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()