from sqlalchemy.ext.asyncio import AsyncSession
from .user import UserRepository
from ..domain import User


class Repository:
    def __init__(self, session: AsyncSession):
        self.session = session
        self.user = UserRepository(User, session)

    async def commit(self) -> None:
        await self.session.commit()

    async def rollback(self) -> None:
        await self.session.rollback()