def register():
error = None
form = SignupForm()
if request.method == 'POST':
if request.form.get('submit') == 'Cancel':
return redirect(url_for('UsersFiles.login'))
elif form.validate_on_submit():
try:
user = Users(
user_name=form.user_name.data,
user_password=form.user_password1.data
)
db.create_all()
db.session.add(user)
db.session.commit()
Log("SIGNUP")
login_user(user)
Log("LOGIN")
# add hash
addCompHash(str(current_user))
flash('You are just login!')
return redirect(url_for('HomeFiles.home'))
except:
error = 'Signup Fail'
db.session.rollback()
Log("SIGNUP FAIL")
# trace= get_current_traceback(skip=1, show_hidden_frames=True,
# ignore_system_exceptions=False)
# return redirect(url_for('ShareFiles.stack_trace', trace=trace))
return render_template('register.html', form = form, error = error)
def login():
error = None
form = LoginForm()
if request.method == 'POST':
if request.form.get('submit') == 'Register':
return redirect(url_for('UsersFiles.register'))
elif form.validate_on_submit():
session.permanent = True
Log("LOGIN")
# compare hash
addCompHash(str(current_user))
flash('You are just login!')
try:
if isHack(str(current_user)):
flash('!!!!! YOU GOT HACK !!!!!!')
except:
flash('First time login')
return redirect(url_for('HomeFiles.home'))
else:
error = 'Invalid username or password. Please try again.'
Log("LOGIN FAIL")
return render_template('login.html', form=form, error=error)
elif request.method == 'GET':
return render_template('login.html', form=form)