Add --version argument

2 files changed, 8 insertions, 0 deletions

diff --git a/firewatch/__init__.py b/firewatch/__init__.py
index e69de29..0eea61c 100644
--- a/firewatch/__init__.py
+++ b/firewatch/__init__.py
@@ -0,0 +1 @@
+from .version import *
diff --git a/firewatch/firewatch.py b/firewatch/firewatch.py
index df17e8a..5360ad2 100644
--- a/firewatch/firewatch.py
+++ b/firewatch/firewatch.py
@@ -189,8 +189,15 @@ def main():
                         help=f'i[{"|".join([x for x in time_units.keys()])}]'
                         ' (120s, 12h, 1d, 2w, 3m, 4y)')
 
+    parser.add_argument('--version', '-V', action='store_true',
+                        help='Display software version')
     args = parser.parse_args()
 
+    if args.version:
+        from . import __version__
+        print(__version__)
+        exit(0)
+
     order = False  # Ascending
     if args.order != 'asc':
         order = True  # Descending