add gpstime argument parsing to search and analyze

Makes it easier to enter search/analyze times.

locklost/analyze.py

@@ -97,10 +97,11 @@ def analyze_condor(event):
 ##################################################
 
 def _parser_add_arguments(parser):
+    from .util import GPSTimeParseAction
     egroup = parser.add_mutually_exclusive_group(required=True)
-    egroup.add_argument('event', type=float, nargs='?',
+    egroup.add_argument('event', action=GPSTimeParseAction, nargs='?',
                         help="event ID / GPS second")
-    egroup.add_argument('--condor', type=float, nargs=2, metavar='GPS',
+    egroup.add_argument('--condor', action=GPSTimeParseAction, nargs=2, metavar='GPS',
                         help="condor analyze all events within GPS range")
     parser.add_argument('--rerun', action='store_true',
                         help="condor re-analyze events")

locklost/search.py

@@ -105,9 +105,10 @@ def search_iterate(segment=None, event_callback=None, stat_file=None):
 ##################################################
 
 def _parser_add_arguments(parser):
-    parser.add_argument('start', type=float,
+    from .util import GPSTimeParseAction
+    parser.add_argument('start', action=GPSTimeParseAction,
                         help="search start time")
-    parser.add_argument('end', type=float,
+    parser.add_argument('end', action=GPSTimeParseAction,
                         help="search end time")
     parser.add_argument('--condor', action='store_true',
                         help="condor search for events")

locklost/util.py

+import argparse
+
+import gpstime
+
+
+class GPSTimeParseAction(argparse.Action):
+    def __call__(self, parser, namespace, values, option_string=False):
+        try:
+            gps = gpstime.parse(values).gps()
+        except TypeError:
+            gps = [gpstime.parse(value).gps() for value in values]
+        except gpstime.GPSTimeException:
+            parser.error("Could not parse date/time string '{}'".format(values))
+        setattr(namespace, self.dest, gps)