-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
88 lines (72 loc) · 2.85 KB
/
app.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
87
from flask import Flask, redirect, render_template, request, url_for
from flask_login import (LoginManager, UserMixin, current_user, login_required,login_user, logout_user)
from flask_principal import Permission, Principal, RoleNeed
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:https:///bookings.db'
app.config['SECRET_KEY'] = 'secret_key'
db = SQLAlchemy(app)
login_manager = LoginManager(app)
principal = Principal(app)
class User(UserMixin, db.Model):
id = db.Column(db.Integer, primary_key = True)
username = db.Column(db.String(50), unique = True, nullable = False )
password = db.Column(db.String(50), nullable = False)
@login_manager.user_loader
def load_user(user_id):
return User.query.get(int(user_id))
@app.route('/login', methods = ['GET', 'POST'])
def login():
if request.method == 'POST':
username = request.form.get('username')
password = request.form.get('password')
user = User.query.filter_by(username=username, password=password).first()
if user:
login_user(user)
return redirect(url_for('index'))
else:
return 'Invalid username or password! Please check your credentials'
return render_template('login.html')
@app.route('/logout')
def logout():
logout_user()
return redirect(url_for('index'))
@app.route('/register', methods = ['GET', 'POST'])
def register():
if request.method == 'POST':
username = request.form.get('username')
password = request.form.get('password')
existing_user = User.query.filter_by(username=username).first()
if existing_user:
return 'Username already exists! Try with a different username'
new_user = User(username=username, password=password)
db.session.add(new_user)
db.session.commit()
return redirect(url_for('login'))
return render_template('register.html')
class Booking(db.Model):
id = db.Column(db.Integer, primary_key = True)
name = db.Column(db.String(50), nullable = False)
service = db.Column(db.String(50), nullable = False)
@app.route('/')
def index():
return render_template('index.html')
@app.route('/booking')
def booking():
return render_template('booking.html')
@app.route('/book', methods =['POST'])
def book():
name = request.form.get('name')
service = request.form.get('service')
booking = Booking (name=name, service=service)
db.session.add(booking)
db.session.commit()
return redirect(url_for('bookings'))
@app.route('/bookings') # to show all bookings details
def bookings():
all_bookings = Booking.query.all()
return render_template('bookings.html', bookings=all_bookings)
# if __name__ == '__main__':
# with app.app_context():
# db.create_all() # create database tables before running the app
# app.run(debug=True)