-
Notifications
You must be signed in to change notification settings - Fork 0
/
application.py
158 lines (122 loc) · 4.64 KB
/
application.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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
import os, re
from cs50 import SQL
from flask import Flask, flash, jsonify, redirect, render_template, request, session, url_for
from flask_session import Session
from tempfile import mkdtemp
from werkzeug.exceptions import default_exceptions, HTTPException, InternalServerError
from werkzeug.security import check_password_hash, generate_password_hash
from helpers import apology, login_required
# Configure application
app = Flask(__name__)
# Ensure templates are auto-reloaded
app.config["TEMPLATES_AUTO_RELOAD"] = True
# Ensure responses aren't cached
@app.after_request
def after_request(response):
response.headers["Cache-Control"] = "no-cache, no-store, must-revalidate"
response.headers["Expires"] = 0
response.headers["Pragma"] = "no-cache"
return response
# Configure session to use filesystem (instead of signed cookies)
app.config["SESSION_FILE_DIR"] = mkdtemp()
app.config["SESSION_PERMANENT"] = False
app.config["SESSION_TYPE"] = "filesystem"
Session(app)
# Configure CS50 Library to use SQLite database
db = SQL("sqlite:///users.db")
@app.before_request
def make_session_permanent():
session.permanent = True
@app.route("/")
def index():
return render_template("index.html")
@app.route("/buy", methods=["GET", "POST"])
@login_required
def buy():
"""Buy shares of stock"""
return apology("TODO")
@app.route("/homepage")
@login_required
def homepage():
return render_template("homepage.html")
@app.route("/login", methods=["GET", "POST"])
def login():
"""Log user in"""
# User reached route via POST (as by submitting a form via POST)
if request.method == "POST":
# Ensure username was submitted
if not request.form.get("username"):
flash("Please enter a username.")
return redirect("/login")
# Ensure password was submitted
elif not request.form.get("password"):
flash("Please enter a password.")
return redirect("/login")
# Query database for username
rows = db.execute("SELECT * FROM users WHERE username = :username",
username=request.form.get("username"))
# Ensure username exists and password is correct
if len(rows) != 1 or not check_password_hash(rows[0]["hash"], request.form.get("password")):
flash("Invalid username and/or password")
return redirect("/login")
# Remember which user has logged in
session["user_id"] = rows[0]["id"]
# Redirect user to home page
flash("You have successfully logged in.")
return redirect("/homepage")
# User reached route via GET (as by clicking a link or via redirect)
else:
return render_template("login.html")
@app.route("/logout")
def logout():
"""Log user out"""
# Forget any user_id
session.clear()
# Redirect user to homepage
flash("You have successfully logged out.")
return redirect("/")
@app.route("/quote", methods=["GET", "POST"])
@login_required
def quote():
"""Get stock quote."""
return apology("TODO")
@app.route("/register", methods=["GET", "POST"])
def register():
"""Register user"""
if request.method == "GET":
return render_template("register.html")
else:
username = request.form.get("username")
if not username:
flash("Please enter a username.")
return redirect("/register")
password = request.form.get("password")
confirmation = request.form.get("confirmation")
if not password:
flash("Please enter a password.")
return redirect("/register")
if len(password) < 8 or (not re.search("\W|_", password)) or (not re.search("[A-Z]", password)):
flash("Password does not meet requirements.")
return redirect("/register")
if password != confirmation:
flash("Passwords do not match.")
return redirect("/register")
username_exist = db.execute("SELECT COUNT(*) FROM users WHERE username = ?", username)
if username_exist[0]["COUNT(*)"] != 0:
flash("Username already taken.")
return redirect("/register")
db.execute("INSERT INTO users (username, hash) VALUES (?, ?)", username, generate_password_hash(password))
return redirect("/")
@app.route("/sell", methods=["GET", "POST"])
@login_required
def sell():
"""Sell shares of stock"""
return apology("TODO")
def errorhandler(e):
"""Handle error"""
if not isinstance(e, HTTPException):
e = InternalServerError()
return apology("Internal Server Error")
# Listen for errors
for code in default_exceptions:
app.errorhandler(code)(errorhandler)