Commit 128880ea authored by fangzhipeng's avatar fangzhipeng

添加获取token方法

修改登录校验逻辑
parent c5bb270b
...@@ -24,7 +24,7 @@ def login_require(func): ...@@ -24,7 +24,7 @@ def login_require(func):
raise BussinessException(error_type.NEED_LOGIN) raise BussinessException(error_type.NEED_LOGIN)
user_token = user_dao.select_user_token_info(token) user_token = user_dao.select_user_token_info(token)
#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) raise BussinessException(error_type.NEED_LOGIN)
return func(*args, **kwargs) return func(*args, **kwargs)
......
...@@ -7,7 +7,6 @@ from daylife.dao.models import UserInfo, UserToken ...@@ -7,7 +7,6 @@ from daylife.dao.models import UserInfo, UserToken
""" """
session = sqlengin.getSession() session = sqlengin.getSession()
session.commit()
session.close() session.close()
""" """
...@@ -38,3 +37,13 @@ def create_token(user_info): ...@@ -38,3 +37,13 @@ def create_token(user_info):
session.commit() session.commit()
session.close() session.close()
return token 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