dekun
555 lines
20 KiB
Python
555 lines
20 KiB
Python
"""期权本地交易统计(胜率 / 盈亏 / 持仓时长)."""
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Any
|
|
|
|
from lib.instance.instance_embed_context_lib import profit_loss_ratio_from_averages
|
|
from lib.options.options_db import init_options_tables
|
|
|
|
|
|
def _safe_float(v: Any) -> float | None:
|
|
if v is None or v == "":
|
|
return None
|
|
try:
|
|
return float(v)
|
|
except (TypeError, ValueError):
|
|
return None
|
|
|
|
|
|
def _underlying_index_usdt(ex: Any, underly: str) -> float | None:
|
|
"""取标的 USDT 近似指数(币本位已平盈亏折 U).优先公开 ticker,避免私钥失败."""
|
|
u = (underly or "ETH").strip().upper() or "ETH"
|
|
pubs: list[Any] = []
|
|
try:
|
|
from lib.sim.hooks import _APP_MODULE, _sim_public_exchange
|
|
|
|
pub = _sim_public_exchange(ex) if _APP_MODULE is not None else None
|
|
if pub is not None:
|
|
pubs.append(pub)
|
|
except Exception:
|
|
pass
|
|
if ex is not None and ex not in pubs:
|
|
pubs.append(ex)
|
|
from lib.exchange.okx_options_lib import fetch_index_price
|
|
|
|
for pub in pubs:
|
|
try:
|
|
if hasattr(pub, "public_get_market_ticker"):
|
|
rows = (pub.public_get_market_ticker({"instId": f"{u}-USDT"}) or {}).get("data") or []
|
|
if rows:
|
|
last = _safe_float(rows[0].get("last") or rows[0].get("lastPx"))
|
|
if last is not None and last > 0:
|
|
return float(last)
|
|
except Exception:
|
|
pass
|
|
try:
|
|
px = fetch_index_price(pub, f"{u}-USD")
|
|
if px is not None and float(px) > 0:
|
|
return float(px)
|
|
except Exception:
|
|
pass
|
|
try:
|
|
t = pub.fetch_ticker(f"{u}/USDT") or {}
|
|
last = _safe_float(t.get("last") or t.get("close"))
|
|
if last is not None and last > 0:
|
|
return float(last)
|
|
except Exception:
|
|
continue
|
|
return None
|
|
|
|
|
|
def history_pnl_to_usdt(history: list[dict[str, Any]], ex: Any = None) -> list[dict[str, Any]]:
|
|
"""
|
|
统计用:优先已冻结的 realized_pnl_usdt(平仓当时口径,含手续费后净额);
|
|
否则币本位用 close_index_px(再否则实时指数)折 U;USDC 原样.
|
|
折算失败的币仓剔除盈亏字段,避免把「币数量」当成 U.
|
|
"""
|
|
from lib.options.options_margin_mode_lib import margin_mode_from_inst_id, premium_ccy_for_mode
|
|
|
|
idx_cache: dict[str, float | None] = {}
|
|
out: list[dict[str, Any]] = []
|
|
for row in history:
|
|
r = dict(row)
|
|
if r.get("status") == "open":
|
|
out.append(r)
|
|
continue
|
|
frozen = _safe_float(r.get("realized_pnl_usdt"))
|
|
if frozen is not None:
|
|
r["realized_pnl"] = round(float(frozen), 4)
|
|
out.append(r)
|
|
continue
|
|
pnl = _safe_float(r.get("realized_pnl"))
|
|
if pnl is None:
|
|
out.append(r)
|
|
continue
|
|
inst = str(r.get("inst_id") or "")
|
|
underly = str(r.get("underlying") or (inst.split("-")[0] if inst else "ETH") or "ETH")
|
|
ccy = str(r.get("premium_ccy") or "").strip().upper()
|
|
if not ccy:
|
|
ccy = premium_ccy_for_mode(margin_mode_from_inst_id(inst), underly)
|
|
if ccy in ("ETH", "BTC"):
|
|
idx = _safe_float(r.get("close_index_px"))
|
|
if idx is None or idx <= 0:
|
|
if underly not in idx_cache:
|
|
idx_cache[underly] = _underlying_index_usdt(ex, underly)
|
|
idx = idx_cache.get(underly)
|
|
if idx is None or idx <= 0:
|
|
r["realized_pnl"] = None
|
|
else:
|
|
r["realized_pnl"] = round(float(pnl) * float(idx), 4)
|
|
else:
|
|
r["realized_pnl"] = round(float(pnl), 4)
|
|
out.append(r)
|
|
return out
|
|
|
|
|
|
def _parse_ts(raw: Any) -> datetime | None:
|
|
if raw is None or raw == "":
|
|
return None
|
|
s = str(raw).strip().replace(" ", "T", 1)
|
|
try:
|
|
return datetime.fromisoformat(s)
|
|
except (TypeError, ValueError):
|
|
return None
|
|
|
|
|
|
def _hold_seconds(created_at: Any, closed_at: Any) -> float | None:
|
|
start = _parse_ts(created_at)
|
|
end = _parse_ts(closed_at)
|
|
if start is None or end is None:
|
|
return None
|
|
sec = (end - start).total_seconds()
|
|
return sec if sec >= 0 else None
|
|
|
|
|
|
def _avg_seconds(values: list[float]) -> float | None:
|
|
if not values:
|
|
return None
|
|
return round(sum(values) / len(values), 1)
|
|
|
|
|
|
def compute_options_stats_from_history(history: list[dict[str, Any]]) -> dict[str, Any]:
|
|
"""基于期权历史列表(交易所)计算统计."""
|
|
wins: list[float] = []
|
|
losses: list[float] = []
|
|
win_holds: list[float] = []
|
|
loss_holds: list[float] = []
|
|
all_holds: list[float] = []
|
|
open_holds: list[float] = []
|
|
now = datetime.now()
|
|
|
|
for row in history:
|
|
if row.get("status") == "open":
|
|
start = _parse_ts(row.get("created_at"))
|
|
if start is not None:
|
|
sec = (now - start).total_seconds()
|
|
if sec >= 0:
|
|
open_holds.append(sec)
|
|
continue
|
|
pnl_raw = row.get("realized_pnl")
|
|
if pnl_raw is None:
|
|
continue
|
|
try:
|
|
pnl = float(pnl_raw)
|
|
except (TypeError, ValueError):
|
|
continue
|
|
hold = _hold_seconds(row.get("created_at"), row.get("closed_at"))
|
|
if hold is not None:
|
|
all_holds.append(hold)
|
|
if pnl > 0:
|
|
wins.append(pnl)
|
|
if hold is not None:
|
|
win_holds.append(hold)
|
|
elif pnl < 0:
|
|
losses.append(pnl)
|
|
if hold is not None:
|
|
loss_holds.append(hold)
|
|
|
|
total_closed = len(wins) + len(losses)
|
|
win_rate = round(len(wins) / total_closed * 100, 2) if total_closed else 0
|
|
avg_win = sum(wins) / len(wins) if wins else None
|
|
avg_loss = sum(losses) / len(losses) if losses else None
|
|
|
|
total_profit = round(sum(wins), 4) if wins else 0.0
|
|
total_loss = round(abs(sum(losses)), 4) if losses else 0.0
|
|
net_realized = round(sum(wins) + sum(losses), 4)
|
|
return {
|
|
"total_closed": total_closed,
|
|
"win_count": len(wins),
|
|
"loss_count": len(losses),
|
|
"win_rate": win_rate,
|
|
"profit_loss_ratio": profit_loss_ratio_from_averages(avg_win, avg_loss),
|
|
"avg_win": round(avg_win, 4) if avg_win is not None else None,
|
|
"avg_loss": round(abs(avg_loss), 4) if avg_loss is not None else None,
|
|
"total_profit": total_profit,
|
|
"total_loss": total_loss,
|
|
"net_realized_pnl": net_realized,
|
|
"avg_hold_sec": _avg_seconds(all_holds),
|
|
"avg_win_hold_sec": _avg_seconds(win_holds),
|
|
"avg_loss_hold_sec": _avg_seconds(loss_holds),
|
|
"open_count": len(open_holds),
|
|
"avg_open_hold_sec": _avg_seconds(open_holds),
|
|
}
|
|
|
|
|
|
_STATS_WINDOWS = ("today", "7d", "30d", "all")
|
|
|
|
|
|
def normalize_stats_window(raw: Any) -> str:
|
|
w = str(raw or "7d").strip().lower()
|
|
if w in ("1d", "day", "今日"):
|
|
return "today"
|
|
if w in ("7", "week"):
|
|
return "7d"
|
|
if w in ("30", "month"):
|
|
return "30d"
|
|
if w in ("0", "全部", "total"):
|
|
return "all"
|
|
return w if w in _STATS_WINDOWS else "7d"
|
|
|
|
|
|
def stats_window_bounds(window: str, *, now: datetime | None = None) -> tuple[datetime | None, datetime | None]:
|
|
"""返回闭区间起止(本地/业务时区近似用本地 now).all → (None, None)."""
|
|
w = normalize_stats_window(window)
|
|
end = now or datetime.now()
|
|
if w == "all":
|
|
return None, None
|
|
if w == "today":
|
|
start = end.replace(hour=0, minute=0, second=0, microsecond=0)
|
|
return start, end
|
|
days = 7 if w == "7d" else 30
|
|
from datetime import timedelta
|
|
|
|
start = end - timedelta(days=days)
|
|
return start, end
|
|
|
|
|
|
def filter_history_for_stats_window(
|
|
history: list[dict[str, Any]],
|
|
window: str,
|
|
*,
|
|
now: datetime | None = None,
|
|
) -> list[dict[str, Any]]:
|
|
"""窗口内已平仓 + 当前持仓(open)."""
|
|
start, end = stats_window_bounds(window, now=now)
|
|
out: list[dict[str, Any]] = []
|
|
for row in history or []:
|
|
if row.get("status") == "open":
|
|
out.append(row)
|
|
continue
|
|
if start is None:
|
|
out.append(row)
|
|
continue
|
|
ts = _parse_ts(row.get("closed_at")) or _parse_ts(row.get("created_at"))
|
|
if ts is None:
|
|
continue
|
|
if ts < start:
|
|
continue
|
|
if end is not None and ts > end:
|
|
continue
|
|
out.append(row)
|
|
return out
|
|
|
|
|
|
def compute_options_stats_for_window(
|
|
history: list[dict[str, Any]],
|
|
window: str,
|
|
*,
|
|
now: datetime | None = None,
|
|
) -> dict[str, Any]:
|
|
w = normalize_stats_window(window)
|
|
filtered = filter_history_for_stats_window(history, w, now=now)
|
|
stats = compute_options_stats_from_history(filtered)
|
|
start, end = stats_window_bounds(w, now=now)
|
|
stats["window"] = w
|
|
stats["window_start"] = start.strftime("%Y-%m-%d %H:%M:%S") if start else None
|
|
stats["window_end"] = end.strftime("%Y-%m-%d %H:%M:%S") if end else None
|
|
return stats
|
|
|
|
|
|
def _row_get(row: Any, key: str, default: Any = None) -> Any:
|
|
try:
|
|
if hasattr(row, "keys") and key in row.keys():
|
|
return row[key]
|
|
except Exception:
|
|
pass
|
|
if isinstance(row, dict):
|
|
return row.get(key, default)
|
|
return default
|
|
|
|
|
|
def _closed_trade_dicts(closed_rows) -> list[dict[str, Any]]:
|
|
out: list[dict[str, Any]] = []
|
|
for row in closed_rows or []:
|
|
out.append(
|
|
{
|
|
"status": "closed",
|
|
"realized_pnl": _row_get(row, "realized_pnl"),
|
|
"realized_pnl_usdt": _row_get(row, "realized_pnl_usdt"),
|
|
"close_index_px": _row_get(row, "close_index_px"),
|
|
"created_at": _row_get(row, "created_at"),
|
|
"closed_at": _row_get(row, "closed_at"),
|
|
"inst_id": _row_get(row, "inst_id") or "",
|
|
"underlying": _row_get(row, "underlying") or "",
|
|
"premium_ccy": _row_get(row, "premium_ccy") or "",
|
|
}
|
|
)
|
|
return out
|
|
|
|
|
|
_CLOSED_STATS_SELECT = """
|
|
SELECT realized_pnl, realized_pnl_usdt, close_index_px,
|
|
created_at, closed_at, inst_id, underlying, premium_ccy
|
|
FROM options_trades
|
|
WHERE status = 'closed'
|
|
AND (realized_pnl IS NOT NULL OR realized_pnl_usdt IS NOT NULL)
|
|
"""
|
|
|
|
|
|
def freeze_missing_realized_pnl_usdt(conn, ex: Any = None) -> int:
|
|
"""旧已平仓缺冻结 U 时按 close_index/当前指数写一次,之后不再跟盘变.
|
|
|
|
注意:已有 realized_pnl 视为净额直接冻结,不再二次扣 open_fee/close_fee
|
|
(避免对 OKX realizedPnl 或已扣费本地净额重复减费).
|
|
"""
|
|
init_options_tables(conn)
|
|
rows = conn.execute(
|
|
"""
|
|
SELECT id, realized_pnl, realized_pnl_usdt, close_index_px,
|
|
inst_id, underlying, premium_ccy
|
|
FROM options_trades
|
|
WHERE status = 'closed'
|
|
AND realized_pnl_usdt IS NULL
|
|
AND realized_pnl IS NOT NULL
|
|
ORDER BY id DESC
|
|
LIMIT 300
|
|
"""
|
|
).fetchall()
|
|
if not rows:
|
|
return 0
|
|
from lib.options.options_pnl_settle_lib import (
|
|
freeze_pnl_to_usdt,
|
|
resolve_close_index_px,
|
|
)
|
|
|
|
updated = 0
|
|
for row in rows:
|
|
ccy = str(_row_get(row, "premium_ccy") or "").strip().upper()
|
|
idx = _safe_float(_row_get(row, "close_index_px"))
|
|
if idx is None or idx <= 0:
|
|
idx = resolve_close_index_px(
|
|
ex,
|
|
inst_id=str(_row_get(row, "inst_id") or ""),
|
|
underlying=str(_row_get(row, "underlying") or ""),
|
|
)
|
|
pnl = _safe_float(_row_get(row, "realized_pnl"))
|
|
if pnl is None:
|
|
continue
|
|
settled = freeze_pnl_to_usdt(pnl, premium_ccy=ccy or "USDC", index_px=idx)
|
|
usdt = settled.get("realized_pnl_usdt")
|
|
close_idx = settled.get("close_index_px")
|
|
if usdt is None:
|
|
continue
|
|
conn.execute(
|
|
"""
|
|
UPDATE options_trades
|
|
SET realized_pnl_usdt = ?,
|
|
close_index_px = COALESCE(close_index_px, ?)
|
|
WHERE id = ? AND realized_pnl_usdt IS NULL
|
|
""",
|
|
(float(usdt), close_idx, int(row["id"])),
|
|
)
|
|
updated += 1
|
|
return updated
|
|
|
|
|
|
def recompute_closed_realized_pnl_usdt(conn, ex: Any = None) -> int:
|
|
"""强制按「平仓收回×平仓指数 − 开仓成本U − 手续费」重算已平冻结 U.
|
|
|
|
旧逻辑用 (收回−已付)×平仓指数,等于把开仓成本也按平仓价重估,ETH 上涨时会偏亏,
|
|
导致合计盈亏与余额对不上.有 budget_usdt/open_index_px 时优先用真实开仓成本.
|
|
"""
|
|
init_options_tables(conn)
|
|
rows = conn.execute(
|
|
"""
|
|
SELECT id, realized_pnl, realized_pnl_usdt, close_index_px, open_index_px,
|
|
inst_id, underlying, premium_ccy, premium_paid, premium_received,
|
|
open_fee, close_fee, fee_ccy, budget_usdt, close_quote, eth_amount,
|
|
open_quote
|
|
FROM options_trades
|
|
WHERE status = 'closed'
|
|
AND premium_paid IS NOT NULL
|
|
ORDER BY id DESC
|
|
LIMIT 300
|
|
"""
|
|
).fetchall()
|
|
if not rows:
|
|
return 0
|
|
from lib.options.options_pnl_settle_lib import resolve_close_index_px, settle_closed_pnl
|
|
from lib.options.options_pricing_lib import total_premium
|
|
|
|
updated = 0
|
|
for row in rows:
|
|
paid = _safe_float(_row_get(row, "premium_paid"))
|
|
recv = _safe_float(_row_get(row, "premium_received"))
|
|
if recv is None:
|
|
cq = _safe_float(_row_get(row, "close_quote"))
|
|
eth = _safe_float(_row_get(row, "eth_amount"))
|
|
if cq is not None and eth is not None and eth > 0:
|
|
recv = float(total_premium(cq, eth))
|
|
if paid is None or recv is None:
|
|
continue
|
|
ofee = abs(_safe_float(_row_get(row, "open_fee")) or 0.0)
|
|
cfee = abs(_safe_float(_row_get(row, "close_fee")) or 0.0)
|
|
ccy = str(_row_get(row, "premium_ccy") or "").strip().upper() or "ETH"
|
|
close_idx = _safe_float(_row_get(row, "close_index_px"))
|
|
open_idx = _safe_float(_row_get(row, "open_index_px"))
|
|
budget = _safe_float(_row_get(row, "budget_usdt"))
|
|
if close_idx is None or close_idx <= 0:
|
|
close_idx = resolve_close_index_px(
|
|
ex,
|
|
inst_id=str(_row_get(row, "inst_id") or ""),
|
|
underlying=str(_row_get(row, "underlying") or ""),
|
|
)
|
|
if (open_idx is None or open_idx <= 0) and budget and paid > 1e-12:
|
|
open_idx = float(budget) / float(paid)
|
|
settled = settle_closed_pnl(
|
|
premium_paid=paid,
|
|
premium_received=recv,
|
|
open_fee=ofee,
|
|
close_fee=cfee,
|
|
premium_ccy=ccy,
|
|
index_px=close_idx,
|
|
open_index_px=open_idx,
|
|
budget_usdt=budget,
|
|
fee_ccy=str(_row_get(row, "fee_ccy") or ccy),
|
|
)
|
|
new_usdt = settled.get("realized_pnl_usdt")
|
|
if new_usdt is None:
|
|
continue
|
|
old_usdt = _safe_float(_row_get(row, "realized_pnl_usdt"))
|
|
new_pnl = settled.get("realized_pnl")
|
|
# 有变化才写,避免无意义刷库
|
|
if (
|
|
old_usdt is not None
|
|
and abs(float(old_usdt) - float(new_usdt)) < 1e-6
|
|
and _safe_float(_row_get(row, "realized_pnl")) is not None
|
|
and new_pnl is not None
|
|
and abs(float(_row_get(row, "realized_pnl")) - float(new_pnl)) < 1e-10
|
|
):
|
|
continue
|
|
conn.execute(
|
|
"""
|
|
UPDATE options_trades
|
|
SET realized_pnl = COALESCE(?, realized_pnl),
|
|
close_index_px = COALESCE(?, close_index_px),
|
|
open_index_px = COALESCE(open_index_px, ?),
|
|
realized_pnl_usdt = ?
|
|
WHERE id = ?
|
|
""",
|
|
(
|
|
new_pnl,
|
|
settled.get("close_index_px"),
|
|
settled.get("open_index_px"),
|
|
float(new_usdt),
|
|
int(row["id"]),
|
|
),
|
|
)
|
|
updated += 1
|
|
return updated
|
|
|
|
|
|
def repair_double_counted_fee_pnl(conn, ex: Any = None) -> int:
|
|
"""兼容旧名:改为强制按开仓成本重算冻结 U."""
|
|
return recompute_closed_realized_pnl_usdt(conn, ex)
|
|
|
|
|
|
def load_local_options_stats_history(conn, ex: Any = None) -> list[dict[str, Any]]:
|
|
"""本地已平(含冻结 U/手续费净额) + 当前 open,供统计/中控窗口复用."""
|
|
init_options_tables(conn)
|
|
try:
|
|
recompute_closed_realized_pnl_usdt(conn, ex)
|
|
except Exception:
|
|
pass
|
|
try:
|
|
freeze_missing_realized_pnl_usdt(conn, ex)
|
|
except Exception:
|
|
pass
|
|
closed_rows = conn.execute(_CLOSED_STATS_SELECT).fetchall()
|
|
open_rows = conn.execute(
|
|
"SELECT created_at FROM options_trades WHERE status = 'open'"
|
|
).fetchall()
|
|
history = _closed_trade_dicts(closed_rows)
|
|
for row in open_rows:
|
|
history.append({"status": "open", "created_at": _row_get(row, "created_at")})
|
|
return history
|
|
|
|
|
|
def compute_options_stats(get_db, ex: Any = None) -> dict[str, Any]:
|
|
conn = get_db()
|
|
try:
|
|
history = load_local_options_stats_history(conn, ex)
|
|
try:
|
|
conn.commit()
|
|
except Exception:
|
|
pass
|
|
# 优先冻结 U;缺省再按指数折(与 /api/options/stats 一致)
|
|
return compute_options_stats_from_history(history_pnl_to_usdt(history, ex))
|
|
finally:
|
|
conn.close()
|
|
|
|
|
|
def _stats_from_option_trade_rows(closed_rows, open_rows, *, ex: Any = None) -> dict[str, Any]:
|
|
"""兼容旧调用;内部统一折 U 后走 compute_options_stats_from_history."""
|
|
history = _closed_trade_dicts(closed_rows)
|
|
for row in open_rows or []:
|
|
history.append({"status": "open", "created_at": _row_get(row, "created_at")})
|
|
return compute_options_stats_from_history(history_pnl_to_usdt(history, ex))
|
|
|
|
|
|
def header_options_stats_for_window(
|
|
conn,
|
|
list_window: dict[str, Any],
|
|
app_tz,
|
|
ex: Any = None,
|
|
) -> dict[str, Any]:
|
|
"""顶栏总交易/胜率/盈亏比:纯期权(options_trades),全站导航共用,按列表窗过滤.
|
|
|
|
币本位 realized_pnl 为币数量,须折 U 后再算盈亏比,否则 BTC/ETH 混算会严重失真.
|
|
"""
|
|
from lib.common.history_window_lib import utc_window_to_bj_sql_strings
|
|
|
|
init_options_tables(conn)
|
|
start_bj, end_bj = utc_window_to_bj_sql_strings(
|
|
list_window["start_utc"], list_window["end_utc"], app_tz
|
|
)
|
|
try:
|
|
freeze_missing_realized_pnl_usdt(conn, ex)
|
|
except Exception:
|
|
pass
|
|
closed_rows = conn.execute(
|
|
"""
|
|
SELECT realized_pnl, realized_pnl_usdt, close_index_px,
|
|
created_at, closed_at, inst_id, underlying, premium_ccy
|
|
FROM options_trades
|
|
WHERE status = 'closed'
|
|
AND (realized_pnl IS NOT NULL OR realized_pnl_usdt IS NOT NULL)
|
|
AND COALESCE(closed_at, created_at) >= ?
|
|
AND COALESCE(closed_at, created_at) <= ?
|
|
""",
|
|
(start_bj, end_bj),
|
|
).fetchall()
|
|
open_rows = conn.execute(
|
|
"SELECT created_at FROM options_trades WHERE status = 'open'"
|
|
).fetchall()
|
|
history = _closed_trade_dicts(closed_rows)
|
|
for row in open_rows:
|
|
history.append({"status": "open", "created_at": _row_get(row, "created_at")})
|
|
stats = compute_options_stats_from_history(history_pnl_to_usdt(history, ex))
|
|
return {
|
|
"total": int(stats.get("total_closed") or 0),
|
|
"rate": float(stats.get("win_rate") or 0),
|
|
"profit_loss_ratio": stats.get("profit_loss_ratio"),
|
|
}
|