def login():
if request.method == "POST":
# check if username is in database
existing_user = mongo.db.users.find_one(
{"username": request.form.get("username").lower()})
if existing_user:
# confirm password
if check_password_hash(existing_user["password"],
request.form.get("password")):
session["user"] = request.form.get("username").lower()
first_name = mongo.db.users.find_one(
{"username": session["user"]})["first_name"]
return render_template(
"profile.html",
username=session["user"],
first_name=first_name,
transactions=transactions(),
transaction_lst=transaction_lst(session),
stock_aapl=stock_aapl,
yesterday=yesterday,
funds_available=stringify_number(wallet()),
profit_loss=stringify_number(profit_loss(session)))
else:
# if the password doesn't match
flash("Incorrect Username and/or Password")
return redirect("login")
else:
# username doesn't match
flash("Incorrect Username and/or Password")
return redirect(url_for("login"))
return render_template("login.html")
def purchaseStocks():
return render_template("purchase-stocks.html",
username=getUsername(session),
stock_aapl=stock_aapl,
yesterday=yesterday,
funds_available=stringify_number(wallet()),
profit_loss=stringify_number(profit_loss(session)))
def openPositions():
return render_template("open-positions.html",
username=getUsername(session),
transactions=transactions(),
transaction_lst=transaction_lst(session),
stock_aapl=stock_aapl,
yesterday=yesterday,
funds_available=stringify_number(wallet()),
profit_loss=stringify_number(profit_loss(session)))
def closedPositions():
# Retrieve all data from mongoDB's "closed transactions" entries
closed_positions = mongo.db.closed_positions.find()
return render_template("closed-positions.html",
username=getUsername(session),
transactions=transactions(),
transaction_lst=transaction_lst(session),
stock_aapl=stock_aapl,
yesterday=yesterday,
funds_available=stringify_number(wallet()),
closed_positions=closed_positions,
profit_loss=stringify_number(profit_loss(session)))
def profile(username):
first_name = mongo.db.users.find_one({"username":
session["user"]})["first_name"]
if session["user"]:
return render_template("profile.html",
username=getUsername(session),
first_name=first_name,
transactions=transactions,
transaction_lst=transaction_lst(session),
stock_aapl=stock_aapl,
yesterday=yesterday,
funds_available=stringify_number(wallet()),
profit_loss=stringify_number(
profit_loss(session)))
return render_template("login")
def register():
if request.method == "POST":
# check if username already exists
existing_user = mongo.db.users.find_one(
{"username": request.form.get("username").lower()})
if existing_user:
flash("Username already exists")
return redirect(url_for("register"))
# Form to be inserted into Mongo's database
# These are inputs in the register form
register = {
"first_name": request.form.get("first_name"),
"last_name": request.form.get("last_name"),
"username": request.form.get("username").lower(),
"password": generate_password_hash(request.form.get("password"))
}
# Inserts "register"
mongo.db.users.insert_one(register)
# Variable created to pass in render_template to
# display on profile page
session["user"] = request.form.get("username").lower()
# Variable created to pass in render_template
# to display on profile page
first_name = mongo.db.users.find_one({"username":
session["user"]})["first_name"]
return render_template("profile.html",
username=session["user"],
first_name=first_name,
transactions=transactions,
transaction_lst=transaction_lst(session),
stock_aapl=stock_aapl,
yesterday=yesterday,
funds_available=stringify_number(wallet()),
profit_loss=stringify_number(
profit_loss(session)))
return render_template("register.html")
def add_funds():
first_name = mongo.db.users.find_one({"username":
session["user"]})["first_name"]
if request.method == "POST":
# When the user click on the button in the profile page an additional
# 10k will be added to the available funds
wallet_transaction = {
"money_amount": float(10000),
"created_by": session["user"]
}
mongo.db.wallet_transactions.insert_one(wallet_transaction)
return render_template("profile.html",
username=getUsername(session),
first_name=first_name,
transactions=transactions,
transaction_lst=transaction_lst(session),
stock_aapl=stock_aapl,
yesterday=yesterday,
funds_available=stringify_number(wallet()),
profit_loss=stringify_number(
profit_loss(session)))
def sell(position_id):
if request.method == "POST":
# If the user tries to sell less stock that the amount
# owned the databse has to be updated
if int(request.form.get("stock_amount_sell")) < int(
request.form.get("stocks_owned")):
# This calculates the amount of stocks remaining to
# update Monog's database with "update"
remaining_stock = int(request.form.get("stocks_owned")) - int(
request.form.get("stock_amount_sell"))
update = {
"purchase_date": yesterday,
"ticker": request.form.get('ticker'),
"stock_amount": str(remaining_stock),
"purchase_price": request.form.get("purchase_price_sell"),
"money_amount": request.form.get("money_amount_sell"),
"created_by": session["user"]
}
mongo.db.transactions.update({"_id": ObjectId(position_id)},
update)
# Data inserted into another database in order to have an
# updated amount of the funds available to the user
wallet_transaction = {
"money_amount": float(request.form.get("money_amount_sell")),
"created_by": session["user"]
}
mongo.db.wallet_transactions.insert_one(wallet_transaction)
# The sold stocks are then store in closed_positions
# in order to have history
sell = {
"selling_date": yesterday,
"ticker": request.form.get('ticker'),
"stock_amount": request.form.get("stock_amount_sell"),
"purchase_price": request.form.get("purchase_price_sell"),
"selling_price": request.form.get("selling_price"),
"created_by": session["user"]
}
mongo.db.closed_positions.insert_one(sell)
return redirect(url_for('openPositions'))
# If the user wants to sell the same amount of stocks owned data
# has to be deleted from the database
elif int(request.form.get("stock_amount_sell")) == int(
request.form.get("stocks_owned")):
mongo.db.transactions.remove({"_id": ObjectId(position_id)})
# The sold stocks are then store in closed_positions
# in order to have history
sell = {
"selling_date": yesterday,
"ticker": request.form.get('ticker'),
"stock_amount": request.form.get("stock_amount_sell"),
"purchase_price": request.form.get("purchase_price_sell"),
"money_amount": request.form.get("money_amount_sell"),
"selling_price": request.form.get("selling_price"),
"stocks_sold": request.form.get("money_amount_sell"),
"created_by": session["user"]
}
mongo.db.closed_positions.insert_one(sell)
# Data inserted into another database in order to have an
# updated amount of the funds available to the user
wallet_transaction = {
"money_amount": float(request.form.get("money_amount_sell")),
"created_by": session["user"]
}
mongo.db.wallet_transactions.insert_one(wallet_transaction)
return redirect(url_for('openPositions'))
# If the user wants to seel more stocks that the amount owned
# nothing will happen and a flash message will be displayed
elif int(request.form.get("stock_amount_sell")) > int(
request.form.get("stocks_owned")):
flash("Insufficient stock owned")
open_position = mongo.db.transactions.find_one(
{"_id": ObjectId(position_id)})
return render_template("sell-stocks.html",
username=getUsername(session),
stock_aapl=stock_aapl,
yesterday=yesterday,
funds_available=stringify_number(wallet()),
open_position=open_position,
profit_loss=stringify_number(profit_loss(session)))