Skip to content

Instantly share code, notes, and snippets.

@justinTM

justinTM/write_previous_deployment_yml.py

Created April 6, 2022 20:29
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save justinTM/f7e7e7e55a825b0ca32ba910e5deba0d to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save justinTM/f7e7e7e55a825b0ca32ba910e5deba0d to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. justinTM created this gist Apr 6, 2022.
    63 changes: 63 additions & 0 deletions write_previous_deployment_yml.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,63 @@
    import gitlab
    import os
    import subprocess
    from gitlab.v4.objects.jobs import ProjectJob
    from gitlab.v4.objects.pipelines import ProjectPipeline



    def pprint_job(job: ProjectJob):
    return f"ProjectJob<name={job.attributes.get('name', None)}, finished_at={job.attributes.get('finished_at')[0:16]}>"



    def pprint_pipeline(p: ProjectPipeline):
    return f"ProjectPipeline<id={p.get_id()}, commit_sha={p.attributes.get('sha')[0:8]}>"



    def download_artifact_locally_with_python(gl: gitlab.Gitlab):
    '''When running locally, use developer's with GitLab Artifacts API to download last successful master deployment file.
    '''

    # get last successful master branch Build Job Definition job
    build_job: ProjectJob = None
    print(f"searching pipelines for successful master job {os.environ['BUILD_JOB_NAME']}...")
    project = gl.projects.get(os.environ['CI_PROJECT_ID'], lazy=False)
    for p in project.pipelines.list():
    if p.attributes.get('ref', None) == 'master':
    for j in p.jobs.list():
    if j.attributes.get('name', None) == os.environ['BUILD_JOB_NAME']:
    if j.attributes.get('status', None) == 'success':
    build_job = project.jobs.get(j.id, lazy=False)
    print(f'found {pprint_job(build_job)} from {pprint_pipeline(p)}')
    break
    if build_job is not None:
    break

    # write previous job definition
    artifact_path = os.environ['FILEPATH_FINAL_JOB_DEFINITION']
    print(f'downloading artifact {artifact_path}...')

    try:
    contents = build_job.artifact(artifact_path) if build_job else b'{}'
    with open(os.environ['FILEPATH_PREVIOUS_JOB_DEFINITION'], mode='wb') as f:
    f.write(contents)
    print(os.environ['FILEPATH_PREVIOUS_JOB_DEFINITION'])
    except gitlab.exceptions.GitlabGetError:
    print(f'ERROR: could not find artifact {artifact_path} from {pprint_job(build_job)}')
    artifacts = build_job.attributes.get('attributes', None)
    print(f'available artifacts: {artifacts}')



    if __name__ == '__main__':
    token: str = None
    gl: gitlab.Gitlab = None
    try:
    gl = gitlab.Gitlab(url='https://gitlab.com', private_token=os.environ['GITLAB_API_TOKEN'])
    except KeyError:
    print('no GITLAB_API_TOKEN found (running on GitLab.com?)')
    print('exiting to run with curl using CI_JOB_TOKEN...')
    else:
    download_artifact_locally_with_python(gl)