def login():
if request.method == 'POST':
zid = request.form['zid']
password = request.form['password']
if Eventsystem.check_digital(zid):
try:
user = Eventsystem.validate_login(int(zid), password)
login_user(user)
return redirect(url_for('index'))
except ErrorMessage as error:
return render_template('login.html',
val=True,
message=error.msg)
else:
try:
user = Eventsystem.validate_login_guest(zid, password)
login_user(user)
return redirect(url_for('index'))
except ErrorMessage as error:
return render_template('login.html',
val=True,
message=error.msg)
return render_template('login.html')
def test_guest_validate_login(self):
print('test_guest_validate_login')
try:
Eventsystem.validate_login_guest('*****@*****.**', 'li199980812')
except ErrorMessage as error:
assert error.msg == 'Please ensure the username and password'
def test_guest_login(self):
print('test_guest_login')
user = Eventsystem.validate_login_guest('*****@*****.**', 'li1998')
assert user is not None
assert user.role == 'guest'
