Fix incorrect imputation of City Loop stations for Flinders Street direct trains
This commit is contained in:
parent
c351302f71
commit
66058888ba
@ -70,7 +70,12 @@ def parse_departure(departure, departures, timenow):
|
||||
stops['stops'].sort(key=lambda x: x['stop_sequence'])
|
||||
route_stops[(departure['route_id'], departure['direction_id'])] = stops['stops']
|
||||
|
||||
route_stops_dir = route_stops[(departure['route_id'], departure['direction_id'])]
|
||||
route_stops_dir = []
|
||||
for stop in route_stops[(departure['route_id'], departure['direction_id'])]:
|
||||
# Cut off at Flinders Street
|
||||
route_stops_dir.append(stop)
|
||||
if stop_to_name(stop, departure['route_id']) == 'Flinders Street':
|
||||
break
|
||||
|
||||
# Calculate stopping pattern
|
||||
result['stops'] = []
|
||||
@ -79,10 +84,18 @@ def parse_departure(departure, departures, timenow):
|
||||
ps_route_start = next((k for k, stop in enumerate(pattern_stops) if stop[0] == int(flask.request.args['stop_id'])), 0)
|
||||
ps_route_end = next((k for k, stop in enumerate(pattern_stops) if stop[0] == pattern['runs'][str(departure['run_id'])]['final_stop_id']), len(pattern_stops)-1)
|
||||
|
||||
ps_route_flinders = next((k for k, stop in enumerate(pattern_stops) if stop[0] == 1071), len(pattern_stops)-1)
|
||||
if ps_route_flinders > ps_route_start:
|
||||
ps_route_end = min(ps_route_end, ps_route_flinders) # Cut off at Flinders Street
|
||||
|
||||
# Find this run along route_stops_dir
|
||||
rsd_route_start = next((k for k, stop in enumerate(route_stops_dir) if stop['stop_id'] == int(flask.request.args['stop_id'])), 0)
|
||||
rsd_route_end = next((k for k, stop in enumerate(route_stops_dir) if stop['stop_id'] == pattern['runs'][str(departure['run_id'])]['final_stop_id']), len(route_stops_dir)-1)
|
||||
|
||||
rsd_route_flinders = next((k for k, stop in enumerate(route_stops_dir) if stop['stop_id'] == 1071), len(route_stops_dir)-1)
|
||||
if rsd_route_flinders > rsd_route_start:
|
||||
rsd_route_end = min(rsd_route_end, rsd_route_flinders)
|
||||
|
||||
# Add express stops
|
||||
for k, stop in enumerate(route_stops_dir[rsd_route_start:rsd_route_end+1]):
|
||||
if any(x[0] == stop['stop_id'] for x in pattern_stops):
|
||||
@ -113,15 +126,18 @@ def parse_departure(departure, departures, timenow):
|
||||
num_city_loop += 1
|
||||
|
||||
# Impute remainder of journey
|
||||
if result['stops'][-1] == 'Parliament':
|
||||
if result['stops'] and result['stops'][-1] == 'Parliament':
|
||||
result['stops'].append('Melbourne Central')
|
||||
result['stops'].append('Flagstaff')
|
||||
result['stops'].append('Southern Cross')
|
||||
result['stops'].append('Flinders Street')
|
||||
num_city_loop += 4
|
||||
|
||||
result['dest'] = result['stops'][-1]
|
||||
if result['dest'] in LOOP_STATIONS:
|
||||
result['dest'] = result['stops'][-1] if result['stops'] else None
|
||||
if result['dest'] is None:
|
||||
result['dest'] = 'Arrival'
|
||||
result['desc'] = ''
|
||||
elif result['dest'] in LOOP_STATIONS:
|
||||
# Up train
|
||||
# Is this a City Loop train?
|
||||
if num_city_loop >= 3:
|
||||
|
Loading…
Reference in New Issue
Block a user