73 lines
1.9 KiB
Python
73 lines
1.9 KiB
Python
import asyncio
|
|
from sqlalchemy.orm import selectinload, exc
|
|
from orm.user import User
|
|
from base.orm import local_session
|
|
|
|
|
|
class UserStorage:
|
|
users = {}
|
|
lock = asyncio.Lock()
|
|
|
|
@staticmethod
|
|
def init(session):
|
|
self = UserStorage
|
|
users = (
|
|
session.query(User)
|
|
.options(selectinload(User.roles), selectinload(User.ratings))
|
|
.all()
|
|
)
|
|
self.users = dict([(user.id, user) for user in users])
|
|
print("[auth.users] %d precached" % len(self.users))
|
|
|
|
@staticmethod
|
|
async def get_user(id):
|
|
with local_session() as session:
|
|
try:
|
|
user = (
|
|
session.query(User).options(
|
|
selectinload(User.roles),
|
|
selectinload(User.ratings)
|
|
).filter(
|
|
User.id == id
|
|
).one()
|
|
)
|
|
return user
|
|
except exc.NoResultFound:
|
|
return None
|
|
|
|
@staticmethod
|
|
async def get_all_users():
|
|
self = UserStorage
|
|
async with self.lock:
|
|
aaa = list(self.users.values())
|
|
aaa.sort(key=lambda user: user.createdAt)
|
|
return aaa
|
|
|
|
@staticmethod
|
|
async def get_top_users():
|
|
self = UserStorage
|
|
async with self.lock:
|
|
aaa = list(self.users.values())
|
|
aaa.sort(key=lambda user: user.rating)
|
|
return aaa
|
|
|
|
@staticmethod
|
|
async def get_user_by_slug(slug):
|
|
self = UserStorage
|
|
async with self.lock:
|
|
for user in self.users.values():
|
|
if user.slug == slug:
|
|
return user
|
|
|
|
@staticmethod
|
|
async def add_user(user):
|
|
self = UserStorage
|
|
async with self.lock:
|
|
self.users[user.id] = user
|
|
|
|
@staticmethod
|
|
async def del_user(id):
|
|
self = UserStorage
|
|
async with self.lock:
|
|
del self.users[id]
|