140 lines
5.3 KiB
Python
140 lines
5.3 KiB
Python
|
"""按 fin_product 为指定客户生成 fin_holdings 持仓数据（本地开发/演示用）。
|
|||
|
|
|
|||
|
|
用法：
|
|||
|
|
python scripts/seed_holdings_for_customer.py --customer-id 18
|
|||
|
|
python scripts/seed_holdings_for_customer.py --customer-id 18 --count 5 --force
|
|||
|
|
|
|||
|
|
规则：
|
|||
|
|
- 只挑选 fin_product 中 status=在售 的产品，按 id 升序取前 count 只；
|
|||
|
|
- 每笔持仓的买入净值 = 当前净值 × (1 - 浮动)，浮动由固定 seed 生成，保证可复现；
|
|||
|
|
- shares = cost_amount / 买入净值（4 位小数），current_value = shares × 当前净值；
|
|||
|
|
- profit_loss / profit_ratio 由上述字段推导；
|
|||
|
|
- 客户已有持仓时默认拒绝，需 --force 才会追加。
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import argparse
|
|||
|
|
import asyncio
|
|||
|
|
import random
|
|||
|
|
import sys
|
|||
|
|
from decimal import Decimal, ROUND_HALF_UP
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
|
|||
|
|
|
|||
|
|
from sqlalchemy import select
|
|||
|
|
|
|||
|
|
from config.database.mysql import get_session_factory
|
|||
|
|
from model.fin_holdings import FinHoldings
|
|||
|
|
from model.fin_product import FinProduct
|
|||
|
|
from model.sys_user import SysUser
|
|||
|
|
|
|||
|
|
_MONEY = Decimal("0.01")
|
|||
|
|
_SHARES = Decimal("0.0001")
|
|||
|
|
_RATIO = Decimal("0.0001")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _money(value: Decimal) -> Decimal:
|
|||
|
|
return value.quantize(_MONEY, rounding=ROUND_HALF_UP)
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def seed(customer_id: int, *, count: int, force: bool) -> list[dict]:
|
|||
|
|
session_factory = get_session_factory()
|
|||
|
|
async with session_factory() as db:
|
|||
|
|
user = await db.get(SysUser, customer_id)
|
|||
|
|
if user is None or user.user_type != "CUSTOMER":
|
|||
|
|
raise SystemExit(f"customer_id={customer_id} 不是客户用户或不存在")
|
|||
|
|
|
|||
|
|
existing = (
|
|||
|
|
await db.execute(
|
|||
|
|
select(FinHoldings).where(FinHoldings.customer_id == customer_id)
|
|||
|
|
)
|
|||
|
|
).scalars().all()
|
|||
|
|
if existing and not force:
|
|||
|
|
raise SystemExit(
|
|||
|
|
f"客户 {customer_id} 已有 {len(existing)} 条持仓，如需追加请加 --force"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
products = (
|
|||
|
|
await db.execute(
|
|||
|
|
select(FinProduct)
|
|||
|
|
.where(FinProduct.status == "在售")
|
|||
|
|
.order_by(FinProduct.id)
|
|||
|
|
.limit(count)
|
|||
|
|
)
|
|||
|
|
).scalars().all()
|
|||
|
|
if not products:
|
|||
|
|
raise SystemExit("fin_product 中没有在售产品")
|
|||
|
|
|
|||
|
|
rng = random.Random(f"holdings-{customer_id}")
|
|||
|
|
created: list[dict] = []
|
|||
|
|
for product in products:
|
|||
|
|
nav = product.nav or Decimal("1.000000")
|
|||
|
|
cost_amount = _money(Decimal(rng.randint(5_000, 50_000)))
|
|||
|
|
# 买入净值在当前净值的 88%~97% 之间浮动，形成有涨有跌的持仓
|
|||
|
|
buy_nav = _money(nav * Decimal(str(round(rng.uniform(0.88, 0.97), 6))))
|
|||
|
|
if buy_nav <= 0:
|
|||
|
|
buy_nav = Decimal("1.0000")
|
|||
|
|
shares = (cost_amount / buy_nav).quantize(_SHARES, rounding=ROUND_HALF_UP)
|
|||
|
|
current_value = _money(shares * nav)
|
|||
|
|
profit_loss = _money(current_value - cost_amount)
|
|||
|
|
profit_ratio = (profit_loss / cost_amount).quantize(_RATIO, rounding=ROUND_HALF_UP)
|
|||
|
|
|
|||
|
|
db.add(
|
|||
|
|
FinHoldings(
|
|||
|
|
customer_id=customer_id,
|
|||
|
|
product_id=product.id,
|
|||
|
|
shares=shares,
|
|||
|
|
cost_amount=cost_amount,
|
|||
|
|
current_value=current_value,
|
|||
|
|
profit_loss=profit_loss,
|
|||
|
|
profit_ratio=profit_ratio,
|
|||
|
|
status="持有中",
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
created.append(
|
|||
|
|
{
|
|||
|
|
"product_id": product.id,
|
|||
|
|
"product_name": product.product_name,
|
|||
|
|
"buy_nav": str(buy_nav),
|
|||
|
|
"nav": str(nav),
|
|||
|
|
"shares": str(shares),
|
|||
|
|
"cost_amount": str(cost_amount),
|
|||
|
|
"current_value": str(current_value),
|
|||
|
|
"profit_loss": str(profit_loss),
|
|||
|
|
"profit_ratio": str(profit_ratio),
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
await db.commit()
|
|||
|
|
return created
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main() -> None:
|
|||
|
|
parser = argparse.ArgumentParser(description="按 fin_product 生成客户持仓数据")
|
|||
|
|
parser.add_argument("--customer-id", type=int, required=True)
|
|||
|
|
parser.add_argument("--count", type=int, default=8, help="生成持仓条数（默认 8）")
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--force", action="store_true", help="客户已有持仓时仍允许追加"
|
|||
|
|
)
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
created = asyncio.run(seed(args.customer_id, count=args.count, force=args.force))
|
|||
|
|
total_cost = sum(Decimal(item["cost_amount"]) for item in created)
|
|||
|
|
total_value = sum(Decimal(item["current_value"]) for item in created)
|
|||
|
|
print(f"客户 {args.customer_id} 新增持仓 {len(created)} 条：")
|
|||
|
|
for item in created:
|
|||
|
|
print(
|
|||
|
|
f" [{item['product_id']}] {item['product_name'][:24]}… "
|
|||
|
|
f"买入净值={item['buy_nav']} 当前净值={item['nav']} "
|
|||
|
|
f"份额={item['shares']} 成本={item['cost_amount']} "
|
|||
|
|
f"市值={item['current_value']} 盈亏={item['profit_loss']} "
|
|||
|
|
f"({item['profit_ratio']})"
|
|||
|
|
)
|
|||
|
|
print(f"合计：成本 {total_cost} 元，市值 {total_value} 元，"
|
|||
|
|
f"盈亏 {total_value - total_cost} 元")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|