dirty version check

keotl · Jan 26, 2019 · 5a16dc8 · 5a16dc8
1 parent 2f05c82
commit 5a16dc8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -14,8 +14,11 @@
         init_file = f.read()
     if "@@VERSION@@" in init_file:
         with open(path.join(here, 'jivago/__init__.py'), 'w', encoding='utf-8') as f:
-            f.write(init_file.replace("@@VERSION@@",
-                                      check_output(git_version_command.split()).decode('utf-8').strip().split("-")[0]))
+            raw_version_info = check_output(git_version_command.split()).decode('utf-8').strip().split("-")
+            version_number = raw_version_info[0]
+            if not raw_version_info[1] == '0':
+                version_number += raw_version_info[2]
+            f.write(init_file.replace("@@VERSION@@", version_number))
 
 setup(
     name='jivago',