From d88697dfaf9f6b881f488f6f40762d6e5e08cbae Mon Sep 17 00:00:00 2001 From: Thomas Perl Date: Wed, 28 Sep 2011 20:54:41 +0200 Subject: [PATCH] scotty: Ask interactively for origin and destination --- scotty | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/scotty b/scotty index 4f88de1..30a701f 100755 --- a/scotty +++ b/scotty @@ -16,11 +16,20 @@ from gotovienna.routing import * parser = argparse.ArgumentParser(description='Get public transport route for Vienna') parser.add_argument('-ot', metavar='type', type=str, help='origin type: %s' % ' | '.join(POSITION_TYPES), default='stop', choices=POSITION_TYPES) parser.add_argument('-dt', metavar='type', type=str, help='destination type: %s' % ' | '.join(POSITION_TYPES), default='stop', choices=POSITION_TYPES) -parser.add_argument('origin') -parser.add_argument('destination') +parser.add_argument('origin', nargs='?') +parser.add_argument('destination', nargs='?') args = parser.parse_args() + +if not args.origin: + args.origin = raw_input('Origin: ') + +if not args.destination: + args.destination = raw_input('Destination: ') + +print >>sys.stderr, 'Searching...', html = search((args.origin, args.ot), (args.destination, args.dt)).read() +print >>sys.stderr, 'done.' parser = sParser(html) state = parser.check_page() -- 1.7.9.5