sdk/cwl/arvados_version.py

   1 from builtins import str
   2 from builtins import next
   3 # Copyright (C) The Arvados Authors. All rights reserved.
   4 #
   5 # SPDX-License-Identifier: Apache-2.0
   6
   7 import subprocess
   8 import time
   9 import os
  10 import re
  11
  12 SETUP_DIR = os.path.dirname(__file__) or '.'
  13
  14 def git_latest_tag():
  15     gittags = subprocess.check_output(['git', 'tag', '-l']).split()
  16     gittags.sort(key=lambda s: [int(u) for u in s.split(b'.')],reverse=True)
  17     return str(next(iter(gittags)).decode('utf-8'))
  18
  19 def choose_version_from():
  20     sdk_ts = subprocess.check_output(
  21         ['git', 'log', '--first-parent', '--max-count=1',
  22          '--format=format:%ct', os.path.join(SETUP_DIR, "../python")]).strip()
  23     cwl_ts = subprocess.check_output(
  24         ['git', 'log', '--first-parent', '--max-count=1',
  25          '--format=format:%ct', SETUP_DIR]).strip()
  26     if int(sdk_ts) > int(cwl_ts):
  27         getver = os.path.join(SETUP_DIR, "../python")
  28     else:
  29         getver = SETUP_DIR
  30     return getver
  31
  32 def git_timestamp_tag():
  33     gitinfo = subprocess.check_output(
  34         ['git', 'log', '--first-parent', '--max-count=1',
  35          '--format=format:%ct', choose_version_from()]).strip()
  36     return str(time.strftime('.%Y%m%d%H%M%S', time.gmtime(int(gitinfo))))
  37
  38 def save_version(setup_dir, module, v):
  39   with open(os.path.join(setup_dir, module, "_version.py"), 'w') as fp:
  40       return fp.write("__version__ = '%s'\n" % v)
  41
  42 def read_version(setup_dir, module):
  43   with open(os.path.join(setup_dir, module, "_version.py"), 'r') as fp:
  44       return re.match("__version__ = '(.*)'$", fp.read()).groups()[0]
  45
  46 def get_version(setup_dir, module):
  47     env_version = os.environ.get("ARVADOS_BUILDING_VERSION")
  48
  49     if env_version:
  50         save_version(setup_dir, module, env_version)
  51     else:
  52         try:
  53             save_version(setup_dir, module, git_latest_tag() + git_timestamp_tag())
  54         except (subprocess.CalledProcessError, OSError):
  55             pass
  56
  57     return read_version(setup_dir, module)