def verify_pw(username, password): # check if user exists if not user_exists(username): return False check_user = User.objects(username=username).first() if check_password_hash(check_user['password'], password): return True else: return False
def login(): if current_user.is_authenticated == True: return redirect(url_for('auth.dashboard')) form = LoginForm() if request.method == 'POST': if form.validate(): check_user = User.objects(username=form.username.data).first() if check_user: if check_password_hash(check_user['password'], form.password.data): login_user(check_user) return redirect(url_for('auth.dashboard')) return render_template('auth/login.html', form=form)
def register(): form = RegisterForm() if request.method == 'POST': if form.validate(): existing_user = User.objects(username=form.username.data).first() if existing_user is None: username = form.username.data debt = 0.0 own = 0.0 hashpass = generate_password_hash(form.password.data, method='sha256') user = User(username=username, password=hashpass, own=own, debt=debt) user.save() login_user(user) return redirect(url_for('auth.dashboard')) return render_template('auth/register.html', form=form)
def post(self): data = request.get_json() username = data['username'] password = data['password'] money = data['amount'] ret, error = verify_credentials(username, password) if error: return jsonify(ret) if money <= 0: return jsonify(generate_status(304, 'The money entered must be greater thar 0')) # users cash cash = User.objects(username=username).first() # add the remaining money to user update_account(username, cash['own'] + money) return jsonify(generate_status(200, "Amount added successfully to account"))
def load_user(user_id): return User.objects(emailId=user_id).first()
def dashboard(): users = [] for u in User.objects(username=current_user.username): users.append({"credit": u.own, "debt": u.debt, "username":u.username}) return render_template('auth/dashboard.html', users=users, name=current_user.username)
def load_user(user_id): return User.objects(pk=user_id).first()
def update_debt(username, balance): User.objects(username=username).update(set__debt=balance)
def user_exists(username): existing_user = User.objects(username=username).first() if existing_user: return True else: return False
def update_account(username, balance): User.objects(username=username).update(set__own=balance) '''