def get_next_stop_times(stop_id, route_id, next_times=3): now = datetime.now() two_min_ago = now - timedelta(minutes=2) service_ids = get_service_ids_for_date() stop_times = StopTime.objects.filter(stop=stop_id, trip__route=route_id, trip__service__in=service_ids, departure_time__gt=gtfs.datetime_to_string(two_min_ago)).order_by('departure_time', 'stop_sequence')[0:next_times] # print connection.queries # TODO: show AM times as end of the previous day. times = formats.stop_times_to_dict(stop_times) return times
def get_schedule(stop_id, route_id, date=date.today()): service_ids = get_service_ids_for_date(date) stop_times = StopTime.objects.filter(stop=stop_id, trip__route=route_id, trip__service__in=service_id).order_by('departure_time', 'stop_sequence') times = formats.stop_times_to_dict(stop_times) return times