16306: Merge branch 'master'
[arvados.git] / sdk / python / 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 import sys
10
11 SETUP_DIR = os.path.dirname(os.path.abspath(__file__))
12 VERSION_PATHS = {
13         SETUP_DIR,
14         os.path.abspath(os.path.join(SETUP_DIR, "../../build/version-at-commit.sh"))
15         }
16
17 def choose_version_from():
18     ts = {}
19     for path in VERSION_PATHS:
20         ts[subprocess.check_output(
21             ['git', 'log', '--first-parent', '--max-count=1',
22              '--format=format:%ct', path]).strip()] = path
23
24     sorted_ts = sorted(ts.items())
25     getver = sorted_ts[-1][1]
26     print("Using "+getver+" for version number calculation of "+SETUP_DIR, file=sys.stderr)
27     return getver
28
29 def git_version_at_commit():
30     curdir = choose_version_from()
31     myhash = subprocess.check_output(['git', 'log', '-n1', '--first-parent',
32                                        '--format=%H', curdir]).strip()
33     myversion = subprocess.check_output([SETUP_DIR+'/../../build/version-at-commit.sh', myhash]).strip().decode()
34     return myversion
35
36 def save_version(setup_dir, module, v):
37     v = v.replace("~dev", ".dev").replace("~rc", "rc")
38     with open(os.path.join(setup_dir, module, "_version.py"), 'wt') as fp:
39         return fp.write("__version__ = '%s'\n" % v)
40
41 def read_version(setup_dir, module):
42     with open(os.path.join(setup_dir, module, "_version.py"), 'rt') as fp:
43         return re.match("__version__ = '(.*)'$", fp.read()).groups()[0]
44
45 def get_version(setup_dir, module):
46     env_version = os.environ.get("ARVADOS_BUILDING_VERSION")
47
48     if env_version:
49         save_version(setup_dir, module, env_version)
50     else:
51         try:
52             save_version(setup_dir, module, git_version_at_commit())
53         except (subprocess.CalledProcessError, OSError) as err:
54             print("ERROR: {0}".format(err), file=sys.stderr)
55             pass
56
57     return read_version(setup_dir, module)
58
59 # Called from calculate_python_sdk_cwl_package_versions() in run-library.sh
60 if __name__ == '__main__':
61     print(get_version(SETUP_DIR, "arvados"))