from sqlalchemy.ext.asyncio import AsyncSession

from src.db.models.budget_form import BudgetForm
from src.repository.sheet_repository import SheetRepository



class SheetService:
    def __init__(self, db: AsyncSession):
        self.db = db
        self.sheet_repo = SheetRepository(db)

    async def get(
            self, 
            sheet: str, 
            direction: str, 
            sections: list[str] | None = None,
            form_id: int | None = None,
            form: BudgetForm | None = None
    ) -> list[tuple]:        
        return await self.sheet_repo.get(
            form_id=form_id,
            sheet=sheet,
            direction=direction,
            sections=sections,
        )