pipeline {
    parameters {
        string(
            defaultValue: 'marcinkostrzewski',
            description: 'Kaggle username',
            name: 'KAGGLE_USERNAME',
            trim: false
        )
        password(
            defaultValue: '',
            description: 'Kaggle token taken from kaggle.json file, as described in https://github.com/Kaggle/kaggle-api#api-credentials',
            name: 'KAGGLE_KEY'
        )
    }

    agent {
        dockerfile {
            additionalBuildArgs "--build-arg KAGGLE_USERNAME=${params.KAGGLE_USERNAME} --build-arg KAGGLE_KEY=${params.KAGGLE_KEY} -t s444409-create-dataset"
        }
    }
    
    stages {
        stage('Get arifacts') {
            steps {
                copyArtifacts fingerprintArtifacts: true, projectName: 's444409-create-dataset', selector: lastSuccessful()
            }
        }

        stage('Train model') {
            steps {
                sh "python train_model.py"
            }
        }
    }

    post {
        always {
            archiveArtifacts artifacts: 'model_out', 
            onlyIfSuccessful: true
        }
    }
}