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