-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
250 lines (187 loc) · 8.04 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
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
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
import os
from cs50 import SQL
from flask import Flask, flash, redirect, render_template, request, session
from flask_session import Session
from tempfile import mkdtemp
from werkzeug.security import check_password_hash, generate_password_hash
from helpers import apology, login_required, lookup, usd
# Configure application
app = Flask(__name__)
# Ensure templates are auto-reloaded
app.config["TEMPLATES_AUTO_RELOAD"] = True
# Custom filter
app.jinja_env.filters["usd"] = usd
# Configure session to use filesystem (instead of signed cookies)
app.config["SESSION_PERMANENT"] = False
app.config["SESSION_TYPE"] = "filesystem"
Session(app)
# Configure CS50 Library to use SQLite database
db = SQL("sqlite:///finance.db")
# Make sure API key is set
if not os.environ.get("API_KEY"):
raise RuntimeError("API_KEY not set")
@app.after_request
def after_request(response):
"""Ensure responses aren't cached"""
response.headers["Cache-Control"] = "no-cache, no-store, must-revalidate"
response.headers["Expires"] = 0
response.headers["Pragma"] = "no-cache"
return response
@app.route("/")
@login_required
def index():
"""Show portfolio of stocks"""
user_id = session["user_id"]
stocks = db.execute(
"SELECT symbol, name, price, SUM(shares) as totalShares FROM transactions WHERE user_id = ? GROUP BY symbol", user_id)
cash = db.execute("SELECT cash FROM users WHERE id = ?", user_id)[0]["cash"]
total = cash
for stock in stocks:
total += stock["price"] * stock["totalShares"]
return render_template("index.html", stocks=stocks, cash=cash, usd=usd, total=total)
@app.route("/buy", methods=["GET", "POST"])
@login_required
def buy():
"""Buy shares of stock"""
# Submit the user’s input via POST to /buy.
if (request.method == "POST"):
# Require that a user input a stock’s symbol,Render an apology if the input is blank or the symbol does not exist (as per the return value of lookup).
symbol = request.form.get("symbol").upper()
item = lookup(symbol)
if not symbol:
return apology("Stock name is required")
if not item:
return apology("Invalid symbol")
try:
shares = int(request.form.get("shares"))
except:
return apology("Share needs to be a number")
if (shares <= 0):
return apology("Number needs to be positive")
user_id = session["user_id"]
cash = db.execute("SELECT cash FROM users WHERE id = ?", user_id)[0]["cash"]
print(f"{{cash}}")
item_name = item["name"]
item_price = item["price"]
total_price = item_price * shares
if cash < total_price:
return apology("Insufficent money")
else:
db.execute("UPDATE users SET cash = ? WHERE id = ?", cash - total_price, user_id)
db.execute("INSERT INTO transactions (user_id, name, shares, price, type, symbol) VALUES(?, ?, ?, ?, ?, ?)",
user_id, item_name, shares, item_price, 'buy', symbol)
return redirect("/")
else:
return render_template("buy.html")
@app.route("/history")
@login_required
def history():
"""Show history of transactions"""
user_id = session["user_id"]
transactions = db.execute("SELECT type, symbol, price, shares, time FROM transactions WHERE user_id = ?", user_id)
return render_template("history.html", transactions=transactions, usd=usd)
@app.route("/login", methods=["GET", "POST"])
def login():
"""Log user in"""
# Forget any user_id
session.clear()
# 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"):
return apology("must provide username", 403)
# Ensure password was submitted
elif not request.form.get("password"):
return apology("must provide password", 403)
# Query database for username
rows = db.execute("SELECT * FROM users WHERE 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")):
return apology("invalid username and/or password", 403)
# Remember which user has logged in
session["user_id"] = rows[0]["id"]
# Redirect user to home page
return redirect("/")
# 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 login form
return redirect("/")
@app.route("/quote", methods=["GET", "POST"])
@login_required
def quote():
"""Get stock quote."""
# Submit the user’s input via POST to /quote.
if (request.method == "POST"):
# Require that a user input a stock’s symbol, implemented as a text field whose name is symbol.
symbol = request.form.get("symbol").upper()
if not symbol:
return apology("The stock name is required")
item = lookup(symbol)
if not item:
return apology("Invalid symbol")
return render_template("quoted.html", item=item, usd_function=usd)
else:
return render_template("quote.html")
@app.route("/register", methods=["GET", "POST"])
def register():
"""Register user"""
# submit user info via POST
if (request.method == "POST"):
username = request.form.get("username")
password = request.form.get("password")
confirmation = request.form.get("confirmation")
specialChars = set("$#!&?*^~")
# check if name is blank or alredy exists
if not username:
return apology("Username is required")
# check if password is the same and not blank
elif not password:
return apology("Password is required")
elif not confirmation:
return apology("Password confirmation is required")
if password != confirmation:
return apology("The passwords need to match")
if not specialChars.intersection(password):
return apology("The password needs to have at least one special character")
# Insert user into db and hash the password and keep the hash in the database
hash = generate_password_hash(password)
try:
db.execute("INSERT INTO users (username, hash) VALUES (?, ?)", username, hash)
return redirect("/")
except:
return apology("Username already exists")
else:
return render_template("register.html")
@app.route("/sell", methods=["GET", "POST"])
@login_required
def sell():
"""Sell shares of stock"""
# Submit the user’s input via POST to /sell.
if (request.method == "POST"):
user_id = session["user_id"]
symbol = request.form.get("symbol").upper()
shares = int(request.form.get("shares"))
if shares <= 0:
return apology("Shares needs to be a positive integer")
item_price = lookup(symbol)["price"]
item_name = lookup(symbol)["name"]
price = shares * item_price
shares_owned = db.execute("SELECT shares FROM transactions WHERE user_id = ? AND symbol = ? GROUP BY symbol", user_id, symbol)[
0]["shares"]
if shares_owned < shares:
return apology("Not enough shares")
current_cash = db.execute("SELECT cash FROM users WHERE id= ?", user_id)[0]["cash"]
db.execute("UPDATE users SET cash = ? WHERE id = ?", current_cash + price, user_id)
db.execute("INSERT INTO transactions (user_id, name, shares, price, type, symbol) VALUES (?,?,?,?,?,?)",
user_id, item_name, -shares, item_price, "sell", symbol)
return redirect("/")
else:
user_id = session["user_id"]
symbols = db.execute("SELECT symbol FROM transactions WHERE user_id = ? GROUP BY symbol", user_id)
return render_template("sell.html", symbols=symbols)