sdk/cwl/arvados_version.py

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