Substitute docstring and version from setup.py into pytest_dependency.py.

Author: RKrahl

pytest_dependency.py

@@ -1,14 +1,9 @@
-"""pytest-dependency - Manage dependencies of tests
+"""$DOC"""
 
-This pytest plugin manages dependencies of tests.  It allows to mark
-some tests as dependent from other tests.  These tests will then be
-skipped if any of the dependencies did fail or has been skipped.
-"""
+__version__ = "$VERSION"
 
 import pytest
 
-__version__   = "0.2"
-
 
 class DependencyItemStatus(object):
     """Status of a test item in a dependency manager.

setup.py

@@ -6,17 +6,48 @@
 skipped if any of the dependencies did fail or has been skipped.
 """
 
+__version__ = "0.2"
+
 import sys
 if sys.version_info < (2, 7):
     raise RuntimeError("You are using Python %s.\n"
                        "Please apply python2_6.patch first." 
                        % sys.version.split()[0])
+import os
+import os.path
+import re
 from setuptools import setup
+import setuptools.command.sdist as st_sdist
+
+
+def _filter_file(src, dest, subst):
+    """Copy src to dest doing substitutions on the fly.
+    """
+    substre = re.compile(r'\$(%s)' % '|'.join(subst.keys()))
+    def repl(m):
+        return subst[m.group(1)]
+    with open(src, "rt") as sf, open(dest, "wt") as df:
+        while True:
+            l = sf.readline()
+            if not l:
+                break
+            df.write(re.sub(substre, repl, l))
+
+class sdist(st_sdist.sdist):
+    def make_release_tree(self, base_dir, files):
+        st_sdist.sdist.make_release_tree(self, base_dir, files)
+        if not self.dry_run:
+            src = "pytest_dependency.py"
+            dest = os.path.join(base_dir, src)
+            if hasattr(os, 'link') and os.path.exists(dest):
+                os.unlink(dest)
+            subst = {'DOC': __doc__, 'VERSION': __version__}
+            _filter_file(src, dest, subst)
 
 
 setup(
     name='pytest-dependency',
-    version='0.2',
+    version=__version__,
     description='Manage dependencies of tests',
     author='Rolf Krahl',
     author_email='[email protected]',
@@ -51,4 +82,5 @@
             'dependency = pytest_dependency',
         ],
     },
+    cmdclass = {'sdist': sdist},
 )