-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
executable file
·95 lines (77 loc) · 3.5 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
from setuptools import setup
from setuptools import find_packages
import os
import subprocess
import re
import io
PACKAGE_NAME = 'regtricks'
ROOTDIR = os.path.abspath(os.path.dirname(__file__))
def get_filetext(filename):
""" Get the text of a local file """
with io.open(os.path.join(ROOTDIR, filename), encoding='utf-8') as f:
return f.read()
def git_version():
""" Get the full and python standardized version from Git tags (if possible) """
try:
# Full version includes the Git commit hash
full_version = subprocess.check_output('git describe', shell=True).decode("utf-8").strip(" \n")
# Python standardized version in form major.minor.patch.post<build>
version_regex = re.compile(r"v?(\d+\.\d+\.\d+(-\d+)?).*")
match = version_regex.match(full_version)
if match:
std_version = match.group(1).replace("-", ".post")
else:
raise RuntimeError("Failed to parse version string %s" % full_version)
return full_version, std_version
except Exception:
# Any failure, return None. We may not be in a Git repo at all
return None, None
def git_timestamp():
""" Get the last commit timestamp from Git (if possible)"""
try:
return subprocess.check_output('git log -1 --format=%cd', shell=True).decode("utf-8").strip(" \n")
except Exception:
# Any failure, return None. We may not be in a Git repo at all
return None
def update_metadata(version_str, timestamp_str):
""" Update the version and timestamp metadata in the module _version.py file """
with io.open(os.path.join(ROOTDIR, PACKAGE_NAME, "_version.py"), "w", encoding='utf-8') as f:
f.write("__version__ = '%s'\n" % version_str)
f.write("__timestamp__ = '%s'\n" % timestamp_str)
def get_requirements():
""" Get a list of all entries in the requirements file """
with io.open(os.path.join(ROOTDIR, 'requirements.txt'), encoding='utf-8') as f:
# print([l.strip() for l in f.readlines()])
return [l.strip() for l in f.readlines()]
def get_version():
""" Get the current version number (and update it in the module _version.py file if necessary)"""
version, timestamp = git_version()[1], git_timestamp()
if version is not None and timestamp is not None:
# We got the metadata from Git - update the version file
update_metadata(version, timestamp)
else:
# Could not get metadata from Git - use the version file if it exists
try:
with io.open(os.path.join(ROOTDIR, PACKAGE_NAME, '_version.py'), encoding='utf-8') as f:
md = f.read()
match = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]", md, re.M)
if match:
version = match.group(1)
else:
raise ValueError("Stored version could not be parsed")
except (IOError, ValueError):
version = "unknown"
update_metadata(version, "unknown")
return version
if __name__ == "__main__":
setup(name=PACKAGE_NAME,
version=get_version(),
description="Tools for manipulating and applying registrations",
long_description=get_filetext('README.md'),
long_description_content_type='text/markdown',
author='Tom Kirk',
author_email='[email protected]',
license='BSD-3-clause',
url='https://github.com/tomfrankkirk/regtricks',
install_requires=get_requirements(),
packages=find_packages())