from sqlalchemy import ForeignKey, Integer, String, Text
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.core.base_model import ModelMixin, TenantMixin, UserMixin


class EnergyReportReviewModel(ModelMixin, TenantMixin, UserMixin):
    """能源报告审核任务模型。"""

    __tablename__: str = "energy_report_review"
    __table_args__: dict[str, str] = {"comment": "能源报告审核任务表"}
    __loader_options__: list[str] = [
        "template",
        "file",
        "attachment_file",
        "reviewer",
        "created_by",
        "updated_by",
        "deleted_by",
        "tenant_by",
    ]

    task_name: Mapped[str] = mapped_column(String(128), nullable=False, comment="任务名称", index=True)
    report_type: Mapped[int] = mapped_column(Integer, default=1, nullable=False, comment="报告类型 1:能源报告 2:光伏报告", index=True)
    template_id: Mapped[int | None] = mapped_column(
        Integer,
        ForeignKey("energy_report_template.id", ondelete="SET NULL", onupdate="CASCADE"),
        nullable=True,
        index=True,
        comment="能源报告模板ID",
    )
    file_id: Mapped[int] = mapped_column(
        Integer,
        ForeignKey("common_file.id", ondelete="RESTRICT", onupdate="CASCADE"),
        nullable=False,
        index=True,
        comment="审核文档文件ID",
    )
    attachment_file_id: Mapped[int | None] = mapped_column(
        Integer,
        ForeignKey("common_file.id", ondelete="SET NULL", onupdate="CASCADE"),
        nullable=True,
        index=True,
        comment="附件文件ID",
    )
    attachment_url: Mapped[str | None] = mapped_column(
        String(512),
        default=None,
        nullable=True,
        comment="附件下载地址",
    )
    status: Mapped[int] = mapped_column(
        Integer,
        default=0,
        nullable=False,
        comment="状态 0:待审查 1:审查中 2:已审查 3:审查失败",
        index=True,
    )
    reviewer_id: Mapped[int | None] = mapped_column(
        Integer,
        ForeignKey("sys_user.id", ondelete="SET NULL", onupdate="CASCADE"),
        nullable=True,
        index=True,
        comment="审查人ID",
    )
    description: Mapped[str | None] = mapped_column(Text, default=None, nullable=True, comment="描述")

    template = relationship("EnergyReportTemplateModel", lazy="selectin", foreign_keys=[template_id], uselist=False)
    file = relationship("FileModel", lazy="selectin", foreign_keys=[file_id], uselist=False)
    attachment_file = relationship(
        "FileModel",
        lazy="selectin",
        foreign_keys=[attachment_file_id],
        uselist=False,
    )
    reviewer = relationship("UserModel", lazy="selectin", foreign_keys=[reviewer_id], uselist=False)


class EnergyReportReviewLogModel(ModelMixin, TenantMixin, UserMixin):
    """能源报告审核操作记录模型。"""

    __tablename__: str = "energy_report_review_log"
    __table_args__: dict[str, str] = {"comment": "能源报告审核操作记录表"}
    __loader_options__: list[str] = [
        "review",
        "created_by",
        "updated_by",
        "deleted_by",
        "tenant_by",
    ]

    report_review_id: Mapped[int] = mapped_column(
        Integer,
        ForeignKey("energy_report_review.id", ondelete="CASCADE", onupdate="CASCADE"),
        nullable=False,
        index=True,
        comment="能源报告审核任务ID",
    )
    task_name: Mapped[str] = mapped_column(String(128), nullable=False, comment="任务名称快照", index=True)
    action_type: Mapped[str] = mapped_column(
        String(32),
        nullable=False,
        index=True,
        comment="操作类型:create_task/start_review/review_pass/review_reject",
    )
    action_name: Mapped[str] = mapped_column(String(64), nullable=False, comment="操作名称")
    description: Mapped[str | None] = mapped_column(Text, default=None, nullable=True, comment="操作描述")

    review = relationship("EnergyReportReviewModel", lazy="selectin", foreign_keys=[report_review_id], uselist=False)