def on_get(self, req, resp, args, route_id):
if args.get('is_active'):
trips = Trip.get_active_trips_for_route(self.session, route_id)
else:
trips = self.session.query(Trip).filter(Trip.route_id == route_id)
trip_schema = TripSchema_v1()
dump_data = trip_schema.dump(trips, many=True).data
body = dict(results=dump_data)
resp.body = json.dumps(body, ensure_ascii=False)
def on_get(self, req, resp, args, stop_id):
trips = Trip.get_trips_for_stop(session=self.session,
stop_id=stop_id,
route_id=args.get('route_id'),
only_active=args.get(
'is_active', False))
trip_schema = TripSchema_v1()
results = trip_schema.dump(trips, many=True).data
body = dict(results=results, )
resp.body = json.dumps(body, ensure_ascii=False)
def setUp(self):
super(StopRoutesTestCase_v3, self).setUp()
self.agency = Agency(agency_id='TS',
agency_name='Transantiago',
agency_url='http://www.transantiago.cl',
agency_timezone='America/Santiago')
self.session.add(self.agency)
self.stop = Stop(
stop_id='PA1',
stop_code='PA1',
stop_name='PA1-Parada 6 / (M) Quinta Normal',
stop_lat='-33.4401169274166',
stop_lon='-33.4401169274166',
stop_url='http://www.metro.cl/estacion/vicuna-mackenna-l4')
self.session.add(self.stop)
self.route = Route(route_id='106',
agency_id='TS',
route_short_name='106',
route_long_name='Nueva San Martin - La Florida',
route_type='3',
route_color='00D5FF',
route_text_color='000000')
self.session.add(self.route)
self.route_stop = RouteStop(
route_id=self.route.route_id,
direction_id=0,
stop_id=self.stop.stop_id,
order=1,
start_date=arrow.now().date(),
end_date=arrow.now().date(),
)
self.session.add(self.route_stop)
self.trip = Trip(trip_id='106-I-L_V34-B02',
route_id=self.route.route_id,
service_id='L_V34',
shape_id=None,
trip_headsign='La Florida',
direction_id='0')
self.session.add(self.trip)
self.stop_time = StopTime(trip_id='106-I-L_V34-B02',
arrival_time=None,
departure_time=None,
stop_id=self.stop.stop_id,
stop_sequence=1)
self.session.add(self.stop_time)
self.session.commit()