def login():
if request.method == 'GET':
return render_template('login.html')
else:
account = request.form['account']
password = request.form['password']
if len(account) == 0 | len(password) == 0:
return render_template('login.html')
cursor.execute('select account from coursefilemanagement.user')
accounts = cursor.fetchall()
for account in accounts:
if request.form['account'] == account[0]:
cursor.execute(
'select password,usertype from coursefilemanagement.user where account=%s',
(account[0]))
detail = cursor.fetchall()
if request.form['password'] == detail[0][0]:
curr_user = User()
curr_user.id = account[0]
print(curr_user.id)
login_user(curr_user)
return redirect(url_for('.homepage'))
return '<h>账号、密码错误！</h>'
def load_user(user_id):
curr_user = User()
curr_user.id = user_id
cursor.execute(
'select usertype '
'from coursefilemanagement.user '
'where account=%s', user_id)
detail = cursor.fetchall()
ac_type = detail[0][0]
curr_user.utype = ac_type
return curr_user
def setUp(self):
user = User()
user.id = 1
user.username = '******'
user.email = '*****@*****.**'
self.user = user
