group_fqcd_jr/tools/seed_sim_account_demo.py

"""场内模拟交易演示种子（§T 用户自助）。

按 D4 决策：1 个客户 + 10 万初始资金 + 2 只基金的初始持仓。

执行：
    python -m tools.seed_sim_account_demo [--customer-id N] [--quotes-only]

## 已知表结构问题（2026-09-12 实测）

底座 `fin_*` 系列表的 `id` 列**没有** `AUTO_INCREMENT` 属性（仅 `fin_knowledge_meta` 有）。
SQLAlchemy ORM 默认期望 auto-increment，会在 INSERT 时省略 id → MySQL 报
`Field 'id' doesn't have a default value`。**底层基础规则不允许 DDL 改动**，
所以本脚本显式查 `MAX(id) + 1` 分配下一个 id，写入时携带 id（不依赖 auto-increment）。

实现：使用 SQLAlchemy Core `insert(...).values(id=..., ...)`，而不是 ORM `session.add()`
——后者对 BIGINT 不会自动注入 id。

幂等：再次执行不会重复建记录（先查 product_code / customer_id / customer_id+product_id）。
"""

from __future__ import annotations

import argparse
import asyncio
import sys
from datetime import UTC, datetime, timedelta
from decimal import Decimal

from sqlalchemy import func, insert, select, update
from sqlalchemy.orm import Session

from app.core.config import get_settings
from app.infrastructure.db import SessionFactory
from app.model.fund import (
    FundHolding,
    FundMarketPrice,
    FundProduct,
    FundSimAccount,
)


CUSTOMER_ID = 9001
INITIAL_BALANCE = Decimal("100000.00")
DEMO_PRODUCTS = [
    {
        "product_code": "510300",
        "product_name": "沪深300ETF",
        "exchange_code": "SSE",
        "product_category": "ETF",
        "risk_level": "R3",
        "lot_size": Decimal("100"),
        "price_tick": Decimal("0.001"),
        "close_price": Decimal("4.5000"),
        "total_fund_shares": Decimal("10000000000"),
        "initial_quantity": Decimal("1000"),
    },
    {
        "product_code": "510500",
        "product_name": "南方中证500ETF",
        "exchange_code": "SSE",
        "product_category": "ETF",
        "risk_level": "R3",
        "lot_size": Decimal("100"),
        "price_tick": Decimal("0.001"),
        "close_price": Decimal("6.2000"),
        "total_fund_shares": Decimal("8000000000"),
        "initial_quantity": Decimal("800"),
    },
]


async def _next_id(session: Session, model) -> int:
    """返回该表下一个可用的 id（不依赖 AUTO_INCREMENT）。"""
    pk_col = model.__table__.primary_key.columns[0]
    result = await session.execute(select(func.coalesce(func.max(pk_col), 0)))
    return int(result.scalar_one()) + 1


async def _upsert_product(session: Session, spec: dict) -> int:
    existing = (
        await session.execute(
            select(FundProduct.id).where(FundProduct.product_code == spec["product_code"])
        )
    ).scalar_one_or_none()
    if existing is not None:
        return int(existing)
    now = datetime.now(UTC).replace(tzinfo=None)
    next_id = await _next_id(session, FundProduct)
    stmt = insert(FundProduct).values(
        id=next_id,
        product_code=spec["product_code"],
        product_name=spec["product_name"],
        exchange_code=spec["exchange_code"],
        product_category=spec["product_category"],
        risk_level=spec["risk_level"],
        fund_manager="南方基金",
        currency="CNY",
        lot_size=spec["lot_size"],
        price_tick=spec["price_tick"],
        current_nav=spec["close_price"],
        current_nav_at=now,
        min_amount=Decimal("100.00"),
        open_start_at=now - timedelta(days=365),
        open_end_at=None,
        transaction_fee_rate=None,
        single_investor_max_holding_ratio=Decimal("5.0000"),
        management_fee_rate=Decimal("0.50"),
        custodian_fee_rate=Decimal("0.10"),
        risk_disclosure_required=0,
        second_confirmation_required=0,
        recording_required=0,
        status="上市",
        created_at=now,
        updated_at=now,
    )
    await session.execute(stmt)
    return next_id


async def _upsert_market_price(session: Session, product_id: int, spec: dict) -> None:
    today = datetime.now(UTC).date()
    now = datetime.now(UTC).replace(tzinfo=None)
    existing = (
        await session.execute(
            select(FundMarketPrice.id).where(
                FundMarketPrice.product_id == product_id,
                FundMarketPrice.trade_date == today,
            )
        )
    ).scalar_one_or_none()
    if existing is not None:
        await session.execute(
            update(FundMarketPrice)
            .where(FundMarketPrice.id == existing)
            .values(
                open_price=spec["close_price"],
                high_price=spec["close_price"] + Decimal("0.05"),
                low_price=spec["close_price"] - Decimal("0.05"),
                close_price=spec["close_price"],
                volume=Decimal("1000000"),
                turnover_amount=spec["close_price"] * Decimal("1000000"),
                total_fund_shares=spec["total_fund_shares"],
                source="eastmoney_demo_seed",
                source_updated_at=now,
            )
        )
        return
    next_id = await _next_id(session, FundMarketPrice)
    stmt = insert(FundMarketPrice).values(
        id=next_id,
        product_id=product_id,
        trade_date=today,
        open_price=spec["close_price"],
        high_price=spec["close_price"] + Decimal("0.05"),
        low_price=spec["close_price"] - Decimal("0.05"),
        close_price=spec["close_price"],
        volume=Decimal("1000000"),
        turnover_amount=spec["close_price"] * Decimal("1000000"),
        total_fund_shares=spec["total_fund_shares"],
        source="eastmoney_demo_seed",
        source_updated_at=now,
        created_at=now,
    )
    await session.execute(stmt)


async def _upsert_account(
    session: Session, customer_id: int
) -> tuple[FundSimAccount, bool]:
    existing = (
        await session.execute(
            select(FundSimAccount.id).where(FundSimAccount.customer_id == customer_id)
        )
    ).scalar_one_or_none()
    if existing is not None:
        account = (
            await session.execute(
                select(FundSimAccount).where(FundSimAccount.customer_id == customer_id)
            )
        ).scalar_one()
        return account, False
    now = datetime.now(UTC).replace(tzinfo=None)
    next_id = await _next_id(session, FundSimAccount)
    stmt = insert(FundSimAccount).values(
        id=next_id,
        account_no=f"FSA{customer_id:06d}",
        customer_id=customer_id,
        currency="CNY",
        cash_balance=INITIAL_BALANCE,
        available_cash=INITIAL_BALANCE,
        frozen_cash=Decimal("0"),
        initial_balance=INITIAL_BALANCE,
        status="正常",
        version=0,
        created_at=now,
        updated_at=now,
    )
    await session.execute(stmt)
    account = (
        await session.execute(
            select(FundSimAccount).where(FundSimAccount.customer_id == customer_id)
        )
    ).scalar_one()
    return account, True


async def _upsert_holding(
    session: Session, customer_id: int, product_id: int, spec: dict
) -> Decimal:
    existing = (
        await session.execute(
            select(FundHolding.id).where(
                FundHolding.customer_id == customer_id,
                FundHolding.product_id == product_id,
            )
        )
    ).scalar_one_or_none()
    if existing is not None:
        return Decimal("0")
    qty = spec["initial_quantity"]
    cost = (qty * spec["close_price"]).quantize(Decimal("0.01"))
    now = datetime.now(UTC).replace(tzinfo=None)
    next_id = await _next_id(session, FundHolding)
    stmt = insert(FundHolding).values(
        id=next_id,
        customer_id=customer_id,
        trade_account=f"FSA{customer_id:06d}",
        product_id=product_id,
        total_quantity=qty,
        # `shares`/`current_value` 在 docs/00 §6.2 标注为"生成列"，但实测 MySQL 表
        # `GENERATION_EXPRESSION=''`——表结构里是普通 NOT NULL 列、无默认值。
        # 写时必须显式给值，按 §6.2 的语义填（shares=total_quantity, current_value=cost）。
        shares=qty,
        available_quantity=qty,
        frozen_quantity=Decimal("0"),
        average_cost=spec["close_price"],
        cost_amount=cost,
        market_value=None,
        current_value=cost,
        profit_loss=None,
        profit_loss_ratio=None,
        status="持有中",
        first_acquired_at=now,
        version=0,
        updated_at=now,
    )
    await session.execute(stmt)
    return cost


async def run(customer_id: int, *, quotes_only: bool = False) -> None:
    s = get_settings()
    print(f"数据库：{s.mysql_dsn.split('@')[-1]}")
    print(f"目标客户 ID = {customer_id}")
    async with SessionFactory() as session:
        async with session.begin():
            product_ids: list[int] = []
            for spec in DEMO_PRODUCTS:
                pid = await _upsert_product(session, spec)
                await _upsert_market_price(session, pid, spec)
                product_ids.append(pid)
            print(f"  [完成] 演示产品 {len(product_ids)} 个 + 当日行情")
            if quotes_only:
                return
            account, account_created = await _upsert_account(session, customer_id)
            added_cost = Decimal("0")
            for pid, spec in zip(product_ids, DEMO_PRODUCTS, strict=True):
                added_cost += await _upsert_holding(session, customer_id, pid, spec)
            if added_cost:
                starting_cash = INITIAL_BALANCE if account_created else account.cash_balance
                account.cash_balance = (starting_cash - added_cost).quantize(Decimal("0.01"))
                account.available_cash = account.cash_balance
                account.updated_at = datetime.now(UTC).replace(tzinfo=None)
                await session.execute(
                    update(FundSimAccount)
                    .where(FundSimAccount.id == account.id)
                    .values(
                        cash_balance=account.cash_balance,
                        available_cash=account.available_cash,
                        updated_at=account.updated_at,
                    )
                )
        print(f"  [完成] 虚拟账户 {account.account_no} 初始余额 CNY {INITIAL_BALANCE}")
        print(f"  [完成] 持仓已建立，本次新增持仓成本 CNY {added_cost}")
        print(f"  [完成] 当前账户现金 CNY {account.cash_balance}")


def main() -> int:
    parser = argparse.ArgumentParser(description="场内模拟交易演示种子")
    parser.add_argument("--customer-id", type=int, default=CUSTOMER_ID)
    parser.add_argument(
        "--quotes-only",
        action="store_true",
        help="仅刷新演示产品与当日行情，不修改账户、持仓或余额",
    )
    args = parser.parse_args()
    asyncio.run(run(args.customer_id, quotes_only=args.quotes_only))
    return 0


if __name__ == "__main__":
    sys.exit(main())