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