"""trade_order 仓储：按单号/客户查申请单 + 状态流转（条件更新，不 commit）。"""
from __future__ import annotations

from sqlalchemy import select, update

from model.trade_order import TradeOrder
from repositories.base import BaseRepository


class TradeOrderRepo(BaseRepository):
    model = TradeOrder

    async def get_by_order_no(self, order_no: str) -> TradeOrder | None:
        return await self.db.scalar(
            select(TradeOrder).where(TradeOrder.order_no == order_no)
        )

    async def list_by_customer(
        self, customer_id: int, status: str | None = None
    ) -> list[TradeOrder]:
        """按客户查申请单，可按状态过滤，按创建时间倒序。"""
        stmt = select(TradeOrder).where(TradeOrder.customer_id == customer_id)
        if status is not None:
            stmt = stmt.where(TradeOrder.status == status)
        stmt = stmt.order_by(TradeOrder.id.desc())
        return list((await self.db.scalars(stmt)).all())

    async def update_status(self, order_id: int, *, status: str, **fields) -> int:
        """更新订单状态（可附带 risk_alert_id / confirm_time / cancel_reason 等）。

        不 commit，由 service 层事务统一提交；返回 rowcount（0 表示订单不存在）。
        """
        result = await self.db.execute(
            update(TradeOrder)
            .where(TradeOrder.id == order_id)
            .values(status=status, **fields)
        )
        return result.rowcount