"""财务数据模型。

资产负债表、利润表、现金流量表、科目余额。
"""

import uuid
from datetime import datetime, timezone

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

from app.core.database import Base
from app.core.types import JSONBType


class FinancialData(Base):
    """财务数据。"""

    __tablename__ = "financial_data"

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    company_id: Mapped[str] = mapped_column(String(36), ForeignKey("companies.id"), nullable=False, index=True)
    period_year: Mapped[int] = mapped_column(Integer, nullable=False, comment="年份")
    period_month: Mapped[int] = mapped_column(Integer, nullable=False, comment="月份")
    statement_type: Mapped[str] = mapped_column(String(50), nullable=False, comment="balance_sheet/income/cash_flow")
    data_json: Mapped[dict | None] = mapped_column(JSONBType, nullable=True, comment="财务数据 JSON")
    source: Mapped[str | None] = mapped_column(String(100), nullable=True, comment="数据来源")
    credibility_score: Mapped[float | None] = mapped_column(Float, nullable=True, comment="可信度评分（0-100）")
    validation_result: Mapped[dict | None] = mapped_column(JSONBType, nullable=True, comment="校验结果")
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), default=lambda: datetime.now(timezone.utc)
    )
    updated_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True),
        default=lambda: datetime.now(timezone.utc),
        onupdate=lambda: datetime.now(timezone.utc),
    )