Print version during runtime

Version is specified in ulozto_downloader package.
setnicka · Nov 17, 2020 · d6ae9de · d6ae9de
1 parent 7d63317
commit d6ae9de
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -3,12 +3,14 @@
 from distutils.core import setup
 import setuptools  # noqa
 
+from ulozto_downloader import __version__
+
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
 setup(
   name='ulozto-downloader',
-  version='1.6',
+  version=__version__,
   license='MIT',
   description='Uloz.to quick multiple sessions downloader.',
   long_description=long_description,

diff --git a/ulozto-downloader b/ulozto-downloader
@@ -5,7 +5,7 @@ import argparse
 import sys
 import signal
 
-from ulozto_downloader import downloader, captcha
+from ulozto_downloader import downloader, captcha, __version__
 
 
 def main():
@@ -20,6 +20,7 @@ def main():
     parser.add_argument('--output', metavar='DIRECTORY', type=str, default="./", help='Target directory')
     parser.add_argument('--auto-captcha', default=False, action="store_true",
                         help='Try to solve captchas automatically using TensorFlow')
+    parser.add_argument('--version', action='version', version=__version__)
 
     args = parser.parse_args()
 

diff --git a/ulozto_downloader/__init__.py b/ulozto_downloader/__init__.py
@@ -0,0 +1 @@
+__version__ = "1.6.1"