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('该邮箱未注册')