def saveAsset(self, emailAddress):
existing_assets = self.findAssets(emailAddress)
if not existing_assets:
asset = {
"userID": User.findUser(emailAddress)["_id"],
"symbol": self.symbol,
"amount": self.amount,
"avarage_price": self.avarage_price
}
else:
update_amount = int(existing_assets['amount']) + int(self.amount)
find_col = {
"name": "userID",
"value": User.findUser(emailAddress)["_id"]
}
set_col = {"name": "amount", "value": update_amount}
return print(Database_manager().update_one(find_col, set_col,
self.collection))
Database_manager().save_one(asset, self.collection)
def login():
if ("username" in session):
flash("You already logged in", "error")
return render_template("welcomescreen.html", user=session['username'])
if request.method == "POST":
existing_user = User.findUser(request.form.get('email').lower())
if(existing_user):
if(PasswordHelper.checkPassword(request.form.get('password'),
existing_user['password'])):
user = User(existing_user['firstname'],
existing_user['emailaddress'],
existing_user['password'])
flash("Welcome back {}".format(user.firstName), "success")
session['username'] = user.firstName
session['emailaddress'] = user.emailAdress
return render_template('assets.html',
user=session['username'])
else:
flash("Username and/or password are incorrect", "error")
else:
flash("Username does not exist", "error")
return render_template('login.html')
def findAssets(emailAddress):
user = User.findUser(emailAddress)
asset = {"userID": user['_id']}
collection = "assets"
return Database_manager().find_one(asset, collection)
