forked from hanyuen/train-booking
/
app.py
76 lines (58 loc) · 2.01 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
__author__ = 'hanyuen'
from flask import Flask, render_template,\
request,redirect, url_for, session, flash
from functools import wraps
from flask_admin import Admin
from flask_admin.contrib.sqla import ModelView
from flask_restless import APIManager
from database import db
from models import Train
app = Flask(__name__)
app.config.from_object("config.DevelopmentConfig")
db.init_app(app) #this changes the app
with app.app_context():
api_manager = APIManager(app, flask_sqlalchemy_db=db)
# api_manager.init_app(app) #this cannot work
#api_manager.create_api(Train, methods=["GET", 'POST', 'DELETE'])
api_manager.create_api(Train, include_columns =["id", "fromCity", "toCity"], methods=["GET", 'POST', 'PUT', 'DELETE'])
#def get_trains():
def create_admin():
admin = Admin(app, name="TrainsBooking")
admin.add_view(ModelView(Train, db.session))
def login_required(f):
@wraps(f)
def wrap(*args, **kwargs):
if "logged_in" in session:
return f(*args, **kwargs)
else:
flash("you need to log in first")
return redirect((url_for("login")))
return wrap
@app.route("/")
@login_required
def home():
trains = db.session.query(Train).all()
return render_template("index.html", trains = trains)
@app.route("/welcome")
def welcome():
return render_template("welcome.html")
@app.route("/login", methods=['GET', 'POST'])
def login():
error = None
if request.method == 'POST':
if (request.form["user"] != 'admin' or request.form["password"] != 'admin'):
error = "Invalid credentials. Please try again"
else:
session["logged_in"] = True
flash("you were just logged in")
return redirect(url_for('home'))
return render_template("login.html", error = error)
@app.route("/logout")
@login_required
def logout():
session.pop('logged_in', None)
flash("you were just logged out")
return redirect (url_for("welcome"))
if __name__ == '__main__':
create_admin()
app.run()