def login():
form = LoginForm()
if form.validate_on_submit():
user = User.first(email=form.email.data)
login_user(user, remember=form.rememberMe.data)
redirectRoute = 'user.profile'
if user.is_admin:
redirectRoute = 'admin.index'
elif user.is_company:
redirectRoute = 'company.profile'
return redirect(url_for(redirectRoute))
return render_template('login.html', form=form)
def load_user(user_id):
return User.first(id=user_id)
def validate_email(self, field):
if User.first(email=field.data):
raise ValidationError('邮箱已经存在')
def validate_username(self, field):
if User.first(name=field.data):
raise ValidationError('名字已经存在')
def validate_password(self, field):
user = User.first(email=self.email.data)
if user and not user.check_password(field.data):
raise ValidationError('密码错误')
def validate_email(self, field: Field):
if field.data and not User.first(email=field.data):
raise ValidationError('该邮箱未注册')
