From 164d370ebb8356fa9f397f599dd53702852edc47 Mon Sep 17 00:00:00 2001 From: Markus Holtermann Date: Wed, 8 Oct 2014 00:37:27 +0200 Subject: Update script help --- bvg-grabber.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) (limited to 'bvg-grabber.py') diff --git a/bvg-grabber.py b/bvg-grabber.py index b5aa710..63f37d5 100755 --- a/bvg-grabber.py +++ b/bvg-grabber.py @@ -11,14 +11,16 @@ from bvggrabber.api.scheduleddeparture import ScheduledDepartureQueryApi, Vehicl if __name__ == '__main__': + vehicle_choices = ('S', 'U', 'TRAM', 'BUS', 'FERRY', 'RB', 'IC') + parser = argparse.ArgumentParser( description='Query the BVG-website for departures') parser.add_argument('station', type=str, help='The station to query') - parser.add_argument('file', type=str, help='Path to file') + parser.add_argument('file', type=str, help='Path to file. Use - for stdout') parser.add_argument('--vehicle', type=str, nargs='*', - help='''Vehicles which shall be queried, - if non given actualdepartue (bus) - will be used''') + choices=vehicle_choices, + help='Vehicles which shall be queried, if non given ' + 'actualdepartue (bus) will be used') parser.add_argument('--limit', type=int, help='Max departures to query') args = parser.parse_args() @@ -57,7 +59,7 @@ if __name__ == '__main__': res = query.call() if args.file in ('stdout', '-'): - print(res.to_json, file=sys.stdout) + print(res.to_json) else: file = open(args.file, 'w') print(res.to_json, file=file) -- cgit v1.2.3