Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Branch: experimental
Fetching contributors…

Cannot retrieve contributors at this time

executable file 110 lines (89 sloc) 3.821 kB
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import sys
from gotovienna import argparse
from gotovienna.utils import *
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=None, choices=POSITION_TYPES)
parser.add_argument('-dt', metavar='type', type=str,
help='destination type: %s' % ' | '.join(POSITION_TYPES),
default=None, choices=POSITION_TYPES)
parser.add_argument('origin', nargs='?',
help='origin station name')
parser.add_argument('destination', nargs='?',
help='destination station name')
args = parser.parse_args()
if not args.origin:
args.origin = raw_input('Origin: ')
if not args.destination:
args.destination = raw_input('Destination: ')
def do_search(args):
if isinstance(args.origin, unicode):
args.origin = args.origin.encode('utf-8', 'ignore')
elif isinstance(args.destination, unicode):
args.destination = args.destination.encode('utf-8', 'ignore')
result = search((args.origin, args.ot),
(args.destination, args.dt))
return sParser(result.read())
print >>sys.stderr, 'Searching...\n',
parser = do_search(args)
print >>sys.stderr, 'done.'
while parser.state == PageType.CORRECTION:
origin_corr, destination_corr = parser.get_correction()
if origin_corr:
print
print '* Origin ambiguous:'
lo = None
while not lo or not lo.isdigit() or int(lo) > len(origin_corr):
for idx, correction in enumerate(origin_corr):
print '%3d. %s' % (idx+1, correction)
lo = sys.stdin.readline().strip()
args.origin = origin_corr[int(lo) - 1]
if destination_corr:
print
print '* Destination ambiguous:'
ld = None
while not ld or not ld.isdigit() or int(ld) > len(destination_corr):
for idx, correction in enumerate(destination_corr):
print '%3d. %s' % (idx+1, correction)
ld = sys.stdin.readline().strip()
args.destination = destination_corr[int(ld) - 1]
parser = do_search(args)
if parser.state == PageType.RESULT:
parser = parser.get_result()
overviews = parser.overview
details = parser.details
l = ''
while not l == 'q':
for idx, overview in enumerate(overviews):
timespan = overview['timespan']
if not timespan:
# XXX: Bogus data for e.g. Pilgramgasse->Karlsplatz?!
continue
str_timespan = timespan[0].strftime('[%Y-%d-%m] %H:%M')
str_timespan += '-' + timespan[1].strftime('%H:%M')
timedelta = timespan[1] - timespan[0]
print '%d. %s (%s)' % (idx + 1,
str_timespan,
timedelta)
print 'q. Quit'
l = sys.stdin.readline().strip()
print
print '~' * 79
if l.isdigit() and int(l) <= len(details):
for detail in details[int(l) - 1]:
if detail['timespan'] and detail['station']:
time = '%s - %s' % (detail['timespan'][0].strftime(TIMEFORMAT),
detail['timespan'][1].strftime(TIMEFORMAT))
print '[%s] %s\n%s' % (time,
' -> '.join(detail['station']),
'\n'.join(detail['info']))
else:
print '\n'.join(detail['info'])
print '-' * 79
print
else:
print 'Error - unknown page returned.'
Jump to Line
Something went wrong with that request. Please try again.