diff --git a/app.py b/app.py index 73041ca..8a78623 100644 --- a/app.py +++ b/app.py @@ -68,7 +68,7 @@ def buy(): return render_template("buy.html") else: symbol = request.form.get("symbol") - shares = float(request.form.get("shares")) + shares = request.form.get("shares") if not symbol: return apology("Not Symbol") @@ -78,7 +78,7 @@ def buy(): if stock == None: return apology("Symbol not found") - if not shares == "": + if shares: transaction_value = float(shares) * stock["price"] user_id = session["user_id"] @@ -95,9 +95,11 @@ def buy(): date = datetime.datetime.now() 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!") + else: + return apology("Missing shares", 400) return redirect("/") # return apology("TODO") @@ -208,7 +210,6 @@ def register(): username_exists = db.execute("SELECT * FROM users WHERE username = ?", username) if username_exists: - flash("Username exists already!") return apology("username already exists!", 400) db.execute("INSERT INTO users(username, hash) VALUES(?, ?)", diff --git a/finance.db b/finance.db index f4d2422..d338736 100644 Binary files a/finance.db and b/finance.db differ diff --git a/flask_session/16fa13c5890feb947b5d5c907cfedf5b b/flask_session/16fa13c5890feb947b5d5c907cfedf5b new file mode 100644 index 0000000..7faa085 Binary files /dev/null and b/flask_session/16fa13c5890feb947b5d5c907cfedf5b differ diff --git a/flask_session/2029240f6d1128be89ddc32729463129 b/flask_session/2029240f6d1128be89ddc32729463129 index 8b04914..7f5741f 100644 Binary files a/flask_session/2029240f6d1128be89ddc32729463129 and b/flask_session/2029240f6d1128be89ddc32729463129 differ diff --git a/templates/buy.html b/templates/buy.html index b34ebb5..2659c6e 100644 --- a/templates/buy.html +++ b/templates/buy.html @@ -10,7 +10,7 @@
- +