def res():
form = ReservationForm(request.form)
if request.method == 'POST' and form.validate():
reservation = Reservation()
save_res(reservation, form, new=True)
flash('Reservation created successfully!')
return redirect('/res')
qry = db_session.query(Reservation)
reservations = qry.all()
reservationsTable = ReservationsTable(reversed(reservations))
reservationsTable.border = True
return render_template('res.html',
form=form,
reservations=reservations,
reservationsTable=reservationsTable)
def make_reservation(train_id=None):
form = ReservationForm()
date = request.args.get('date', type=str)
train_id = request.args.get('train_id', type=int)
departure_station = request.args.get('departure_station', type=str)
departure_time = request.args.get('departure_time', type=str)
arrival_station = request.args.get('arrival_station', type=str)
arrival_time = request.args.get('arrival_time', type=str)
seats_available = request.args.get('seats_available', type=str)
price = request.args.get('price', type=str)
num_adult = request.args.get('num_adult', type=int)
num_child = request.args.get('num_child', type=int)
num_senior = request.args.get('num_senior', type=int)
num_military = request.args.get('num_military', type=int)
num_pets = request.args.get('num_pets', type=int)
if request.method == 'GET':
return render_template('makereservation.html',
form=form,
date=date,
train_id=train_id,
departure_station=departure_station,
departure_time=departure_time,
arrival_station=arrival_station,
arrival_time=arrival_time,
seats_available=seats_available,
price=price,
passengers=[
num_adult, num_child, num_senior,
num_military, num_pets
])
elif request.method == 'POST':
if form.validate():
booker = {
'first_name': form.first_name.data,
'last_name': form.last_name.data,
'email': form.email.data
}
passengers = [
int(form.num_adult.data),
int(form.num_child.data),
int(form.num_senior.data),
int(form.num_military.data),
int(form.num_pets.data)
]
print('PASSENGERS FROM FORM: {}'.format(passengers))
create_reservation_and_trips(train_id, departure_station,
departure_time, arrival_station,
passengers, booker, date)
message = 'The reservation has been made successfully!'
return render_template('success.html', message=message)
else:
return render_template('makereservation.html',
form=form,
date=date,
train_id=train_id,
departure_station=departure_station,
departure_time=departure_time,
arrival_station=arrival_station,
arrival_time=arrival_time,
seats_available=seats_available,
price=price,
passengers=[
num_adult, num_child, num_senior,
num_military, num_pets
])
