Merge pull request #5 from pyfidelity/version-injection

Version injection

README.rst

@@ -30,6 +30,10 @@ this package allows to extract it from the underlying Git repository:
 Changes
 -------
 
+1.0.4 - 2016-06-22
+
+- [feature] allow to build a package using a git-based version without modifying it upfront
+
 1.0.3 - 2015-04-23
 ++++++++++++++++++
 

setup.py

@@ -3,7 +3,7 @@
 
 setup(
     name='setuptools-git-version',
-    version='1.0.3',
+    version='1.0.4',
     url='https://github.com/pyfidelity/setuptools-git-version',
     author='pyfidelity UG',
     author_email='mail@pyfidelity.com',

setuptools_git_version.py

@@ -3,6 +3,7 @@
 
 
 command = 'git describe --tags --long --dirty'
+fmt = '{tag}.{commitcount}+{gitsha}'
 
 
 def validate_version_format(dist, attr, value):
@@ -15,11 +16,29 @@ def validate_version_format(dist, attr, value):
     dist.metadata.version = version
 
 
-def format_version(version, fmt):
+def format_version(version, fmt=fmt):
     parts = version.split('-')
     assert len(parts) in (3, 4)
     dirty = len(parts) == 4
     tag, count, sha = parts[:3]
     if count == '0' and not dirty:
         return tag
     return fmt.format(tag=tag, commitcount=count, gitsha=sha.lstrip('g'))
+
+
+if __name__ == "__main__":
+    # determine version from git
+    git_version = check_output(command.split()).decode('utf-8').strip()
+    git_version = format_version(version=git_version)
+
+    # monkey-patch `setuptools.setup` to inject the git version
+    import setuptools
+    original_setup = setuptools.setup
+
+    def setup(version=None, *args, **kw):
+        return original_setup(version=git_version, *args, **kw)
+
+    setuptools.setup = setup
+
+    # import the packages's setup module
+    import setup