fix get user by uid

This commit is contained in:
pycook 2019-11-19 21:46:53 +08:00
parent 1c6d056d58
commit bc7b51c544
2 changed files with 3 additions and 3 deletions

View File

@ -34,7 +34,7 @@ API_PACKAGE = "api"
@login_manager.user_loader @login_manager.user_loader
def load_user(user_id): def load_user(user_id):
"""Load user by ID.""" """Load user by ID."""
return User.get_by_id(int(user_id)) return User.get_by(uid=int(user_id), first=True, to_dict=False)
class ReverseProxy(object): class ReverseProxy(object):

View File

@ -44,7 +44,7 @@ class UserCRUD(object):
@staticmethod @staticmethod
def update(uid, **kwargs): def update(uid, **kwargs):
user = User.get_by_id(uid) or abort(404, "User <{0}> does not exist".format(uid)) user = User.get_by(uid=uid, to_dict=False, first=True) or abort(404, "User <{0}> does not exist".format(uid))
UserCache.clean(uid) UserCache.clean(uid)
@ -59,7 +59,7 @@ class UserCRUD(object):
@classmethod @classmethod
def delete(cls, uid): def delete(cls, uid):
user = User.get_by_id(uid) or abort(404, "User <{0}> does not exist".format(uid)) user = User.get_by(uid=uid, to_dict=False, first=True) or abort(404, "User <{0}> does not exist".format(uid))
UserCache.clean(user) UserCache.clean(user)