|
|
@@ -6,7 +6,7 @@ from werkzeug.security import generate_password_hash, check_password_hash
|
|
|
from .models import User
|
|
|
from . import db
|
|
|
|
|
|
-from .forms import LoginForm, RegForm, MFAForm
|
|
|
+from .forms import LoginForm, RegForm, MFAForm, SearchForm
|
|
|
|
|
|
# MFA
|
|
|
import pyotp
|
|
|
@@ -17,6 +17,8 @@ accounts = Blueprint('accounts', __name__)
|
|
|
def login():
|
|
|
form = LoginForm()
|
|
|
|
|
|
+ seform = SearchForm()
|
|
|
+
|
|
|
if form.validate_on_submit():
|
|
|
email = form.email.data
|
|
|
challenge_passwd = form.passwd.data
|
|
|
@@ -33,7 +35,7 @@ def login():
|
|
|
else:
|
|
|
flash('Unsucessful Login!', category='error')
|
|
|
|
|
|
- return render_template('login.html', user = current_user, form = form)
|
|
|
+ return render_template('login.html', user = current_user, form = form, seform = SearchForm())
|
|
|
|
|
|
@accounts.route('/logout')
|
|
|
@login_required
|
|
|
@@ -46,6 +48,7 @@ def register():
|
|
|
pass_list = list()
|
|
|
|
|
|
form = RegForm()
|
|
|
+ seform = SearchForm()
|
|
|
|
|
|
if form.validate_on_submit():
|
|
|
email = form.email.data
|
|
|
@@ -84,7 +87,7 @@ def register():
|
|
|
return redirect(url_for('dashboards.market'))
|
|
|
else:
|
|
|
flash('Registration Failed', category='error')
|
|
|
- return render_template("register.html", user = current_user, form = form)
|
|
|
+ return render_template("register.html", user = current_user, form = form, seform = seform)
|
|
|
|
|
|
|
|
|
@accounts.route('/mfa', methods=['GET', 'POST'])
|
|
|
@@ -93,6 +96,8 @@ def mfa():
|
|
|
user_chal = request.args['user_chal']
|
|
|
user = User.query.filter_by(id = user_chal).first()
|
|
|
|
|
|
+ seform = SearchForm()
|
|
|
+
|
|
|
# check for existing totphash
|
|
|
if not user.totphash:
|
|
|
# generate random secret key for auth
|
|
|
@@ -119,4 +124,4 @@ def mfa():
|
|
|
return redirect(url_for('accounts.mfa'))
|
|
|
|
|
|
|
|
|
- return render_template('mfa.html', secret = secret, form = form, user = user)
|
|
|
+ return render_template('mfa.html', secret = secret, form = form, user = user, seform = seform)
|