mirror of https://github.com/me50/kukemuna.git
automated commit by check50 [check50=True]
This commit is contained in:
parent
af74bf282d
commit
6e3bc7d7aa
33
app.py
33
app.py
|
|
@ -68,7 +68,7 @@ def buy():
|
||||||
return render_template("buy.html")
|
return render_template("buy.html")
|
||||||
else:
|
else:
|
||||||
symbol = request.form.get("symbol")
|
symbol = request.form.get("symbol")
|
||||||
shares = int(request.form.get("shares"))
|
shares = request.form.get("shares")
|
||||||
|
|
||||||
if not symbol:
|
if not symbol:
|
||||||
return apology("Not Symbol")
|
return apology("Not Symbol")
|
||||||
|
|
@ -78,25 +78,26 @@ def buy():
|
||||||
if stock == None:
|
if stock == None:
|
||||||
return apology("Symbol not found")
|
return apology("Symbol not found")
|
||||||
|
|
||||||
transaction_value = shares * stock["price"]
|
if not shares == "":
|
||||||
|
transaction_value = int(shares) * stock["price"]
|
||||||
|
|
||||||
user_id = session["user_id"]
|
user_id = session["user_id"]
|
||||||
user_cash_db = db.execute("SELECT cash FROM users WHERE id = ?", user_id)
|
user_cash_db = db.execute("SELECT cash FROM users WHERE id = ?", user_id)
|
||||||
user_cash = user_cash_db[0]["cash"]
|
user_cash = user_cash_db[0]["cash"]
|
||||||
|
|
||||||
if user_cash < transaction_value:
|
if user_cash < transaction_value:
|
||||||
return apology("U broke, m8!")
|
return apology("U broke, m8!")
|
||||||
|
|
||||||
free_cash = user_cash - transaction_value
|
free_cash = user_cash - transaction_value
|
||||||
|
|
||||||
db.execute("UPDATE users SET cash = ? WHERE id = ?", free_cash, user_id)
|
db.execute("UPDATE users SET cash = ? WHERE id = ?", free_cash, user_id)
|
||||||
|
|
||||||
date = datetime.datetime.now()
|
date = datetime.datetime.now()
|
||||||
|
|
||||||
db.execute("INSERT INTO transactions (user_id, symbol, shares, price, date) VALUES (?, ?, ?, ?, ?)",
|
db.execute("INSERT INTO transactions (user_id, symbol, shares, price, date) VALUES (?, ?, ?, ?, ?)",
|
||||||
user_id, stock["symbol"], shares, stock["price"], date)
|
user_id, stock["symbol"], shares, stock["price"], date)
|
||||||
|
|
||||||
flash("Bought!")
|
flash("Bought!")
|
||||||
|
|
||||||
return redirect("/")
|
return redirect("/")
|
||||||
# return apology("TODO")
|
# return apology("TODO")
|
||||||
|
|
@ -191,15 +192,15 @@ def register():
|
||||||
if request.method == "POST":
|
if request.method == "POST":
|
||||||
# Ensure username was submitted
|
# Ensure username was submitted
|
||||||
if not request.form.get("username"):
|
if not request.form.get("username"):
|
||||||
return apology("must provide username", 403)
|
return apology("must provide username", 400)
|
||||||
|
|
||||||
# Ensure password was submitted
|
# Ensure password was submitted
|
||||||
elif not request.form.get("password"):
|
elif not request.form.get("password"):
|
||||||
return apology("must provide password", 403)
|
return apology("must provide password", 400)
|
||||||
|
|
||||||
# Ensure password repeat matches
|
# Ensure password repeat matches
|
||||||
if not request.form.get("password") == request.form.get("confirmation"):
|
if not request.form.get("password") == request.form.get("confirmation"):
|
||||||
return apology("passwords don't match", 403)
|
return apology("passwords don't match", 400)
|
||||||
|
|
||||||
username = request.form.get("username")
|
username = request.form.get("username")
|
||||||
password = request.form.get("password")
|
password = request.form.get("password")
|
||||||
|
|
|
||||||
BIN
finance.db
BIN
finance.db
Binary file not shown.
Binary file not shown.
Loading…
Reference in New Issue