from motor.motor_asyncio import AsyncIOMotorClient

from repos.assets_repo import AssetsRepo
from repos.char_repo import CharacterRepo
from repos.generation_repo import GenerationRepo
from repos.user_repo import UsersRepo


class DAO:
    def __init__(self, client: AsyncIOMotorClient, db_name="bot_db"):
        self.chars = CharacterRepo(client, db_name)
        self.assets = AssetsRepo(client, db_name)
        self.generations = GenerationRepo(client, db_name)