ci/gitlab/trigger_cerbero_pipeline.py

   1 #!/usr/bin/python3
   2
   3 import time
   4 import os
   5 import sys
   6 import gitlab
   7
   8 CERBERO_PROJECT = 'gstreamer/cerbero'
   9
  10
  11 class Status:
  12     FAILED = 'failed'
  13     MANUAL = 'manual'
  14     CANCELED = 'canceled'
  15     SUCCESS = 'success'
  16     SKIPPED = 'skipped'
  17     CREATED = 'created'
  18
  19     @classmethod
  20     def is_finished(cls, state):
  21         return state in [
  22             cls.FAILED,
  23             cls.MANUAL,
  24             cls.CANCELED,
  25             cls.SUCCESS,
  26             cls.SKIPPED,
  27         ]
  28
  29
  30 def fprint(msg):
  31     print(msg, end="")
  32     sys.stdout.flush()
  33
  34
  35 if __name__ == "__main__":
  36     server = os.environ['CI_SERVER_URL']
  37     gl = gitlab.Gitlab(server,
  38                        private_token=os.environ.get('GITLAB_API_TOKEN'),
  39                        job_token=os.environ.get('CI_JOB_TOKEN'))
  40
  41     cerbero = None
  42     cerbero_name = None
  43     # We do not want to run on (often out of date) user upstream branch
  44     if os.environ["CI_COMMIT_REF_NAME"] != os.environ['GST_UPSTREAM_BRANCH']:
  45         try:
  46             cerbero_name = f'{os.environ["CI_PROJECT_NAMESPACE"]}/cerbero'
  47             cerbero = gl.projects.get(cerbero_name)
  48             if os.environ["CI_COMMIT_REF_NAME"] in [b.name for b in cerbero.branches.list()]:
  49                 cerbero_branch = os.environ["CI_COMMIT_REF_NAME"]
  50             else:
  51                 # No branch with a same name on the user cerbero repo... trigger
  52                 # on upstream project
  53                 cerbero = None
  54         except gitlab.exceptions.GitlabGetError:
  55             pass
  56
  57     if cerbero is None:
  58         cerbero_name = CERBERO_PROJECT
  59         cerbero = gl.projects.get(cerbero_name)
  60         cerbero_branch = os.environ["GST_UPSTREAM_BRANCH"]
  61
  62     fprint(f"-> Triggering on branch {cerbero_branch} in {cerbero_name}\n")
  63
  64     # CI_PROJECT_URL is not necessarily the project where the branch we need to
  65     # build resides, for instance merge request pipelines can be run on
  66     # 'gstreamer' namespace. Fetch the branch name in the same way, just in
  67     # case it breaks in the future.
  68     if 'CI_MERGE_REQUEST_SOURCE_PROJECT_URL' in os.environ:
  69         project_url = os.environ['CI_MERGE_REQUEST_SOURCE_PROJECT_URL']
  70         project_branch = os.environ['CI_MERGE_REQUEST_SOURCE_BRANCH_NAME']
  71     else:
  72         project_url = os.environ['CI_PROJECT_URL']
  73         project_branch = os.environ['CI_COMMIT_REF_NAME']
  74
  75     pipe = cerbero.trigger_pipeline(
  76         token=os.environ['CI_JOB_TOKEN'],
  77         ref=cerbero_branch,
  78         variables={
  79             "CI_GSTREAMER_URL": project_url,
  80             "CI_GSTREAMER_REF_NAME": project_branch,
  81             # This tells cerbero CI that this is a pipeline started via the
  82             # trigger API, which means it can use a deps cache instead of
  83             # building from scratch.
  84             "CI_GSTREAMER_TRIGGERED": "true",
  85         }
  86     )
  87
  88     fprint(f'Cerbero pipeline running at {pipe.web_url} ')
  89     while True:
  90         time.sleep(15)
  91         pipe.refresh()
  92         if Status.is_finished(pipe.status):
  93             fprint(f": {pipe.status}\n")
  94             sys.exit(0 if pipe.status == Status.SUCCESS else 1)
  95         else:
  96             fprint(".")