Commit 9f5b10ca authored by shenshuo's avatar shenshuo

把邮箱也放入cookie

parent 31002a8f
...@@ -12,10 +12,7 @@ import jwt ...@@ -12,10 +12,7 @@ import jwt
class BaseHandler(RequestHandler): class BaseHandler(RequestHandler):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
self.new_csrf_key = str(shortuuid.uuid()) self.new_csrf_key = str(shortuuid.uuid())
self.user_id = None self.user_id, self.username, self.nickname, self.email, self.is_super = None, None, None, None, False
self.username = None
self.nickname = None
self.is_super = False
self.is_superuser = self.is_super self.is_superuser = self.is_super
super(BaseHandler, self).__init__(*args, **kwargs) super(BaseHandler, self).__init__(*args, **kwargs)
...@@ -58,6 +55,7 @@ class BaseHandler(RequestHandler): ...@@ -58,6 +55,7 @@ class BaseHandler(RequestHandler):
self.user_id = user_info.get('user_id', None) self.user_id = user_info.get('user_id', None)
self.username = user_info.get('username', None) self.username = user_info.get('username', None)
self.nickname = user_info.get('nickname', None) self.nickname = user_info.get('nickname', None)
self.email = user_info.get('email', None)
self.is_super = user_info.get('is_superuser', False) self.is_super = user_info.get('is_superuser', False)
if not self.user_id: if not self.user_id:
...@@ -67,6 +65,7 @@ class BaseHandler(RequestHandler): ...@@ -67,6 +65,7 @@ class BaseHandler(RequestHandler):
self.set_secure_cookie("user_id", self.user_id) self.set_secure_cookie("user_id", self.user_id)
self.set_secure_cookie("nickname", self.nickname) self.set_secure_cookie("nickname", self.nickname)
self.set_secure_cookie("username", self.username) self.set_secure_cookie("username", self.username)
self.set_secure_cookie("email", self.email)
self.is_superuser = self.is_super self.is_superuser = self.is_super
### SDK 不用处理鉴权 ### SDK 不用处理鉴权
# my_verify = MyVerify(self.user_id) # my_verify = MyVerify(self.user_id)
...@@ -92,6 +91,9 @@ class BaseHandler(RequestHandler): ...@@ -92,6 +91,9 @@ class BaseHandler(RequestHandler):
def get_current_nickname(self): def get_current_nickname(self):
return self.nickname return self.nickname
def get_current_email(self):
return self.email
def is_superuser(self): def is_superuser(self):
return self.is_superuser return self.is_superuser
......
...@@ -29,6 +29,7 @@ class AuthToken: ...@@ -29,6 +29,7 @@ class AuthToken:
'user_id': kwargs.get('user_id', ''), 'user_id': kwargs.get('user_id', ''),
'username': kwargs.get('username', ''), 'username': kwargs.get('username', ''),
'nickname': kwargs.get('nickname', ''), 'nickname': kwargs.get('nickname', ''),
'email': kwargs.get('email', ''),
'is_superuser': kwargs.get('is_superuser', False) 'is_superuser': kwargs.get('is_superuser', False)
} }
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment