瀏覽代碼

Merge pull request #786 from flying1020/2.1/bugfix/未分配空间用户登录错误

修复bug,未分配用户登录的时候,前端提示服务器错误。改为提示具体的错误信息。
walle-web.io 6 年之前
父節點
當前提交
d7eb05d875
共有 1 個文件被更改,包括 7 次插入2 次删除
  1. 7 2
      walle/api/passport.py

+ 7 - 2
walle/api/passport.py

@@ -16,6 +16,8 @@ from walle.api.api import ApiResource
 from walle.form.user import LoginForm
 from walle.model.user import UserModel
 from walle.service.code import Code
+from walle.service.error import WalleError
+
 
 class PassportAPI(ApiResource):
     actions = ['login', 'logout']
@@ -46,8 +48,11 @@ class PassportAPI(ApiResource):
             user = UserModel.query.filter_by(email=form.email.data).first()
 
             if user is not None and user.verify_password(form.password.data):
-                login_user(user)
-                user.fresh_session()
+                try:
+                    login_user(user)
+                    user.fresh_session()
+                except WalleError as e:
+                    return self.render_json(code=e.code, data=Code.code_msg[e.code])
                 return self.render_json(data=current_user.to_json())
 
         return self.render_json(code=Code.error_pwd, data=form.errors)