from app.core.base_crud import CRUDBase
from app.core.base_schema import AuthSchema

from .model import EnergyReportGenerationLogModel, EnergyReportGenerationModel
from .schema import (
    EnergyReportGenerationCreateSchema,
    EnergyReportGenerationLogCreateSchema,
    EnergyReportGenerationUpdateSchema,
)


class EnergyReportGenerationCRUD(CRUDBase[EnergyReportGenerationModel, EnergyReportGenerationCreateSchema, EnergyReportGenerationUpdateSchema]):
    """能源报告生成任务数据层。"""

    def __init__(self, auth: AuthSchema) -> None:
        super().__init__(model=EnergyReportGenerationModel, auth=auth)


class EnergyReportGenerationLogCRUD(
    CRUDBase[EnergyReportGenerationLogModel, EnergyReportGenerationLogCreateSchema, EnergyReportGenerationLogCreateSchema]
):
    """能源报告生成操作记录数据层。"""

    def __init__(self, auth: AuthSchema) -> None:
        super().__init__(model=EnergyReportGenerationLogModel, auth=auth)