def login():
try:
if request.method == 'POST':
email = request.form['email']
password = request.form['password']
userModel = UserModel()
if userModel.is_valid(User(email, password)):
hashed_password = userModel.get(User(email)).password
if sha256_crypt.verify(password, hashed_password):
flash("Successfully logged!", 'success')
session['logged_in'] = True
session['email'] = email
else:
flash("Wrong credentials!", 'warning')
else:
flash("Invalid credentials. Try Again.", 'warning')
except Exception as exception:
flash('Database error', 'warning')
print(exception)
finally:
return redirect(url_for('home'))
def registration_page():
try:
if request.method == "POST":
email = request.form['email']
password = sha256_crypt.encrypt(request.form['password'])
userModel = UserModel()
if userModel.get(User(email=email)):
flash("User already exist", 'warning')
else:
userModel.create(User(email, password))
session['logged_in'] = True
session['email'] = email
flash("Thanks for registration :)", 'success')
except Exception as e:
print(e)
flash("Database error", 'warning')
else:
return redirect(url_for('home'))
