Modify setup.py to accept VERSION as an argument as well as an environment variable.

1 files changed, 4 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index c70d580..fab0326 100755
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,10 @@ import io
 
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'pym'))
 
-__version__ = os.getenv('VERSION', default=os.getenv('PVR', default='9999'))
+if ( len(sys.argv) > 2 ) and ( sys.argv[1] == "set_version" ):
+	__version__ = sys.argv[2]
+else:
+	__version__ = os.getenv('VERSION', default=os.getenv('PVR', default='9999'))
 
 cwd = os.getcwd()