Commit 2f7c6e45 authored by shenshuo's avatar shenshuo

生成令牌可以进行过期修改

parent e10d33f6
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding:utf-8 -*- # -*- coding:utf-8 -*-
import jwt, datetime,hashlib import jwt, datetime, hashlib
from .configs import configs as my_configs from .configs import configs as my_configs
...@@ -8,7 +8,7 @@ class AuthToken: ...@@ -8,7 +8,7 @@ class AuthToken:
def __init__(self): def __init__(self):
self.token_secret = my_configs.get('token_secret', '3AIiOq18i~H=WWTIGq4ODQyMzcsIdfghs') self.token_secret = my_configs.get('token_secret', '3AIiOq18i~H=WWTIGq4ODQyMzcsIdfghs')
def encode_auth_token(self, **kargs): def encode_auth_token(self, **kwargs):
""" """
生成认证Token 生成认证Token
:param user_id: string :param user_id: string
...@@ -17,18 +17,19 @@ class AuthToken: ...@@ -17,18 +17,19 @@ class AuthToken:
:return: string :return: string
""" """
try: try:
exp_time = kwargs.get('exp_time', 1)
payload = { payload = {
'exp': datetime.datetime.utcnow() + datetime.timedelta(days=1, seconds=10), 'exp': datetime.datetime.utcnow() + datetime.timedelta(days=exp_time, seconds=10),
'nbf': datetime.datetime.utcnow() - datetime.timedelta(seconds=10), 'nbf': datetime.datetime.utcnow() - datetime.timedelta(seconds=10),
'iat': datetime.datetime.utcnow(), 'iat': datetime.datetime.utcnow(),
'iss': 'auth: ss', 'iss': 'auth: ss',
'sub': 'my token', 'sub': 'my token',
'id': '15618718060', 'id': '15618718060',
'data': { 'data': {
'user_id': kargs.get('user_id',''), 'user_id': kwargs.get('user_id', ''),
'username': kargs.get('username',''), 'username': kwargs.get('username', ''),
'nickname': kargs.get('nickname',''), 'nickname': kwargs.get('nickname', ''),
'is_superuser': kargs.get('is_superuser', False) 'is_superuser': kwargs.get('is_superuser', False)
} }
} }
return jwt.encode( return jwt.encode(
...@@ -58,6 +59,7 @@ class AuthToken: ...@@ -58,6 +59,7 @@ class AuthToken:
except jwt.InvalidTokenError: except jwt.InvalidTokenError:
return dict(status=-2, msg='无效Token') return dict(status=-2, msg='无效Token')
def gen_md5(pd): def gen_md5(pd):
m2 = hashlib.md5() m2 = hashlib.md5()
m2.update(pd.encode("utf-8")) m2.update(pd.encode("utf-8"))
......
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