Commit 128880ea authored by fangzhipeng's avatar fangzhipeng

添加获取token方法

修改登录校验逻辑
parent c5bb270b
......@@ -24,7 +24,7 @@ def login_require(func):
raise BussinessException(error_type.NEED_LOGIN)
user_token = user_dao.select_user_token_info(token)
#token信息不存在或与存在信息不符合需要进行登录
if not user_token or user_token.user_id != user_id:
if not (user_token and user_token.user_id == user_id):
raise BussinessException(error_type.NEED_LOGIN)
return func(*args, **kwargs)
......
......@@ -7,7 +7,6 @@ from daylife.dao.models import UserInfo, UserToken
"""
session = sqlengin.getSession()
session.commit()
session.close()
"""
......@@ -38,3 +37,13 @@ def create_token(user_info):
session.commit()
session.close()
return token
def select_user_token_info(token):
session = sqlengin.getSession()
item = session.query(UserToken).filter(UserToken.token == token).filter(UserToken.device_type == 0).first()
session.close()
return item
if __name__ == '__main__':
print(select_user_token_info('89bb221c349511e89cecb8e85634f0ac'))
\ No newline at end of file
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