def test_gets_user_stop_id(self): user_lat = 37.785152 user_lon = -122.406581 route_code = '153' route = gets_route_id_db(route_code) self.assertEqual(gets_user_stop_id(user_lat, user_lon, route), 14) print "complete gets_user_stop_id"
def gets_user_itinerary(agency, route_code, destination_stop, user_inital_stop, route_name): """returns a list of the user's stops from inital to destination >>> agency = u'BART' >>> route_code = u'920' >>> destination_stop = u'32' >>> user_inital_stop = u'14' >>> route_name = u'Daly City - Dublin/Pleasanton' >>> user_itinerary = gets_user_itinerary(agency, route_code, destination_stop, user_inital_stop, route_name) >>> user_itinerary '14, 15, 17, 19, 38, 36, 34, 32' """ route = gets_route_id_db(route_code) if not route: route = gets_route_db(route_code) destination_stop = gets_stop_db(destination_stop) user_inital_stop = gets_stop_db(user_inital_stop) route_stops = gets_stops_from_route(str(route.stop_list), route.agency.agency_id) itinerary = parse_route_stop_for_user(route_stops, user_inital_stop, destination_stop) return itinerary
def setUp(self): """Creating mock data to test aganist""" user_db = User.query.filter_by(user_name="Jessica").first() user_inital_stop = '14' user_inital_stop = gets_stop_db(user_inital_stop) destination_stop = '32' destination_stop = gets_stop_db(destination_stop) route_code = '153' route = gets_route_id_db(route_code) agency = 'BART' arrival_time_datetime = datetime.datetime(2016, 7, 31, 4, 39, 53, 25840) adds_transit_request(user_inital_stop.stop_code, destination_stop.stop_code, agency, route.name, route.route_code, arrival_time_datetime, user_db)
def stops(): """returns routes's stops""" route_id = request.args.get("route_id") route_db = gets_route_id_db(route_id) stops = { stop.name: { "stop_code": stop.stop_code, "name": stop.name, "lat": stop.lat, "lon": stop.lon, } for stop in route_db.stops} return jsonify(stops)
def process_user_info(): """recieves the user data and processes data""" user_name = request.form.get("name") raw_user_phone_num = request.form.get("phone") agency = request.form.get("agency") route_code = request.form.get("route") user_inital_stop = request.form.get("user_stop") destination_stop = request.form.get("destination_stop") user_lat = request.form.get("lat") user_lon = request.form.get("lng") route = gets_route_id_db(route_code) if user_lat: user_inital_stop = gets_user_stop_id(user_lat, user_lon, route) user_inital_stop = gets_stop_db(user_inital_stop) if not user_lat: user_lat = user_inital_stop.lat user_lon = user_inital_stop.lon destination_stop = gets_stop_db(destination_stop) arrival_time_datetime = process_lat_lng_get_arrival_datetime(user_lat, user_lon, destination_stop) user_phone = convert_to_e164(raw_user_phone_num) user_db = checks_user_db(user_name, user_phone) adds_transit_request(user_inital_stop.stop_code, destination_stop.stop_code, agency, route.name, route.route_code, arrival_time_datetime, user_db) return render_template("/thank_you.html", user_fname=user_name, user_phone=user_phone, route=route, user_inital_stop=user_inital_stop, destination_stop=destination_stop)