43 lines
891 B
Python
43 lines
891 B
Python
|
|
||
|
|
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)).all()
|
||
|
|
self.users = dict([(user.id, user) for user in users])
|
||
|
|
print('[storage.users] %d ' % len(self.users))
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
async def get_user(id):
|
||
|
|
self = UserStorage
|
||
|
|
async with self.lock:
|
||
|
|
return self.users.get(id)
|
||
|
|
|
||
|
|
@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]
|