-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.py
86 lines (76 loc) · 3.09 KB
/
main.py
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
from flask import Flask, redirect, render_template, url_for
from flask import request
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SECRET_KEY'] = "2pZt'WFQ45~,"
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///hotel.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)
class Hotel(db.Model):
__tablename__ = "hotel"
id = db.Column(db.Integer, primary_key=True)
check_out = db.Column(db.String(20), nullable=False)
check_in = db.Column(db.String(20), nullable=False)
adult = db.Column(db.String(10), nullable=False)
child = db.Column(db.String(10), nullable=False)
class Booking(db.Model):
__tablename__ = "booking"
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(25), nullable=False)
email = db.Column(db.String(50), nullable=False)
check_out = db.Column(db.String(20), nullable=False)
check_in = db.Column(db.String(20), nullable=False)
adult = db.Column(db.String(10), nullable=False)
child = db.Column(db.String(10), nullable=False)
room = db.Column(db.String(10), nullable=False)
special_request = db.Column(db.String(250))
# with app.app_context():
# db.create_all()
@app.route('/', methods=["GET", "POST"])
def home():
if request.method == "POST":
with app.app_context():
new_hotel = Hotel(
check_out = request.form['check_out'],
check_in = request.form['check_in'],
adult = request.form['adult'],
child = request.form['child']
)
db.session.add(new_hotel)
db.session.commit()
return redirect(url_for('home'))
return render_template('home.html')
@app.route('/rooms', methods=["GET", "POST"])
def rooms():
if request.method == "POST":
with app.app_context():
new_hotel = Hotel(
check_out = request.form['check_out'],
check_in = request.form['check_in'],
adult = request.form['adult'],
child = request.form['child']
)
db.session.add(new_hotel)
db.session.commit()
return redirect(url_for('home'))
return render_template('rooms.html')
@app.route('/booking', methods=["GET", "POST"])
def booking():
if request.method == "POST":
with app.app_context():
new_booking = Booking(
name = request.form['name'],
email = request.form['email'],
check_out = request.form['check_out'],
check_in = request.form['check_in'],
adult = request.form['adult'],
child = request.form['child'],
room = request.form['room'],
special_request = request.form['special_request']
)
db.session.add(new_booking)
db.session.commit()
return redirect(url_for('home'))
return render_template('booking.html')
if __name__ == "__main__":
app.run(port=5000, debug=True)