2 # -*- coding: UTF-8 -*-
7 from gotovienna.utils import *
8 from gotovienna.routing import *
10 parser = argparse.ArgumentParser(description='Get public transport route for Vienna')
11 parser.add_argument('-ot', metavar='type', type=str, help='origin type: %s' % ' | '.join(POSITION_TYPES), default='stop', choices=POSITION_TYPES)
12 parser.add_argument('-dt', metavar='type', type=str, help='destination type: %s' % ' | '.join(POSITION_TYPES), default='stop', choices=POSITION_TYPES)
13 parser.add_argument('origin', nargs='?', help='origin station name')
14 parser.add_argument('destination', nargs='?', help='destination station name')
16 args = parser.parse_args()
19 args.origin = raw_input('Origin: ')
21 if not args.destination:
22 args.destination = raw_input('Destination: ')
25 if isinstance(args.origin, unicode):
26 args.origin = args.origin.encode('utf-8', 'ignore')
27 elif isinstance(args.destination, unicode):
28 args.destination = args.destination.encode('utf-8', 'ignore')
30 result = search((args.origin, args.ot),
31 (args.destination, args.dt))
33 return sParser(result.read())
35 print >>sys.stderr, 'Searching...',
36 parser = do_search(args)
37 print >>sys.stderr, 'done.'
39 while parser.state == PageType.CORRECTION:
40 origin_corr, destination_corr = parser.get_correction()
44 print '* Origin ambiguous:'
46 while not lo or not lo.isdigit() or int(lo) > len(origin_corr):
47 for idx, correction in enumerate(origin_corr):
48 print '%3d. %s' % (idx+1, correction)
49 lo = sys.stdin.readline().strip()
51 args.origin = origin_corr[int(lo) - 1]
55 print '* Destination ambiguous:'
57 while not ld or not ld.isdigit() or int(ld) > len(destination_corr):
58 for idx, correction in enumerate(destination_corr):
59 print '%3d. %s' % (idx+1, correction)
60 ld = sys.stdin.readline().strip()
62 args.destination = destination_corr[int(ld) - 1]
64 parser = do_search(args)
66 if parser.state == PageType.RESULT:
67 parser = parser.get_result()
68 overviews = parser.overview
69 details = parser.details
72 for idx, overview in enumerate(overviews):
73 if not overview['date'] or not overview['time']:
74 # XXX: Bogus data for e.g. Pilgramgasse->Karlsplatz?!
77 print '%d. [%s] %s-%s (%s)' % (idx + 1,
83 l = sys.stdin.readline().strip()
87 if l.isdigit() and int(l) <= len(details):
88 for detail in details[int(l) - 1]:
89 if detail['time'] and detail['station']:
90 time = '%s - %s' % (detail['time'][0].strftime(TIMEFORMAT), detail['time'][1].strftime(TIMEFORMAT))
91 print '[%s] %s\n%s' % (time, ' -> '.join(detail['station']), '\n'.join(detail['info']))
93 print '\n'.join(detail['info'])
97 print 'Error - unknown page returned.'