def login():
# 创建表单对象
form = LoginForm()
# 判断表单是否提交
if form.validate_on_submit():
# 判断登录方式
if LoginForm.is_username_or_email(form.username.data):
# 创建user模型对象
u = User.query.filter_by(username=form.username.data).first()
else:
# 创建user模型对象
u = User.query.filter_by(email=form.username.data).first()
# 校验密码正确性
if u.check_password(form.password.data):
# 登录用户
login_user(u, remember=form.remember.data)
# 重定向到next页面或主页面
return redirect(request.args.get('next') or url_for('main.index'))
else:
flash('密码不正确')
return render_template('user/login.html', form=form)
