def link_itinerary():
"""Links a user to an existing itienrary and returns data as JSON"""
user = helper.get_user_by_email(session['EMAIL'])
itinerary_id = request.form['id']
itinerary = helper.get_itinerary_by_id(itinerary_id)
crud.create_user_itinerary(user.user_id, itinerary_id)
return jsonify({'itinerary_id': itinerary_id, 'trip_name': itinerary.trip_name})
def new_itinerary():
"""Creates a new itinerary for a user and returns data as JSON."""
user = helper.get_user_by_email(session['EMAIL'])
trip_name = request.form['trip_name']
start_date = request.form['start_date']
end_date = request.form['end_date']
num_days = crud.calculate_itinerary_days(start_date, end_date)
lat, lng = crud.get_latitude_longitude_for_itinerary(trip_name)
new_itinerary = crud.create_itinerary(trip_name, start_date, end_date, num_days, lat, lng)
crud.create_user_itinerary(user.user_id, new_itinerary.itinerary_id)
return jsonify({'itinerary_id': new_itinerary.itinerary_id, 'trip_name': new_itinerary.trip_name})
def new_itinerary():
"""Creates a new itinerary for a user."""
email = session["USERNAME"]
user = helper.get_user_by_email(email)
trip_name = request.form.get('trip_name')
start_date = request.form.get('start_date')
end_date = request.form.get('end_date')
num_days = crud.calculate_itinerary_days(start_date, end_date)
lat_lng = crud.get_latitude_longitude_for_itinerary(trip_name)
lat = lat_lng[0]
lng = lat_lng[1]
new_itinerary = crud.create_itinerary(trip_name, start_date, end_date,
num_days, lat, lng)
crud.create_user_itinerary(user.user_id, new_itinerary.itinerary_id)
json_info = {
'itinerary_id': new_itinerary.itinerary_id,
'trip_name': new_itinerary.trip_name
}
return jsonify(json_info)
crud.create_user('*****@*****.**', 'test', 'Grace', 'Graceful')
crud.create_user('*****@*****.**', 'test', 'Hildy', 'Hinter')
crud.create_user('*****@*****.**', 'test', 'Jamie', 'Jameson')
crud.create_user('*****@*****.**', 'test', 'Kat', 'King')
#create itineraries
crud.create_itinerary('London, UK', '2021-01-01', '2021-01-08', 15, 51.4141076, -1.4000882)
crud.create_itinerary('Bali, Indonesia', '2021-05-05', '2021-05-15', 11, -8.4556973, 114.510954)
crud.create_itinerary('San Francisco, CA', '2020-12-01', '2020-12-05', 5, 37.7576793, -122.5076413)
crud.create_itinerary('Chicago, IL', '2020-11-25', '2020-11-28', 4, 41.8333925, -88.0121674)
crud.create_itinerary('Paris, France', '2021-03-03', '2021-03-09', 7, 48.8588377, 2.2770192)
#associate users with itineraries
crud.create_user_itinerary(1, 1)
crud.create_user_itinerary(2, 1)
crud.create_user_itinerary(3, 1)
crud.create_user_itinerary(4, 1)
crud.create_user_itinerary(10, 2)
crud.create_user_itinerary(9, 2)
crud.create_user_itinerary(8, 2)
crud.create_user_itinerary(1, 3)
crud.create_user_itinerary(4, 3)
crud.create_user_itinerary(7, 3)
crud.create_user_itinerary(7, 4)
crud.create_user_itinerary(6, 4)
crud.create_user_itinerary(5, 4)
def test_create_user_itinerary(self):
"""Tests creation of a user and itinerary association in database."""
ui = crud.create_user_itinerary(2, 1)
self.assertIsNotNone(ui.ui_id)