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)
'''
