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)