def metrics_by_interval():
route_id = request.args.get('route_id')
if route_id is None:
route_id = '12'
start_stop_id = request.args.get('start_stop_id')
if start_stop_id is None:
start_stop_id = '3476'
end_stop_id = request.args.get('end_stop_id')
direction_id = request.args.get('direction_id')
start_date_str = request.args.get('start_date')
end_date_str = request.args.get('end_date')
date_str = request.args.get('date')
if date_str is not None:
start_date_str = end_date_str = date_str
else:
if start_date_str is None:
start_date_str = '2019-04-08'
if end_date_str is None:
end_date_str = start_date_str
start_time_str = request.args.get('start_time') # e.g. "14:00" (24h time of day)
end_time_str = request.args.get('end_time') # e.g. "18:00" (24h time of day)
params = {
'start_stop_id': start_stop_id,
'end_stop_id': end_stop_id,
'route_id': route_id,
'direction_id': direction_id,
'start_date': start_date_str,
'end_date': end_date_str,
'date': date_str,
'start_time': start_time_str,
'end_time': end_time_str,
}
data = {
'params': params
}
if start_time_str is not None and end_time_str is not None:
# round start_time down and end_time up to allow for even intervals
start_time = datetime.strptime(start_time_str, '%H:%M').replace(microsecond=0, second=0, minute=0)
end_time = datetime.strptime(end_time_str, '%H:%M').replace(microsecond=0, second=0, minute=0) - timedelta(hours=1)
time_str_intervals = []
while start_time.hour != end_time.hour + 1:
time_str_intervals.append((
start_time.strftime('%H:%M'),
(start_time + timedelta(seconds=3600)).strftime('%H:%M')
))
start_time += timedelta(seconds=3600)
else:
if start_time_str or end_time_str:
return make_error_response(params, f'Need both a start and end time', 404)
time_str_intervals = constants.DEFAULT_TIME_STR_INTERVALS
try:
route_metrics = metrics.RouteMetrics('sf-muni', route_id)
dates = util.get_dates_in_range(start_date_str, end_date_str)
keys = ['count','avg','min','median','max','percentiles']
tz = pytz.timezone('US/Pacific')
def get_interval_stats(start_time_str, end_time_str):
rng = metrics.Range(dates, start_time_str, end_time_str, tz)
return {
'start_time': start_time_str,
'end_time': end_time_str,
'wait_times': route_metrics.get_wait_time_stats(
direction_id, start_stop_id,
rng, keys
),
'trip_times': route_metrics.get_trip_time_stats(
direction_id, start_stop_id, end_stop_id,
rng, keys
),
}
data['intervals'] = [
get_interval_stats(start_time_str, end_time_str)
for start_time_str, end_time_str in time_str_intervals
]
except errors.ArrivalHistoryNotFoundError as ex:
return make_error_response(params, str(ex), 404)
except errors.ValidationError as ex:
return make_error_response(params, str(ex), 400)
return Response(json.dumps(data, indent=2), mimetype='application/json')
def metrics_page():
metrics_start = time.time()
route_id = request.args.get('route_id')
if route_id is None:
route_id = '12'
start_stop_id = request.args.get('start_stop_id')
if start_stop_id is None:
start_stop_id = '3476'
end_stop_id = request.args.get('end_stop_id')
direction_id = request.args.get('direction_id')
start_date_str = request.args.get('start_date')
end_date_str = request.args.get('end_date')
date_str = request.args.get('date')
if date_str is not None:
start_date_str = end_date_str = date_str
else:
if start_date_str is None:
start_date_str = '2019-04-08'
if end_date_str is None:
end_date_str = start_date_str
start_time_str = request.args.get('start_time') # e.g. "14:00" (24h time of day)
end_time_str = request.args.get('end_time') # e.g. "18:00" or "03:00+1" (24h time of day)
params = {
'start_stop_id': start_stop_id,
'end_stop_id': end_stop_id,
'route_id': route_id,
'direction_id': direction_id,
'start_date': start_date_str,
'end_date': end_date_str,
'start_time': start_time_str,
'end_time': end_time_str,
}
data = {
'params': params
}
try:
route_config = nextbus.get_route_config('sf-muni', route_id)
start_stop_info = route_config.get_stop_info(start_stop_id)
if start_stop_info is None:
raise errors.ValidationError(f"Stop {start_stop_id} is not on route {route_id}")
data['start_stop_title'] = start_stop_info.title
if end_stop_id:
end_stop_info = route_config.get_stop_info(end_stop_id)
if end_stop_info is None:
raise errors.ValidationError(f"Stop {end_stop_id} is not on route {route_id}")
data['end_stop_title'] = end_stop_info.title
rng = metrics.Range(
util.get_dates_in_range(start_date_str, end_date_str),
start_time_str,
end_time_str,
pytz.timezone('US/Pacific')
)
route_metrics = metrics.RouteMetrics('sf-muni', route_id)
keys = ['count','avg','min','median','max','percentiles','histogram']
data['wait_times'] = route_metrics.get_wait_time_stats(
direction_id, start_stop_id,
rng, keys
)
data['trip_times'] = route_metrics.get_trip_time_stats(
direction_id, start_stop_id, end_stop_id,
rng, keys
)
data['headway_min'] = route_metrics.get_headway_min_stats(
direction_id, start_stop_id,
rng, keys
)
except errors.ArrivalHistoryNotFoundError as ex:
return make_error_response(params, str(ex), 404)
except errors.ValidationError as ex:
return make_error_response(params, str(ex), 400)
metrics_end = time.time()
data['processing_time'] = (metrics_end - metrics_start)
res = Response(json.dumps(data, indent=2), mimetype='application/json')
if not DEBUG:
res.headers['Cache-Control'] = 'max-age=60'
return res
print(f"Start: {start_time}")
tt = timetable.get_timetable_from_csv(agency_id, route, d, ver)
rc = agency.get_route_config(route)
for stop in stops:
# get direction
nextbus_dir = rc.get_directions_for_stop(stop)
if len(nextbus_dir) == 0:
print(f"Stop {stop} has no directions.")
else:
for direction in nextbus_dir:
tt.pretty_print(stop, direction)
if comparison:
route_metrics = metrics.RouteMetrics(agency_id, route)
df = route_metrics.get_comparison_to_timetable(
d, stop, direction)
if len(df) > 0:
df = df.rename(
{
"arrival_time":
"Scheduled Arrival",
"arrival_headway":
"Scheduled Headway",
"next_arrival":
"Next Arrival",
"next_arrival_delta":
"Delta (Next Arrival)",
"next_arrival_headway":