core/services/auth/roles.py

import asyncio
from sqlalchemy.orm import selectinload
from orm.rbac import Role

class RoleStorage:
	roles = {}
	lock = asyncio.Lock()

	@staticmethod
	def init(session):
		self = RoleStorage
		roles = session.query(Role).\
			options(selectinload(Role.permissions)).all()
		self.roles = dict([(role.id, role) for role in roles])
		print('[service.auth] %d roles' % len(roles))
		

	@staticmethod
	async def get_role(id):
		self = RoleStorage
		async with self.lock:
			return self.roles.get(id)

	@staticmethod
	async def add_role(role):
		self = RoleStorage
		async with self.lock:
			self.roles[id] = role

	@staticmethod
	async def del_role(id):
		self = RoleStorage
		async with self.lock:
			del self.roles[id]