core/services/auth/users.py

import asyncio
from sqlalchemy.orm import selectinload
from orm.user import User


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):
        self = UserStorage
        async with self.lock:
            return self.users.get(id)

    @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_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]