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