-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathapp.py
More file actions
92 lines (77 loc) · 3.32 KB
/
Copy pathapp.py
File metadata and controls
92 lines (77 loc) · 3.32 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
from flask import Flask, render_template, request, redirect, url_for
import pymysql
import config
app = Flask(__name__)
# MySQL connection function
def get_db_connection():
connection = pymysql.connect(
host=config.Config.MYSQL_HOST,
user=config.Config.MYSQL_USER,
password=config.Config.MYSQL_PASSWORD,
database=config.Config.MYSQL_DB,
cursorclass=pymysql.cursors.DictCursor
)
return connection
# Step 1: Select source and destination
@app.route('/')
def index():
connection = get_db_connection()
with connection.cursor() as cur:
# Fetch unique sources and destinations from the database
cur.execute("SELECT DISTINCT source FROM ev_buses")
sources = cur.fetchall()
cur.execute("SELECT DISTINCT destination FROM ev_buses")
destinations = cur.fetchall()
connection.close()
return render_template('index.html', sources=sources, destinations=destinations)
# Step 2: Show available buses based on source, destination, and date
@app.route('/search_buses', methods=['POST'])
def search_buses():
source = request.form['source']
destination = request.form['destination']
connection = get_db_connection()
with connection.cursor() as cur:
# Query to get available buses between selected source and destination
query = '''
SELECT id, bus_name, available_seats FROM ev_buses
WHERE source = %s AND destination = %s
'''
cur.execute(query, (source, destination))
buses = cur.fetchall()
connection.close()
# Render available buses with their names and seat availability
return render_template('available_buses.html', buses=buses, source=source, destination=destination)
# Step 3: Book a seat on a selected bus
@app.route('/book', methods=['POST'])
def book():
bus_id = request.form['bus_id']
first_name = request.form['first_name']
last_name = request.form['last_name']
mobile_number = request.form['mobile_number']
seats_booked = int(request.form['seats_booked'])
connection = get_db_connection()
with connection.cursor() as cur:
# Check if enough seats are available
cur.execute("SELECT available_seats FROM ev_buses WHERE id = %s", (bus_id,))
available_seats = cur.fetchone()['available_seats']
if available_seats >= seats_booked:
# Insert booking into the bookings table
cur.execute(
"INSERT INTO bookings (bus_id, first_name, last_name, mobile_number, seats_booked, booking_date) "
"VALUES (%s, %s, %s, %s, %s, CURDATE())",
(bus_id, first_name, last_name, mobile_number, seats_booked)
)
# Update the available seats in the ev_buses table
cur.execute(
"UPDATE ev_buses SET available_seats = available_seats - %s WHERE id = %s",
(seats_booked, bus_id)
)
connection.commit()
connection.close()
# Pass the first name, bus ID, and seats booked to the success page
return render_template('success.html', first_name=first_name, bus_id=bus_id, seats_booked=seats_booked)
else:
connection.close()
return "Not enough seats available", 400
if __name__ == '__main__':
app.run(debug=True, host='0.0.0.0', port=80)