fixes #18

agilescientific · Dec 9, 2016 · 59daa60 · 59daa60
1 parent 9f18a32
commit 59daa60
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/seisplot.py b/seisplot.py
@@ -12,6 +12,7 @@
 import glob
 import re
 import datetime
+import sys
 
 import yaml
 import numpy as np
@@ -377,7 +378,6 @@ def main(target, cfg):
 
 if __name__ == "__main__":
 
-    Notice.title()
     parser = argparse.ArgumentParser(description='Plot a SEGY file.')
     parser.add_argument("-c", "--config",
                         metavar="config file",
@@ -406,7 +406,14 @@ def main(target, cfg):
     parser.add_argument('-d', '--demo',
                         action='store_true',
                         help='Run with the demo file, data/31_81_PR.png.')
+    parser.add_argument('-v', '--version',
+                        action='store_true',
+                        help='Get the version number.')
     args = parser.parse_args()
+    if args.version:
+        Notice.info(__version__)
+        sys.exit()
+    Notice.title()
     target = args.filename
     with args.config as f:
         cfg = yaml.load(f)