Add __version__ and derive package version from it

1 files changed, 13 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 0430c78..281bd2c 100755
--- a/setup.py
+++ b/setup.py
@@ -1,14 +1,26 @@
 #!/usr/bin/env python
 import os
+import re
 from setuptools import setup
 
 
+def get_version(package):
+    """
+    Return package version as listed in `__version__` in `init.py`.
+    """
+    init_py = open(os.path.join(package, '__init__.py')).read()
+    return re.search("__version__ = ['\"]([^'\"]+)['\"]", init_py).group(1)
+
+
+version = get_version('jwt')
+
 with open(os.path.join(os.path.dirname(__file__), 'README.md')) as readme:
     long_description = readme.read()
 
+
 setup(
     name='PyJWT',
-    version='0.2.2',
+    version=version,
     author='Jeff Lindsay',
     author_email='progrium@gmail.com',
     description='JSON Web Token implementation in Python',