Add __version__ and derive package version from it

2 files changed, 14 insertions, 1 deletions

diff --git a/jwt/__init__.py b/jwt/__init__.py
index 2f9e4d6..0a1dc23 100644
--- a/jwt/__init__.py
+++ b/jwt/__init__.py
@@ -25,6 +25,7 @@ if sys.version_info >= (3, 0, 0):
     basestring = str
 
 
+__version__ = '0.2.2'
 __all__ = ['encode', 'decode', 'DecodeError']
 
 
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',