diff options
author | Markus Holtermann <info@markusholtermann.eu> | 2014-10-10 01:01:18 +0200 |
---|---|---|
committer | Markus Holtermann <info@markusholtermann.eu> | 2014-10-10 01:01:18 +0200 |
commit | 52c014264138af5d4f8a7b34fdf431abe689943d (patch) | |
tree | 36a690ab60a6c0dfcd8eab8514061c02812796cb /bvg-grabber.py | |
parent | e1cb4f1d11c6f9bd7304b608023abc52a33f9026 (diff) | |
download | bvg-grabber-52c014264138af5d4f8a7b34fdf431abe689943d.tar.gz bvg-grabber-52c014264138af5d4f8a7b34fdf431abe689943d.tar.bz2 bvg-grabber-52c014264138af5d4f8a7b34fdf431abe689943d.zip |
Fixed #8 -- Added support for limit to ActualDepartureQueryApi
Diffstat (limited to 'bvg-grabber.py')
-rwxr-xr-x | bvg-grabber.py | 12 |
1 files changed, 5 insertions, 7 deletions
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', '-'): |