add git_version() to setup.py

Author: Ilan Schnell

setup.py

@@ -59,8 +59,8 @@
 ISRELEASED          = False
 VERSION             = '%d.%d.%d' % (MAJOR, MINOR, MICRO)
 
-# Return the svn version as a string, raise a ValueError otherwise
-def svn_version():
+# Return the git revision as a string
+def git_version():
     def _minimal_ext_cmd(cmd):
         # construct minimal environment
         env = {}
@@ -72,30 +72,17 @@ def _minimal_ext_cmd(cmd):
         env['LANGUAGE'] = 'C'
         env['LANG'] = 'C'
         env['LC_ALL'] = 'C'
-        out = subprocess.Popen(cmd,
-                               stdout=subprocess.PIPE, env=env).communicate()[0]
+        out = subprocess.Popen(cmd, stdout = subprocess.PIPE,
+                               env=env).communicate()[0]
         return out
 
     try:
-        out = _minimal_ext_cmd(['svn', 'info'])
+        out = _minimal_ext_cmd(['git', 'rev-parse', 'HEAD'])
+        GIT_REVISION = out.strip().decode('ascii')
     except OSError:
-        print(" --- Could not run svn info --- ")
-        return ""
+        GIT_REVISION = "Unknown"
 
-    r = re.compile('Revision: ([0-9]+)')
-    svnver = ""
-
-    out = out.decode()
-
-    for line in out.split('\n'):
-        m = r.match(line.strip())
-        if m:
-            svnver = m.group(1)
-
-    if not svnver:
-        print("Error while parsing svn version")
-
-    return svnver
+    return GIT_REVISION
 
 # BEFORE importing distutils, remove MANIFEST. distutils doesn't properly
 # update it when the contents of directories change.
@@ -107,36 +94,36 @@ def _minimal_ext_cmd(cmd):
 # a lot more robust than what was previously being used.
 builtins.__NUMPY_SETUP__ = True
 
-FULLVERSION = VERSION
-if not ISRELEASED:
-    FULLVERSION += '.dev'
-    # If in git or something, bypass the svn rev
-    if os.path.exists('.svn'):
-        FULLVERSION += svn_version()
-
 def write_version_py(filename='numpy/version.py'):
     cnt = """
 # THIS FILE IS GENERATED FROM NUMPY SETUP.PY
-short_version='%(version)s'
-version='%(version)s'
-release=%(isrelease)s
+short_version = '%(version)s'
+version = '%(version)s'
+full_version = '%(full_version)s'
+git_revision = '%(git_revision)s'
+release = %(isrelease)s
 
 if not release:
-    version += '.dev'
-    import os
-    svn_version_file = os.path.join(os.path.dirname(__file__),
-                                   'core','__svn_version__.py')
-    if os.path.isfile(svn_version_file):
-        import imp
-        svn = imp.load_module('numpy.core.__svn_version__',
-                              open(svn_version_file),
-                              svn_version_file,
-                              ('.py','U',1))
-        version += svn.version
+    version = full_version
 """
+    FULL_VERSION = VERSION
+    if not ISRELEASED:
+        if os.path.exists('.git'):
+            GIT_REVISION = git_version()
+        elif os.path.exists(filename):
+            # must be a source distribution, use existing version file
+            from numpy.version import git_revision as GIT_REVISION
+        else:
+            GIT_REVISION = "Unknown"
+
+        FULL_VERSION += '.dev-' + GIT_REVISION[:7]
+
     a = open(filename, 'w')
     try:
-        a.write(cnt % {'version': VERSION, 'isrelease': str(ISRELEASED)})
+        a.write(cnt % {'version': VERSION,
+                       'full_version' : FULL_VERSION,
+                       'git_revision' : GIT_REVISION,
+                       'isrelease': str(ISRELEASED)})
     finally:
         a.close()