"""trade_order 交易申请单 ORM 模型（风控阻断闭环主表，状态机单一来源）。"""
from __future__ import annotations

from datetime import datetime
from decimal import Decimal

from sqlalchemy import BigInteger, DateTime, Numeric, String, func
from sqlalchemy.orm import Mapped, mapped_column

from model.base import Base


class TradeOrder(Base):
    __tablename__ = "trade_order"
    __table_args__ = {"comment": "交易申请单（风控阻断闭环主表，状态机单一来源）"}

    id: Mapped[int] = mapped_column(BigInteger, primary_key=True, autoincrement=True)
    order_no: Mapped[str] = mapped_column(String(32), unique=True)
    customer_id: Mapped[int] = mapped_column(BigInteger)
    product_id: Mapped[int] = mapped_column(BigInteger)
    advisor_id: Mapped[int | None] = mapped_column(BigInteger)
    order_type: Mapped[str] = mapped_column(String(16))
    amount: Mapped[Decimal | None] = mapped_column(Numeric(18, 2))
    shares: Mapped[Decimal | None] = mapped_column(Numeric(18, 4))
    nav: Mapped[Decimal | None] = mapped_column(Numeric(12, 6))
    fee: Mapped[Decimal | None] = mapped_column(Numeric(12, 2))
    status: Mapped[str] = mapped_column(String(16), server_default="待确认")
    risk_alert_id: Mapped[int | None] = mapped_column(BigInteger)
    cancel_reason: Mapped[str | None] = mapped_column(String(255))
    create_time: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
    confirm_time: Mapped[datetime | None] = mapped_column(DateTime)