Put version in ais.__version__ variable.

schwehr · Oct 27, 2015 · 4ebe26f · 4ebe26f
1 parent 6b7e1f5
commit 4ebe26f
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/VERSION b/VERSION
diff --git a/ais/__init__.py b/ais/__init__.py
@@ -2,3 +2,7 @@
 
 from _ais import decode
 from _ais import DecodeError
+
+
+__license__ = 'Apache 2.0'
+__version__ = '0.16'
diff --git a/setup.py b/setup.py
@@ -5,8 +5,11 @@
 
 from setuptools import setup, find_packages, Extension
 
-with open('VERSION') as f:
-    VERSION = f.readline().strip()
+VERSION = None
+with open(os.path.join('ais', '__init__.py')) as f:
+    for line in f:
+        if line.strip().startswith('__version__'):
+            VERSION = line.split('=')[1].strip().replace('"', '').replace("'", '').strip()
 
 EXTRA_COMPILE_ARGS = []
 if sys.platform in ('darwin', 'linux', 'linux2'):