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

from .model import EnergyReportReviewLogModel, EnergyReportReviewModel
from .schema import (
    EnergyReportReviewCreateSchema,
    EnergyReportReviewLogCreateSchema,
    EnergyReportReviewUpdateSchema,
)


class EnergyReportReviewCRUD(CRUDBase[EnergyReportReviewModel, EnergyReportReviewCreateSchema, EnergyReportReviewUpdateSchema]):
    """能源报告审核任务数据层。"""

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


class EnergyReportReviewLogCRUD(
    CRUDBase[EnergyReportReviewLogModel, EnergyReportReviewLogCreateSchema, EnergyReportReviewLogCreateSchema]
):
    """能源报告审核操作记录数据层。"""

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