services/nodemanager/arvados_version.py

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