i

void call(Map config = [:], Closure body) {
    defaultConfig = [
        pythonPath: 'python3',
        recreateVenv: false,
        venvDir: '.venv',
        updatePip: true,
        installPipTools: true,
        requirementsFile: 'requirements.in',
        stepName: 'python'
    ]
    runConfig = defaultConfig + config

    venvDir = [
      env.WORKSPACE,
      runConfig.venvDir
    ]
    venvBinDir = venvDir + 'bin'
    venvPython = venvBinDir + 'python'

    venvEnv = [
      "VIRTUAL_ENV=${venvDir.join('/')}",
      "PATH+CUSTOM=${venvBinDir.join('/')}"
    ]

    stage(runConfig.stepName) {
        printBanner(runConfig.stepName)
        ansiColor('css') {
            if (fileExists(runConfig.venvDir) && runConfig.recreateVenv)  {
                echo 'Removing old venv...'
                dir(runConfig.venvDir) {
                    deleteDir()
                }
            }

            if (!fileExists(runConfig.venvDir)) {
                createVenv = "${runConfig.pythonPath} -m venv ${runConfig.venvDir}"
                echo "python create venv cmd is: ${createVenv}"
                sh createVenv
            }

            withEnv(venvEnv) {
                List setupCmds = []
                if (runConfig.updatePip) {
                    updateCmd = "${venvPython.join('/')} -m pip install --upgrade pip"
                    setupCmds.add(updateCmd)
                }
                if (runConfig.installPipTools) {
                    installCmd = "${venvPython.join('/')} -m pip install pip-tools"
                    setupCmds.add(installCmd)
                }
                if (fileExists(runConfig.requirementsFile)) {
                    reqCmd = "pip-compile ${runConfig.requirementsFile}"
                    setupCmds.add(reqCmd)
                }
                if (!setupCmds.isEmpty()) {
                    sh setupCmds.join('\n')
                }
                echo 'Executing closure...'
                body()
            }
        }
    }
}