pipeline {
  	agent {
		dockerfile {
			 additionalBuildArgs "--build-arg KAGGLE_USERNAME=${params.KAGGLE_USERNAME} --build-arg KAGGLE_KEY=51854c15661583860a60db232a7026f0 --build-arg CUTOFF=${params.CUTOFF} -t kgirejko"
		}
	  }
     parameters {
		string(
			defaultValue: 'kgirejko',
			description: 'Kaggle username',
            name: 'KAGGLE_USERNAME',
            trim: false
		)

		password(
			defaultValue: '51854c15661583860a60db232a7026f0',
            description: 'Kaggle token taken from kaggle.json file, as described in https://github.com/Kaggle/kaggle-api#api-credentials',
            name: 'KAGGLE_KEY'
		)
		
		string(
            defaultValue: '1000',
            description: 'Number of epochs',
            name: 'EPOCHS',
            trim: false
        )
     }

   stages {
	stage('Main train'){
		steps {
            copyArtifacts filter: '*', projectName: 's478815-create-dataset'
			sh 'python3 ./DL.py $EPOCHS'
			archiveArtifacts artifacts: 'modelP.pkl', followSymlinks: false
			
            build job: 's478815-evaluation/master/'
		}
	}
    }

	post {
        success {
            emailext body: 'SUCCESS', subject: 's478815-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
        }
        failure {
            emailext body: 'FAILURE', subject: 's478815-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
        }
        unstable {
            emailext body: 'UNSTABLE', subject: 's478815-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
        }
        changed {
            emailext body: 'CHANGED', subject: 's478815-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
        }	
	}
}