add freeze_debianver

1 files changed, 68 insertions, 2 deletions

diff --git a/setup.py b/setup.py
index 8d0c218..777a051 100644
--- a/setup.py
+++ b/setup.py
@@ -18,6 +18,7 @@
 setup file for leap.mx
 """
 import os
+import re
 from setuptools import setup, find_packages
 
 import versioneer
@@ -43,6 +44,68 @@ trove_classifiers = [
     'Topic :: Security :: Cryptography',
 ]
 
+DOWNLOAD_BASE = ('https://github.com/leapcode/leap_mx/'
+                 'archive/%s.tar.gz')
+_versions = versioneer.get_versions()
+VERSION = _versions['version']
+VERSION_FULL = _versions['full']
+DOWNLOAD_URL = ""
+
+# get the short version for the download url
+_version_short = re.findall('\d+\.\d+\.\d+', VERSION)
+if len(_version_short) > 0:
+    VERSION_SHORT = _version_short[0]
+    DOWNLOAD_URL = DOWNLOAD_BASE % VERSION_SHORT
+
+cmdclass = versioneer.get_cmdclass()
+
+
+from setuptools import Command
+
+
+class freeze_debianver(Command):
+    """
+    Freezes the version in a debian branch.
+    To be used after merging the development branch onto the debian one.
+    """
+    user_options = []
+
+    def initialize_options(self):
+        pass
+
+    def finalize_options(self):
+        pass
+
+    def run(self):
+        proceed = str(raw_input(
+            "This will overwrite the file _version.py. Continue? [y/N] "))
+        if proceed != "y":
+            print("He. You scared. Aborting.")
+            return
+        template = r"""
+# This file was generated by the `freeze_debianver` command in setup.py
+# Using 'versioneer.py' (0.7+) from
+# revision-control system data, or from the parent directory name of an
+# unpacked source archive. Distribution tarballs contain a pre-generated copy
+# of this file.
+
+version_version = '{version}'
+version_full = '{version_full}'
+"""
+        templatefun = r"""
+
+def get_versions(default={}, verbose=False):
+        return {'version': version_version, 'full': version_full}
+"""
+        subst_template = template.format(
+            version=VERSION_SHORT,
+            version_full=VERSION_FULL) + templatefun
+        with open(versioneer.versionfile_source, 'w') as f:
+            f.write(subst_template)
+
+
+cmdclass["freeze_debianver"] = freeze_debianver
+
 if os.environ.get("VIRTUAL_ENV", None):
     data_files = None
 else:
@@ -54,12 +117,15 @@ else:
 
 setup(
     name='leap.mx',
-    version=versioneer.get_version(),
-    cmdclass=versioneer.get_cmdclass(),
+    version=VERSION,
+    cmdclass=cmdclass,
     url="http://github.com/leapcode/leap_mx",
+    download_url=DOWNLOAD_URL,
     license='AGPLv3+',
     author='The LEAP Encryption Access Project',
     author_email='info@leap.se',
+    maintainer='Kali Kaneko',
+    maintainer_email='kali@leap.se',
     description=("An asynchronous, transparently-encrypting remailer "
                  "for the LEAP platform"),
     long_description=(