diff options
Diffstat (limited to 'shopdb/__version__.py')
-rw-r--r-- | shopdb/__version__.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/shopdb/__version__.py b/shopdb/__version__.py new file mode 100644 index 0000000..4d42882 --- /dev/null +++ b/shopdb/__version__.py @@ -0,0 +1,29 @@ +"""Define version.""" + + +import pathlib + + +here = pathlib.Path(__file__).resolve().parent + + +__all__ = ["__version__", "__branch__"] + + +# read from VERSION file +with open(str(here / "VERSION")) as f: + __version__ = f.read().strip() + + +# add git branch, if appropriate +p = here.parent / ".git" +if p.is_file(): + with open(str(p)) as f: + p = p.parent / f.readline()[8:].strip() # Strip "gitdir: " +p = p / "HEAD" +if p.exists(): + with open(str(p)) as f: + __branch__ = f.readline().rstrip().split(r"/")[-1] + __version__ += "+" + __branch__ +else: + __branch__ = "" |