version

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__ = ""