improve version printing when run from a checkout

evax · May 15, 2012 · f1e8bfb · f1e8bfb
1 parent c1e1e7b
commit f1e8bfb
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/lib/cdist/__init__.py b/lib/cdist/__init__.py
@@ -19,7 +19,17 @@
 #
 #
 
-VERSION     = "2.0.9"
+import os
+import subprocess
+
+try:
+    with open(os.devnull, 'w') as devnull:
+        here = os.path.dirname(os.path.realpath(__file__))
+        VERSION = subprocess.check_output(
+                    'cd "%s" && git describe' % here,
+                    stderr=devnull, shell=True).decode('utf-8')
+except:
+    VERSION     = "2.0.9"
 
 BANNER = """
              ..          .       .x+=:.        s
@@ -38,8 +48,6 @@
 DOT_CDIST   = ".cdist"
 
 
-import os
-
 class Error(Exception):
     """Base exception class for this project"""
     pass