From 52c014264138af5d4f8a7b34fdf431abe689943d Mon Sep 17 00:00:00 2001 From: Markus Holtermann Date: Fri, 10 Oct 2014 01:01:18 +0200 Subject: Fixed #8 -- Added support for limit to ActualDepartureQueryApi --- bvg-grabber.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) (limited to 'bvg-grabber.py') diff --git a/bvg-grabber.py b/bvg-grabber.py index 63f37d5..3964f83 100755 --- a/bvg-grabber.py +++ b/bvg-grabber.py @@ -21,7 +21,8 @@ if __name__ == '__main__': 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') + parser.add_argument('--limit', type=int, default=9, + help='Max departures to query. Default: 9') args = parser.parse_args() query = None @@ -44,18 +45,15 @@ if __name__ == '__main__': vehicles |= Vehicle.RB elif vehicle == 'IC': vehicles |= Vehicle.IC - limit = 9 - if args.limit: - limit = args.limit - query = ScheduledDepartureQueryApi(args.station, vehicles, limit=limit) + query = ScheduledDepartureQueryApi(args.station, vehicles, limit=args.limit) res = query.call() if bus: - aquery = ActualDepartureQueryApi(args.station) + aquery = ActualDepartureQueryApi(args.station, limit=args.limit) res2 = aquery.call() res.merge(res2) else: - query = ActualDepartureQueryApi(args.station) + query = ActualDepartureQueryApi(args.station, limit=args.limit) res = query.call() if args.file in ('stdout', '-'): -- cgit v1.2.3