better sdist, use command for generating gitrev

diff --git a/setup.py b/setup.py
index 417716184f7ec6df7b92a4f58b5ccccd3526157a..7c52b7ef73809400f3fc53ada6f7eea21bc85d61 100644 (file)
--- a/setup.py
+++ b/setup.py
@@ -3,9 +3,6 @@ from distutils.sysconfig import get_python_lib
 import os, os.path
 import sys
 
-if 'sdist' in sys.argv and sys.platform != "win32" and os.path.isdir('.git'):
-    assert os.system("git show-ref -s HEAD > .gitrev") == 0
-
 if sys.platform == "darwin":
     # Don't create resource files on OS X tar.
     os.environ['COPY_EXTENDED_ATTRIBUTES_DISABLE'] = 'true'
@@ -18,6 +15,15 @@ def add_command_class(name, cls):
     cmdclasses[name] = cls
     setup_args['cmdclass'] = cmdclasses
 
+from distutils.command.sdist import sdist as sdist_orig
+class sdist(sdist_orig):
+    def run(self):
+        if (sys.platform != "win32" and 
+            os.path.isdir('.git')):
+            assert os.system("git show-ref -s HEAD > .gitrev") == 0
+        sdist_orig.run(self)
+add_command_class('sdist', sdist)
+
 if sys.version_info[0] >= 3:
     import lib2to3.refactor
     from distutils.command.build_py \