services/dockercleaner/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_version_at_commit():
  11     curdir = os.path.dirname(os.path.abspath(__file__))
  12     myhash = subprocess.check_output(['git', 'log', '-n1', '--first-parent',
  13                                        '--format=%H', curdir]).strip()
  14     myversion = subprocess.check_output([curdir+'/../../build/version-at-commit.sh', myhash]).strip().decode()
  15     return myversion
  16
  17 def save_version(setup_dir, module, v):
  18     v = v.replace("~dev", ".dev").replace("~rc", "rc")
  19     with open(os.path.join(setup_dir, module, "_version.py"), 'wt') as fp:
  20         return fp.write("__version__ = '%s'\n" % v)
  21
  22 def read_version(setup_dir, module):
  23     with open(os.path.join(setup_dir, module, "_version.py"), 'rt') as fp:
  24         return re.match("__version__ = '(.*)'$", fp.read()).groups()[0]
  25
  26 def get_version(setup_dir, module):
  27     env_version = os.environ.get("ARVADOS_BUILDING_VERSION")
  28
  29     if env_version:
  30         save_version(setup_dir, module, env_version)
  31     else:
  32         try:
  33             save_version(setup_dir, module, git_version_at_commit())
  34         except (subprocess.CalledProcessError, OSError):
  35             pass
  36
  37     return read_version(setup_dir, module)