Implement CTP-authoritative trading UI with event-driven state.
Add in-memory order/position books fed by CTP events, split active orders above positions in the UI, tick-triggered local SL/TP, and 30-second full calibration. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -0,0 +1,201 @@
|
||||
# Copyright (c) 2025-2026 马建军. All rights reserved.
|
||||
# 专有软件 — 未经授权禁止复制、传播、转售。
|
||||
# 详见 LICENSE.zh-CN.txt
|
||||
|
||||
"""CTP 权威内存簿:委托、持仓、同步状态(事件增量 + 定期全量校准)。"""
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
import threading
|
||||
import time
|
||||
from typing import Any, Callable, Optional
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
CALIBRATE_INTERVAL_SEC = 30.0
|
||||
|
||||
|
||||
def position_key(exchange: str, symbol: str, direction: str) -> str:
|
||||
"""统一持仓键:exchange|symbol|direction"""
|
||||
ex = (exchange or "").strip().upper()
|
||||
sym = (symbol or "").strip().lower()
|
||||
d = (direction or "long").strip().lower()
|
||||
if ex:
|
||||
return f"{ex}|{sym}|{d}"
|
||||
return f"{sym}|{d}"
|
||||
|
||||
|
||||
def parse_position_key(key: str) -> tuple[str, str, str]:
|
||||
parts = (key or "").split("|")
|
||||
if len(parts) >= 3:
|
||||
return parts[0], parts[1], parts[2]
|
||||
if len(parts) == 2:
|
||||
return "", parts[0], parts[1]
|
||||
return "", (key or "").lower(), "long"
|
||||
|
||||
|
||||
class CtpTradingState:
|
||||
"""进程内 CTP 快照:柜台回报为准,SQLite 仅挂 SL/TP 元数据。"""
|
||||
|
||||
def __init__(self) -> None:
|
||||
self._lock = threading.RLock()
|
||||
self._orders: dict[str, dict[str, Any]] = {}
|
||||
self._positions: dict[str, dict[str, Any]] = {}
|
||||
self._tick_prices: dict[str, float] = {}
|
||||
self._sync_state = "idle"
|
||||
self._last_event_ts: float = 0.0
|
||||
self._last_calibrate_ts: float = 0.0
|
||||
self._on_change: Optional[Callable[[], None]] = None
|
||||
|
||||
def set_change_callback(self, fn: Optional[Callable[[], None]]) -> None:
|
||||
self._on_change = fn
|
||||
|
||||
def _notify(self) -> None:
|
||||
self._last_event_ts = time.time()
|
||||
fn = self._on_change
|
||||
if fn:
|
||||
try:
|
||||
fn()
|
||||
except Exception as exc:
|
||||
logger.debug("trading state change callback: %s", exc)
|
||||
|
||||
@property
|
||||
def sync_state(self) -> str:
|
||||
with self._lock:
|
||||
return self._sync_state
|
||||
|
||||
def sync_label(self) -> str:
|
||||
st = self.sync_state
|
||||
if st == "syncing":
|
||||
return "同步中…"
|
||||
if st == "ready":
|
||||
return "已同步"
|
||||
return ""
|
||||
|
||||
def begin_sync(self) -> None:
|
||||
with self._lock:
|
||||
self._sync_state = "syncing"
|
||||
|
||||
def finish_sync(self) -> None:
|
||||
with self._lock:
|
||||
self._sync_state = "ready"
|
||||
self._last_calibrate_ts = time.time()
|
||||
|
||||
def needs_calibrate(self) -> bool:
|
||||
with self._lock:
|
||||
if self._sync_state == "idle":
|
||||
return True
|
||||
return (time.time() - self._last_calibrate_ts) >= CALIBRATE_INTERVAL_SEC
|
||||
|
||||
def upsert_order(self, row: dict[str, Any], *, notify: bool = True) -> None:
|
||||
oid = str(row.get("order_id") or row.get("vt_order_id") or "").strip()
|
||||
if not oid:
|
||||
return
|
||||
with self._lock:
|
||||
self._orders[oid] = dict(row)
|
||||
if notify:
|
||||
self._notify()
|
||||
|
||||
def remove_order(self, order_id: str, *, notify: bool = True) -> None:
|
||||
oid = (order_id or "").strip()
|
||||
if not oid:
|
||||
return
|
||||
removed = False
|
||||
with self._lock:
|
||||
if oid in self._orders:
|
||||
del self._orders[oid]
|
||||
removed = True
|
||||
else:
|
||||
for k in list(self._orders.keys()):
|
||||
if k == oid or k.endswith(oid) or oid.endswith(k):
|
||||
del self._orders[k]
|
||||
removed = True
|
||||
break
|
||||
if removed and notify:
|
||||
self._notify()
|
||||
|
||||
def upsert_position(self, row: dict[str, Any], *, notify: bool = True) -> None:
|
||||
lots = int(row.get("lots") or 0)
|
||||
ex = row.get("exchange") or ""
|
||||
sym = row.get("symbol") or ""
|
||||
direction = row.get("direction") or "long"
|
||||
pk = position_key(ex, sym, direction)
|
||||
with self._lock:
|
||||
if lots <= 0:
|
||||
self._positions.pop(pk, None)
|
||||
else:
|
||||
row = dict(row)
|
||||
row["position_key"] = pk
|
||||
self._positions[pk] = row
|
||||
if notify:
|
||||
self._notify()
|
||||
|
||||
def remove_position(self, pk: str, *, notify: bool = True) -> None:
|
||||
with self._lock:
|
||||
self._positions.pop(pk, None)
|
||||
if notify:
|
||||
self._notify()
|
||||
|
||||
def set_tick_price(self, exchange: str, symbol: str, price: float) -> None:
|
||||
if not symbol or price <= 0:
|
||||
return
|
||||
key = f"{(exchange or '').upper()}|{symbol.lower()}"
|
||||
with self._lock:
|
||||
self._tick_prices[key] = float(price)
|
||||
|
||||
def get_tick_price(self, exchange: str, symbol: str) -> Optional[float]:
|
||||
key = f"{(exchange or '').upper()}|{symbol.lower()}"
|
||||
with self._lock:
|
||||
return self._tick_prices.get(key)
|
||||
|
||||
def get_active_orders(self) -> list[dict[str, Any]]:
|
||||
with self._lock:
|
||||
return list(self._orders.values())
|
||||
|
||||
def get_positions(self) -> list[dict[str, Any]]:
|
||||
with self._lock:
|
||||
return list(self._positions.values())
|
||||
|
||||
def position_keys(self) -> set[str]:
|
||||
with self._lock:
|
||||
return set(self._positions.keys())
|
||||
|
||||
def clear(self) -> None:
|
||||
with self._lock:
|
||||
self._orders.clear()
|
||||
self._positions.clear()
|
||||
self._tick_prices.clear()
|
||||
self._sync_state = "idle"
|
||||
|
||||
def calibrate_from_lists(
|
||||
self,
|
||||
orders: list[dict[str, Any]],
|
||||
positions: list[dict[str, Any]],
|
||||
) -> None:
|
||||
"""全量校准:以 vnpy 内存为准重建订单/持仓簿。"""
|
||||
self.begin_sync()
|
||||
new_orders: dict[str, dict[str, Any]] = {}
|
||||
for o in orders or []:
|
||||
oid = str(o.get("order_id") or o.get("vt_order_id") or "").strip()
|
||||
if oid:
|
||||
new_orders[oid] = dict(o)
|
||||
new_positions: dict[str, dict[str, Any]] = {}
|
||||
for p in positions or []:
|
||||
lots = int(p.get("lots") or 0)
|
||||
if lots <= 0:
|
||||
continue
|
||||
ex = p.get("exchange") or ""
|
||||
sym = p.get("symbol") or ""
|
||||
direction = p.get("direction") or "long"
|
||||
pk = position_key(ex, sym, direction)
|
||||
row = dict(p)
|
||||
row["position_key"] = pk
|
||||
new_positions[pk] = row
|
||||
with self._lock:
|
||||
self._orders = new_orders
|
||||
self._positions = new_positions
|
||||
self.finish_sync()
|
||||
self._notify()
|
||||
|
||||
|
||||
trading_state = CtpTradingState()
|
||||
+428
-134
@@ -84,6 +84,7 @@ from trading_context import (
|
||||
trading_mode_label,
|
||||
)
|
||||
from ctp_symbol import ths_to_vnpy_symbol
|
||||
from ctp_trading_state import position_key, trading_state
|
||||
from vnpy_bridge import (
|
||||
_ctp_td_lock,
|
||||
ctp_cancel_order,
|
||||
@@ -99,6 +100,7 @@ from vnpy_bridge import (
|
||||
execute_order,
|
||||
get_bridge,
|
||||
set_position_refresh_callback,
|
||||
set_tick_sl_tp_callback,
|
||||
)
|
||||
|
||||
|
||||
@@ -207,11 +209,17 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
|
||||
def _ctp_pos_to_ths_code(p: dict) -> str:
|
||||
sym = (p.get("symbol") or "").strip()
|
||||
ex = (p.get("exchange") or "").strip()
|
||||
if not sym:
|
||||
return ""
|
||||
codes = ths_to_codes(sym)
|
||||
if codes:
|
||||
return codes.get("ths_code") or sym
|
||||
if ex:
|
||||
from vnpy_bridge import CtpBridge
|
||||
ths = CtpBridge._vnpy_sym_to_ths(sym, ex)
|
||||
if ths:
|
||||
return ths
|
||||
return sym
|
||||
|
||||
def _resolve_position_margin(
|
||||
@@ -334,7 +342,8 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
"""CTP 有持仓但本地无监控时,自动补写一条 active 记录供展示。"""
|
||||
if not ctp_status(mode).get("connected"):
|
||||
return
|
||||
for p in _ctp_positions(mode, refresh_if_empty=True):
|
||||
ctp_positions = _ctp_positions(mode, refresh_if_empty=True)
|
||||
for p in ctp_positions:
|
||||
lots = int(p.get("lots") or 0)
|
||||
if lots <= 0:
|
||||
continue
|
||||
@@ -342,7 +351,7 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
ths = _ctp_pos_to_ths_code(p)
|
||||
if not ths:
|
||||
continue
|
||||
existing = _find_active_monitor(conn, ths, direction)
|
||||
existing = _find_or_revive_monitor(conn, ths, direction)
|
||||
if existing:
|
||||
_sync_monitor_from_ctp(
|
||||
conn, int(existing["id"]), ths, direction, mode, ctp=p,
|
||||
@@ -368,6 +377,43 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
"UPDATE trade_order_monitors SET initial_stop_loss=? WHERE id=?",
|
||||
(initial_sl, mid),
|
||||
)
|
||||
if ctp_positions:
|
||||
return
|
||||
|
||||
def _restore_recent_pending_monitors(conn, mode: str) -> None:
|
||||
"""重启或 vnpy 委托缓存丢失时,恢复当日最近一笔可能仍有效的开仓挂单。"""
|
||||
if not ctp_status(mode).get("connected"):
|
||||
return
|
||||
if conn.execute("SELECT 1 FROM trade_order_monitors WHERE status='pending' LIMIT 1").fetchone():
|
||||
return
|
||||
today = datetime.now().strftime("%Y-%m-%d")
|
||||
row = conn.execute(
|
||||
"""SELECT * FROM trade_order_monitors
|
||||
WHERE status='closed' AND monitor_type='manual'
|
||||
AND vt_order_id IS NOT NULL AND vt_order_id != ''
|
||||
AND open_time LIKE ?
|
||||
ORDER BY id DESC LIMIT 1""",
|
||||
(f"{today}%",),
|
||||
).fetchone()
|
||||
if not row:
|
||||
return
|
||||
mon = dict(row)
|
||||
sym = mon.get("symbol") or ""
|
||||
direction = (mon.get("direction") or "long").strip().lower()
|
||||
if _find_active_monitor(conn, sym, direction):
|
||||
return
|
||||
for p in _ctp_positions(mode, refresh_if_empty=False):
|
||||
if int(p.get("lots") or 0) <= 0:
|
||||
continue
|
||||
if (p.get("direction") or "long") != direction:
|
||||
continue
|
||||
if _match_ctp_symbol(p.get("symbol") or "", sym):
|
||||
return
|
||||
conn.execute(
|
||||
"UPDATE trade_order_monitors SET status='pending' WHERE id=?",
|
||||
(mon["id"],),
|
||||
)
|
||||
logger.info("恢复挂单监控 id=%s sym=%s", mon.get("id"), sym)
|
||||
|
||||
def _match_ctp_symbol(ctp_sym: str, ths: str) -> bool:
|
||||
a = (ctp_sym or "").lower()
|
||||
@@ -540,7 +586,7 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
ctp_st = ctp_status(mode)
|
||||
if ctp_st.get("connected"):
|
||||
for o in _ctp_active_orders(mode):
|
||||
sym = o.get("symbol") or ""
|
||||
sym = _ctp_pos_to_ths_code(o) or (o.get("symbol") or "")
|
||||
offset_s = (o.get("offset") or "").upper()
|
||||
kind = "limit"
|
||||
label = "委托挂单"
|
||||
@@ -548,6 +594,7 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
label = "平仓委托"
|
||||
pending.append({
|
||||
"symbol_code": sym,
|
||||
"symbol": _symbol_display_fields(sym).get("symbol_name") or sym,
|
||||
"direction": o.get("direction") or "long",
|
||||
"direction_label": "做多" if o.get("direction") == "long" else "做空",
|
||||
"lots": int(o.get("lots") or 0),
|
||||
@@ -556,6 +603,7 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
"label": label,
|
||||
"source": "ctp",
|
||||
"order_id": o.get("order_id"),
|
||||
"vt_order_id": o.get("vt_order_id") or o.get("order_id"),
|
||||
"can_cancel_order": is_trading_session(),
|
||||
"cancel_allowed": is_trading_session(),
|
||||
**_symbol_display_fields(sym),
|
||||
@@ -568,14 +616,50 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
except Exception:
|
||||
return []
|
||||
|
||||
def _canonical_position_key(symbol: str, direction: str) -> str:
|
||||
def _canonical_position_key(symbol: str, direction: str, exchange: str = "") -> str:
|
||||
sym = (symbol or "").strip()
|
||||
d = (direction or "long").strip().lower()
|
||||
ex = (exchange or "").strip().upper()
|
||||
try:
|
||||
vnpy_sym, _ = ths_to_vnpy_symbol(sym)
|
||||
return f"{vnpy_sym.lower()}:{d}"
|
||||
vnpy_sym, ex2 = ths_to_vnpy_symbol(sym)
|
||||
sym = vnpy_sym
|
||||
if not ex:
|
||||
ex = ex2
|
||||
except Exception:
|
||||
return f"{sym.lower()}:{d}"
|
||||
sym = sym.lower()
|
||||
return position_key(ex, sym, d)
|
||||
|
||||
def _position_key_from_ctp(p: dict) -> str:
|
||||
return position_key(
|
||||
p.get("exchange") or "",
|
||||
p.get("symbol") or "",
|
||||
p.get("direction") or "long",
|
||||
)
|
||||
|
||||
def _monitor_position_key(mon: dict, exchange: str = "") -> str:
|
||||
sym = (mon.get("symbol") or "").strip()
|
||||
d = (mon.get("direction") or "long").strip().lower()
|
||||
ex = (exchange or "").strip().upper()
|
||||
try:
|
||||
vnpy_sym, ex2 = ths_to_vnpy_symbol(sym)
|
||||
sym = vnpy_sym
|
||||
if not ex:
|
||||
ex = ex2
|
||||
except Exception:
|
||||
sym = sym.lower()
|
||||
return position_key(ex, sym, d)
|
||||
|
||||
def _monitors_by_position_key(conn) -> dict[str, dict]:
|
||||
ensure_monitor_order_columns(conn)
|
||||
out: dict[str, dict] = {}
|
||||
for r in conn.execute(
|
||||
"SELECT * FROM trade_order_monitors WHERE status='active' ORDER BY id DESC"
|
||||
).fetchall():
|
||||
mon = dict(r)
|
||||
pk = _monitor_position_key(mon)
|
||||
if pk not in out:
|
||||
out[pk] = mon
|
||||
return out
|
||||
|
||||
def _find_active_monitor(conn, symbol: str, direction: str) -> Optional[dict]:
|
||||
direction = (direction or "long").strip().lower()
|
||||
@@ -589,6 +673,51 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
return row
|
||||
return None
|
||||
|
||||
def _revive_closed_monitor(conn, symbol: str, direction: str) -> Optional[dict]:
|
||||
"""柜台仍有持仓但本地监控被误关时,恢复最近一条同品种记录。"""
|
||||
direction = (direction or "long").strip().lower()
|
||||
for r in conn.execute(
|
||||
"SELECT * FROM trade_order_monitors WHERE status='closed' ORDER BY id DESC LIMIT 40"
|
||||
).fetchall():
|
||||
row = dict(r)
|
||||
if (row.get("direction") or "long") != direction:
|
||||
continue
|
||||
if not _match_ctp_symbol(symbol, row.get("symbol") or ""):
|
||||
continue
|
||||
if int(row.get("lots") or 0) <= 0:
|
||||
continue
|
||||
conn.execute(
|
||||
"UPDATE trade_order_monitors SET status='active' WHERE id=?",
|
||||
(row["id"],),
|
||||
)
|
||||
row["status"] = "active"
|
||||
logger.info(
|
||||
"恢复误关闭监控 id=%s sym=%s dir=%s",
|
||||
row.get("id"), row.get("symbol"), direction,
|
||||
)
|
||||
return row
|
||||
return None
|
||||
|
||||
def _find_or_revive_monitor(conn, symbol: str, direction: str) -> Optional[dict]:
|
||||
active = _find_active_monitor(conn, symbol, direction)
|
||||
if active:
|
||||
return active
|
||||
return _revive_closed_monitor(conn, symbol, direction)
|
||||
|
||||
def _close_all_monitors_for_sym_dir(conn, symbol: str, direction: str) -> None:
|
||||
direction = (direction or "long").strip().lower()
|
||||
for r in conn.execute(
|
||||
"SELECT id, symbol, direction FROM trade_order_monitors "
|
||||
"WHERE status IN ('active', 'pending')"
|
||||
).fetchall():
|
||||
if (r["direction"] or "long") != direction:
|
||||
continue
|
||||
if _match_ctp_symbol(symbol, r["symbol"] or ""):
|
||||
conn.execute(
|
||||
"UPDATE trade_order_monitors SET status='closed' WHERE id=?",
|
||||
(r["id"],),
|
||||
)
|
||||
|
||||
def _close_duplicate_monitors(conn, symbol: str, direction: str, keep_id: int) -> None:
|
||||
direction = (direction or "long").strip().lower()
|
||||
for r in conn.execute(
|
||||
@@ -950,18 +1079,15 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
"source": "monitor",
|
||||
"monitor_id": mon["id"] if mon else None,
|
||||
})
|
||||
row_key = _canonical_position_key(sym, direction)
|
||||
ctp_st = ctp_status(mode)
|
||||
sync_pending = (
|
||||
mon is not None
|
||||
and ctp is None
|
||||
and bool(ctp_st.get("connected"))
|
||||
row_key = _canonical_position_key(
|
||||
sym, direction, (ctp or {}).get("exchange") or "",
|
||||
)
|
||||
return {
|
||||
"key": row_key,
|
||||
"source": "ctp" if ctp else "local",
|
||||
"position_key": row_key,
|
||||
"source": "ctp",
|
||||
"source_label": source_label,
|
||||
"sync_pending": sync_pending,
|
||||
"sync_pending": False,
|
||||
"monitor_id": mon["id"] if mon else None,
|
||||
"symbol_code": sym,
|
||||
**_symbol_display_fields(sym),
|
||||
@@ -1073,6 +1199,69 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
"trailing_r_locked": int(mon.get("trailing_r_locked") or 0),
|
||||
}
|
||||
|
||||
def _compose_ctp_open_order_row(
|
||||
o: dict,
|
||||
*,
|
||||
mode: str,
|
||||
capital: float,
|
||||
now_iso: str,
|
||||
) -> Optional[dict]:
|
||||
offset_u = (o.get("offset") or "").upper()
|
||||
if offset_u and "OPEN" not in offset_u:
|
||||
return None
|
||||
sym = _ctp_pos_to_ths_code(o) or (o.get("symbol") or "").strip()
|
||||
direction = (o.get("direction") or "long").strip().lower()
|
||||
lots = int(o.get("lots") or 0)
|
||||
if not sym or lots <= 0:
|
||||
return None
|
||||
order_price = float(o.get("price") or 0)
|
||||
pos_metrics = calc_position_metrics(
|
||||
direction, order_price, order_price, order_price, lots, order_price, capital, sym,
|
||||
)
|
||||
timeout_sec = get_pending_order_timeout_sec(get_setting)
|
||||
return {
|
||||
"key": f"{_canonical_position_key(sym, direction)}:pending:ctp:{o.get('order_id') or ''}",
|
||||
"order_state": "pending",
|
||||
"source": "ctp",
|
||||
"source_label": "委托挂单",
|
||||
"sync_pending": True,
|
||||
"monitor_id": None,
|
||||
"order_id": o.get("order_id"),
|
||||
"vt_order_id": o.get("vt_order_id") or o.get("order_id"),
|
||||
"symbol_code": sym,
|
||||
**_symbol_display_fields(sym),
|
||||
"direction": direction,
|
||||
"direction_label": "做多" if direction == "long" else "做空",
|
||||
"lots": lots,
|
||||
"entry_price": order_price,
|
||||
"order_price": order_price,
|
||||
"stop_loss": None,
|
||||
"take_profit": None,
|
||||
"open_time": now_iso,
|
||||
"holding_duration": None,
|
||||
"mark_price": order_price,
|
||||
"current_price": order_price,
|
||||
"margin": pos_metrics.get("margin"),
|
||||
"margin_source": "estimate",
|
||||
"position_pct": pos_metrics.get("position_pct"),
|
||||
"float_pnl": None,
|
||||
"est_fee": None,
|
||||
"can_close": False,
|
||||
"close_allowed": False,
|
||||
"can_cancel_order": is_trading_session(),
|
||||
"cancel_allowed": is_trading_session(),
|
||||
"pending_timeout_sec": timeout_sec,
|
||||
"pending_timeout_min": max(1, timeout_sec // 60),
|
||||
"sl_order_active": False,
|
||||
"tp_order_active": False,
|
||||
"sl_monitoring": False,
|
||||
"tp_monitoring": False,
|
||||
"can_place_orders": False,
|
||||
"pending_orders": [],
|
||||
"trailing_be": False,
|
||||
"trailing_r_locked": 0,
|
||||
}
|
||||
|
||||
def _reconcile_pending(conn, mode: str, *, capital: float = 0.0) -> None:
|
||||
reconcile_pending_orders(
|
||||
conn,
|
||||
@@ -1084,7 +1273,123 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
timeout_sec=get_pending_order_timeout_sec(get_setting),
|
||||
)
|
||||
|
||||
def _build_active_orders(
|
||||
conn,
|
||||
*,
|
||||
mode: str,
|
||||
capital: float,
|
||||
now_iso: str,
|
||||
) -> list[dict]:
|
||||
"""当前委托:以 CTP 柜台为准,本地 pending 开仓单合并展示。"""
|
||||
orders: list[dict] = []
|
||||
seen_keys: set[str] = set()
|
||||
|
||||
if ctp_status(mode).get("connected"):
|
||||
ctp_orders = trading_state.get_active_orders()
|
||||
if not ctp_orders:
|
||||
ctp_orders = _ctp_active_orders(mode)
|
||||
for o in ctp_orders:
|
||||
try:
|
||||
row = _compose_ctp_open_order_row(
|
||||
o, mode=mode, capital=capital, now_iso=now_iso,
|
||||
)
|
||||
if not row:
|
||||
row = _compose_ctp_order_row_any(
|
||||
o, mode=mode, capital=capital, now_iso=now_iso,
|
||||
)
|
||||
if row:
|
||||
orders.append(row)
|
||||
seen_keys.add(row.get("key") or "")
|
||||
except Exception as exc:
|
||||
logger.warning("compose ctp order row failed: %s", exc)
|
||||
|
||||
for r in conn.execute(
|
||||
"SELECT * FROM trade_order_monitors WHERE status='pending' ORDER BY id DESC"
|
||||
).fetchall():
|
||||
mon = dict(r)
|
||||
try:
|
||||
prow = _compose_pending_row(
|
||||
mon, mode=mode, capital=capital, now_iso=now_iso,
|
||||
)
|
||||
if prow and prow.get("key") not in seen_keys:
|
||||
pk = f"{prow.get('symbol_code') or ''}:{prow.get('direction') or ''}"
|
||||
dup = any(
|
||||
(x.get("symbol_code") or "") + ":" + (x.get("direction") or "") == pk
|
||||
and x.get("order_state") == "pending"
|
||||
for x in orders
|
||||
)
|
||||
if not dup:
|
||||
orders.append(prow)
|
||||
except Exception as exc:
|
||||
logger.warning("compose pending order row failed: %s", exc)
|
||||
return orders
|
||||
|
||||
def _compose_ctp_order_row_any(
|
||||
o: dict,
|
||||
*,
|
||||
mode: str,
|
||||
capital: float,
|
||||
now_iso: str,
|
||||
) -> Optional[dict]:
|
||||
"""CTP 任意未成交委托(含平仓)。"""
|
||||
sym = _ctp_pos_to_ths_code(o) or (o.get("symbol") or "").strip()
|
||||
direction = (o.get("direction") or "long").strip().lower()
|
||||
lots = int(o.get("lots") or 0)
|
||||
if not sym or lots <= 0:
|
||||
return None
|
||||
offset_u = (o.get("offset") or "").upper()
|
||||
is_open = not offset_u or "OPEN" in offset_u
|
||||
order_price = float(o.get("price") or 0)
|
||||
pos_metrics = calc_position_metrics(
|
||||
direction, order_price, order_price, order_price, lots, order_price, capital, sym,
|
||||
)
|
||||
label = "开仓委托" if is_open else "平仓委托"
|
||||
timeout_sec = get_pending_order_timeout_sec(get_setting)
|
||||
ex = o.get("exchange") or ""
|
||||
pk = _canonical_position_key(sym, direction, ex)
|
||||
return {
|
||||
"key": f"{pk}:order:{o.get('order_id') or ''}",
|
||||
"order_state": "pending",
|
||||
"source": "ctp",
|
||||
"source_label": label,
|
||||
"sync_pending": False,
|
||||
"monitor_id": None,
|
||||
"order_id": o.get("order_id"),
|
||||
"vt_order_id": o.get("vt_order_id") or o.get("order_id"),
|
||||
"symbol_code": sym,
|
||||
**_symbol_display_fields(sym),
|
||||
"direction": direction,
|
||||
"direction_label": "做多" if direction == "long" else "做空",
|
||||
"lots": lots,
|
||||
"entry_price": order_price,
|
||||
"order_price": order_price,
|
||||
"stop_loss": None,
|
||||
"take_profit": None,
|
||||
"open_time": now_iso,
|
||||
"mark_price": order_price,
|
||||
"current_price": order_price,
|
||||
"margin": pos_metrics.get("margin"),
|
||||
"margin_source": "estimate",
|
||||
"position_pct": pos_metrics.get("position_pct"),
|
||||
"float_pnl": None,
|
||||
"can_close": False,
|
||||
"close_allowed": False,
|
||||
"can_cancel_order": is_trading_session(),
|
||||
"cancel_allowed": is_trading_session(),
|
||||
"pending_timeout_sec": timeout_sec if is_open else None,
|
||||
"pending_timeout_min": max(1, timeout_sec // 60) if is_open else None,
|
||||
"sl_order_active": False,
|
||||
"tp_order_active": False,
|
||||
"sl_monitoring": False,
|
||||
"tp_monitoring": False,
|
||||
"can_place_orders": False,
|
||||
"pending_orders": [],
|
||||
"trailing_be": False,
|
||||
"trailing_r_locked": 0,
|
||||
}
|
||||
|
||||
def _build_trading_live_rows(conn, *, fast: bool = False) -> list[dict]:
|
||||
"""当前持仓:以 CTP 为准,SQLite 仅叠加 SL/TP 元数据。"""
|
||||
from zoneinfo import ZoneInfo
|
||||
tz = ZoneInfo("Asia/Shanghai")
|
||||
now_iso = datetime.now(tz).strftime("%Y-%m-%dT%H:%M")
|
||||
@@ -1092,127 +1397,73 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
capital = _capital(conn)
|
||||
ensure_monitor_order_columns(conn)
|
||||
|
||||
monitors_raw = [
|
||||
dict(r) for r in conn.execute(
|
||||
"SELECT * FROM trade_order_monitors WHERE status='active' ORDER BY id DESC"
|
||||
).fetchall()
|
||||
]
|
||||
monitor_by_key: dict[str, dict] = {}
|
||||
for mon in monitors_raw:
|
||||
key = _canonical_position_key(mon.get("symbol") or "", mon.get("direction") or "long")
|
||||
if key not in monitor_by_key:
|
||||
monitor_by_key[key] = mon
|
||||
|
||||
ctp_list: list[dict] = (
|
||||
_ctp_positions(mode, refresh_if_empty=not fast, refresh_margin=not fast)
|
||||
if ctp_status(mode).get("connected") else []
|
||||
)
|
||||
ctp_by_key: dict[str, dict] = {}
|
||||
for p in ctp_list:
|
||||
if int(p.get("lots") or 0) <= 0:
|
||||
continue
|
||||
key = _canonical_position_key(p.get("symbol") or "", p.get("direction") or "long")
|
||||
ctp_by_key[key] = p
|
||||
monitor_by_pk = _monitors_by_position_key(conn)
|
||||
ctp_list: list[dict] = []
|
||||
if ctp_status(mode).get("connected"):
|
||||
ctp_list = trading_state.get_positions()
|
||||
if not ctp_list:
|
||||
ctp_list = _ctp_positions(
|
||||
mode, refresh_if_empty=not fast, refresh_margin=not fast,
|
||||
)
|
||||
|
||||
rows: list[dict] = []
|
||||
used_ctp_keys: set[str] = set()
|
||||
|
||||
for key, mon in monitor_by_key.items():
|
||||
ctp = ctp_by_key.get(key)
|
||||
if not ctp:
|
||||
for ck, cp in ctp_by_key.items():
|
||||
if ck in used_ctp_keys:
|
||||
for p in ctp_list:
|
||||
lots = int(p.get("lots") or 0)
|
||||
if lots <= 0:
|
||||
continue
|
||||
pk = p.get("position_key") or _position_key_from_ctp(p)
|
||||
mon = monitor_by_pk.get(pk)
|
||||
if not mon:
|
||||
for mk, mv in monitor_by_pk.items():
|
||||
if (mv.get("direction") or "long") != (p.get("direction") or "long"):
|
||||
continue
|
||||
if (cp.get("direction") or "long") != (mon.get("direction") or "long"):
|
||||
continue
|
||||
if _match_ctp_symbol(cp.get("symbol") or "", mon.get("symbol") or ""):
|
||||
ctp = cp
|
||||
used_ctp_keys.add(ck)
|
||||
if _match_ctp_symbol(p.get("symbol") or "", mv.get("symbol") or ""):
|
||||
mon = mv
|
||||
break
|
||||
elif key in ctp_by_key:
|
||||
used_ctp_keys.add(key)
|
||||
if ctp and mon and not fast:
|
||||
ths = _ctp_pos_to_ths_code(p) or (p.get("symbol") or "")
|
||||
if mon and not fast:
|
||||
_sync_monitor_from_ctp(
|
||||
conn, int(mon["id"]), mon.get("symbol") or "",
|
||||
mon.get("direction") or "long", mode, ctp=ctp,
|
||||
capital=capital,
|
||||
conn, int(mon["id"]), mon.get("symbol") or ths,
|
||||
mon.get("direction") or p.get("direction") or "long",
|
||||
mode, ctp=p, capital=capital,
|
||||
)
|
||||
mon = _find_active_monitor(conn, mon.get("symbol") or "", mon.get("direction") or "long") or mon
|
||||
mon = _find_active_monitor(
|
||||
conn, mon.get("symbol") or ths, mon.get("direction") or "long",
|
||||
) or mon
|
||||
try:
|
||||
row = _compose_position_row(
|
||||
conn, mon=mon, ctp=ctp, mode=mode, capital=capital, now_iso=now_iso,
|
||||
fast=fast,
|
||||
conn, mon=mon, ctp=p, mode=mode, capital=capital,
|
||||
now_iso=now_iso, fast=fast,
|
||||
)
|
||||
if row:
|
||||
rows.append(row)
|
||||
except Exception as exc:
|
||||
logger.warning("compose monitor row failed: %s", exc)
|
||||
|
||||
for key, ctp in ctp_by_key.items():
|
||||
if key in used_ctp_keys:
|
||||
continue
|
||||
matched = False
|
||||
for uk in used_ctp_keys:
|
||||
if uk == key:
|
||||
matched = True
|
||||
break
|
||||
if matched:
|
||||
continue
|
||||
for existing in rows:
|
||||
if _match_ctp_symbol(
|
||||
ctp.get("symbol") or "", existing.get("symbol_code") or "",
|
||||
) and (ctp.get("direction") or "long") == (existing.get("direction") or "long"):
|
||||
matched = True
|
||||
break
|
||||
if matched:
|
||||
continue
|
||||
mon = _find_active_monitor(
|
||||
conn, ctp.get("symbol") or "", ctp.get("direction") or "long",
|
||||
)
|
||||
try:
|
||||
row = _compose_position_row(
|
||||
conn, mon=mon, ctp=ctp, mode=mode, capital=capital, now_iso=now_iso,
|
||||
fast=fast,
|
||||
)
|
||||
if row:
|
||||
rows.append(row)
|
||||
except Exception as exc:
|
||||
logger.warning("compose ctp row failed: %s", exc)
|
||||
logger.warning("compose ctp position row failed: %s", exc)
|
||||
|
||||
seen: set[str] = set()
|
||||
deduped: list[dict] = []
|
||||
for row in rows:
|
||||
rk = row.get("key") or f"{row.get('symbol_code')}:{row.get('direction')}"
|
||||
rk = row.get("key") or row.get("position_key") or ""
|
||||
if rk in seen:
|
||||
continue
|
||||
seen.add(rk)
|
||||
deduped.append(row)
|
||||
|
||||
pending_raw = [
|
||||
dict(r) for r in conn.execute(
|
||||
"SELECT * FROM trade_order_monitors WHERE status='pending' ORDER BY id DESC"
|
||||
).fetchall()
|
||||
]
|
||||
for mon in pending_raw:
|
||||
try:
|
||||
prow = _compose_pending_row(
|
||||
mon, mode=mode, capital=capital, now_iso=now_iso,
|
||||
)
|
||||
if prow:
|
||||
deduped.insert(0, prow)
|
||||
except Exception as exc:
|
||||
logger.warning("compose pending row failed: %s", exc)
|
||||
return deduped
|
||||
|
||||
def _build_trading_live_payload(conn, *, fast: bool = False) -> dict:
|
||||
from zoneinfo import ZoneInfo
|
||||
tz = ZoneInfo("Asia/Shanghai")
|
||||
now_iso = datetime.now(tz).strftime("%Y-%m-%dT%H:%M")
|
||||
mode = get_trading_mode(get_setting)
|
||||
ctp_st = ctp_status(mode)
|
||||
capital = _capital(conn)
|
||||
if not fast and ctp_st.get("connected"):
|
||||
if ctp_st.get("connected") and not fast:
|
||||
_reconcile_pending(conn, mode, capital=capital)
|
||||
if ctp_st.get("connected"):
|
||||
_ensure_monitors_from_ctp(conn, mode)
|
||||
_sync_trade_monitors_with_ctp(conn, mode)
|
||||
rows = _build_trading_live_rows(conn, fast=fast)
|
||||
active_orders = _build_active_orders(
|
||||
conn, mode=mode, capital=capital, now_iso=now_iso,
|
||||
)
|
||||
rows = _apply_account_margin_to_rows(rows, mode, capital)
|
||||
_persist_ctp_snapshot_to_monitors(conn, rows, mode)
|
||||
pending_orders = _build_pending_orders(conn, mode)
|
||||
@@ -1220,6 +1471,7 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
return {
|
||||
"ok": True,
|
||||
"rows": rows,
|
||||
"active_orders": active_orders,
|
||||
"pending_orders": pending_orders,
|
||||
"capital": capital,
|
||||
"ctp_status": ctp_st,
|
||||
@@ -1227,16 +1479,26 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
"risk_status": risk,
|
||||
"trading_session": is_trading_session(),
|
||||
"pending_order_timeout_min": get_pending_order_timeout_min(get_setting),
|
||||
"sync_state": trading_state.sync_state,
|
||||
"sync_label": trading_state.sync_label(),
|
||||
}
|
||||
|
||||
def _refresh_trading_live_snapshot(*, fast: bool = False) -> dict:
|
||||
mode = get_trading_mode(get_setting)
|
||||
if not fast and ctp_status(mode).get("connected"):
|
||||
try:
|
||||
with _ctp_td_lock:
|
||||
get_bridge().refresh_positions()
|
||||
except Exception as exc:
|
||||
logger.debug("refresh positions before snapshot: %s", exc)
|
||||
if ctp_status(mode).get("connected"):
|
||||
if not fast:
|
||||
try:
|
||||
with _ctp_td_lock:
|
||||
get_bridge().calibrate_trading_state()
|
||||
except Exception as exc:
|
||||
logger.debug("refresh calibrate: %s", exc)
|
||||
for p in trading_state.get_positions() or _ctp_positions(mode, refresh_if_empty=False):
|
||||
ths = _ctp_pos_to_ths_code(p)
|
||||
if ths:
|
||||
try:
|
||||
get_bridge().subscribe_symbol(ths)
|
||||
except Exception:
|
||||
pass
|
||||
conn = get_db()
|
||||
try:
|
||||
init_strategy_tables(conn)
|
||||
@@ -1246,7 +1508,7 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
def _push_position_snapshot_async(*, fast: bool = False) -> None:
|
||||
def _push_position_snapshot_async(*, fast: bool = True) -> None:
|
||||
def _run() -> None:
|
||||
try:
|
||||
payload = _refresh_trading_live_snapshot(fast=fast)
|
||||
@@ -1256,15 +1518,43 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
|
||||
threading.Thread(target=_run, daemon=True).start()
|
||||
|
||||
def _on_tick_sl_tp(exchange: str, symbol: str, price: float) -> None:
|
||||
from sl_tp_guard import check_sl_tp_on_tick
|
||||
from db_conn import DB_PATH, connect_db
|
||||
|
||||
mode = get_trading_mode(get_setting)
|
||||
if not ctp_status(mode).get("connected"):
|
||||
return
|
||||
conn = connect_db(DB_PATH)
|
||||
try:
|
||||
_init_tables(conn)
|
||||
capital = _capital(conn)
|
||||
n = check_sl_tp_on_tick(
|
||||
conn, mode, exchange, symbol, price,
|
||||
capital=capital, notify_fn=send_wechat_msg,
|
||||
be_tick_mult=get_trailing_be_tick_buffer(get_setting),
|
||||
)
|
||||
if n:
|
||||
conn.commit()
|
||||
_push_position_snapshot_async(fast=True)
|
||||
except Exception as exc:
|
||||
logger.debug("tick sl/tp: %s", exc)
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
def _bootstrap_trading_runtime() -> None:
|
||||
"""进程启动:立刻读库展示持仓,并异步连 CTP。"""
|
||||
"""进程启动:读 CTP 快照推送,事件驱动增量 + 定期全量校准。"""
|
||||
set_position_refresh_callback(
|
||||
lambda: _push_position_snapshot_async(fast=False)
|
||||
lambda: _push_position_snapshot_async(fast=True)
|
||||
)
|
||||
set_tick_sl_tp_callback(_on_tick_sl_tp)
|
||||
|
||||
def _warm() -> None:
|
||||
try:
|
||||
payload = _refresh_trading_live_snapshot(fast=True)
|
||||
mode = get_trading_mode(get_setting)
|
||||
if ctp_status(mode).get("connected"):
|
||||
get_bridge().calibrate_trading_state()
|
||||
payload = _refresh_trading_live_snapshot(fast=False)
|
||||
position_hub.set_snapshot(payload)
|
||||
position_hub.broadcast("positions", payload)
|
||||
except Exception as exc:
|
||||
@@ -1346,6 +1636,7 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
try:
|
||||
init_strategy_tables(conn)
|
||||
payload = _build_trading_live_payload(conn, fast=True)
|
||||
conn.commit()
|
||||
position_hub.set_snapshot(payload)
|
||||
return jsonify(payload)
|
||||
finally:
|
||||
@@ -1656,12 +1947,7 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
symbol_name=(mon.get("symbol_name") or "") if mon else "",
|
||||
market_code=(mon.get("market_code") or "") if mon else "",
|
||||
)
|
||||
if mid:
|
||||
_close_duplicate_monitors(conn, sym, direction, mid)
|
||||
conn.execute(
|
||||
"UPDATE trade_order_monitors SET status='closed' WHERE id=?",
|
||||
(mid,),
|
||||
)
|
||||
_close_all_monitors_for_sym_dir(conn, sym, direction)
|
||||
conn.commit()
|
||||
try:
|
||||
from ctp_trade_sync import sync_trade_logs_from_ctp
|
||||
@@ -2449,26 +2735,34 @@ def install_trading(app, *, login_required, require_nav, get_db, get_setting, se
|
||||
interval=1,
|
||||
)
|
||||
_pos_refresh_tick = {"n": 0}
|
||||
_last_full_calibrate = {"ts": 0.0}
|
||||
|
||||
def _position_worker_refresh() -> dict:
|
||||
import time as _time
|
||||
from ctp_trading_state import CALIBRATE_INTERVAL_SEC
|
||||
|
||||
_pos_refresh_tick["n"] += 1
|
||||
mode = get_trading_mode(get_setting)
|
||||
connected = bool(ctp_status(mode).get("connected"))
|
||||
# 已连接时每 2 秒完整对账;未连接时每 5 秒轻量刷新(禁止 query_position)
|
||||
if connected:
|
||||
use_fast = _pos_refresh_tick["n"] % 2 != 0
|
||||
else:
|
||||
use_fast = _pos_refresh_tick["n"] % 5 != 0
|
||||
payload = _refresh_trading_live_snapshot(fast=use_fast)
|
||||
if connected and use_fast and any(
|
||||
r.get("sync_pending") for r in (payload.get("rows") or [])
|
||||
):
|
||||
now = _time.time()
|
||||
need_full = (
|
||||
connected
|
||||
and (
|
||||
trading_state.needs_calibrate()
|
||||
or (now - _last_full_calibrate["ts"]) >= CALIBRATE_INTERVAL_SEC
|
||||
)
|
||||
)
|
||||
if need_full:
|
||||
_last_full_calibrate["ts"] = now
|
||||
payload = _refresh_trading_live_snapshot(fast=False)
|
||||
else:
|
||||
payload = _refresh_trading_live_snapshot(fast=True)
|
||||
return payload
|
||||
|
||||
start_position_worker(
|
||||
refresh_fn=_position_worker_refresh,
|
||||
interval=1,
|
||||
interval=2,
|
||||
idle_interval=5,
|
||||
)
|
||||
_bootstrap_trading_runtime()
|
||||
start_ctp_fee_worker(
|
||||
|
||||
+75
-13
@@ -80,6 +80,37 @@ def _find_ctp_position(positions: list[dict], sym: str, direction: str) -> Optio
|
||||
return None
|
||||
|
||||
|
||||
def _vt_order_in_active(vt_oid: str, active_orders: dict[str, dict]) -> bool:
|
||||
oid = (vt_oid or "").strip()
|
||||
if not oid:
|
||||
return False
|
||||
if oid in active_orders:
|
||||
return True
|
||||
tail = oid.rsplit("_", 1)[-1]
|
||||
for key in active_orders:
|
||||
if key == oid or key.endswith(tail) or oid.endswith(key):
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def _symbol_open_order_active(
|
||||
orders: list[dict],
|
||||
sym: str,
|
||||
direction: str,
|
||||
match_fn: Callable[[str, str], bool],
|
||||
) -> Optional[dict]:
|
||||
direction = (direction or "long").strip().lower()
|
||||
for o in orders or []:
|
||||
offset_u = (o.get("offset") or "").upper()
|
||||
if offset_u and "OPEN" not in offset_u:
|
||||
continue
|
||||
if (o.get("direction") or "long") != direction:
|
||||
continue
|
||||
if match_fn(o.get("symbol") or "", sym):
|
||||
return o
|
||||
return None
|
||||
|
||||
|
||||
def reconcile_pending_orders(
|
||||
conn,
|
||||
mode: str,
|
||||
@@ -103,13 +134,15 @@ def reconcile_pending_orders(
|
||||
else []
|
||||
)
|
||||
try:
|
||||
active_orders = {
|
||||
str(o.get("order_id") or ""): o
|
||||
for o in ctp_list_active_orders(mode)
|
||||
if o.get("order_id")
|
||||
}
|
||||
active_order_list = ctp_list_active_orders(mode)
|
||||
active_orders = {}
|
||||
for o in active_order_list:
|
||||
for key in (o.get("order_id"), o.get("vt_order_id")):
|
||||
if key:
|
||||
active_orders[str(key)] = o
|
||||
except Exception as exc:
|
||||
logger.debug("list active orders: %s", exc)
|
||||
active_order_list = []
|
||||
active_orders = {}
|
||||
|
||||
rows = conn.execute(
|
||||
@@ -137,7 +170,7 @@ def reconcile_pending_orders(
|
||||
stats["promoted"] += 1
|
||||
continue
|
||||
|
||||
if vt_oid and vt_oid in active_orders:
|
||||
if vt_oid and _vt_order_in_active(vt_oid, active_orders):
|
||||
if age >= limit_sec and is_trading_session():
|
||||
if ctp_cancel_order(mode, vt_oid):
|
||||
conn.execute(
|
||||
@@ -149,13 +182,42 @@ def reconcile_pending_orders(
|
||||
logger.warning("pending auto-cancel failed monitor=%s order=%s", mid, vt_oid)
|
||||
continue
|
||||
|
||||
# 委托已不在活跃列表且无持仓:拒单/撤单/过期
|
||||
if age >= 8:
|
||||
conn.execute(
|
||||
"UPDATE trade_order_monitors SET status='closed' WHERE id=?",
|
||||
(mid,),
|
||||
)
|
||||
stats["closed"] += 1
|
||||
live_open = _symbol_open_order_active(active_order_list, sym, direction, match)
|
||||
if live_open or (vt_oid and age < limit_sec):
|
||||
if live_open and age >= limit_sec and is_trading_session():
|
||||
cancel_oid = (
|
||||
vt_oid
|
||||
or live_open.get("vt_order_id")
|
||||
or live_open.get("order_id")
|
||||
or ""
|
||||
)
|
||||
if cancel_oid and ctp_cancel_order(mode, cancel_oid):
|
||||
conn.execute(
|
||||
"UPDATE trade_order_monitors SET status='closed' WHERE id=?",
|
||||
(mid,),
|
||||
)
|
||||
stats["cancelled"] += 1
|
||||
continue
|
||||
|
||||
if age >= limit_sec:
|
||||
if vt_oid and is_trading_session():
|
||||
if ctp_cancel_order(mode, vt_oid):
|
||||
conn.execute(
|
||||
"UPDATE trade_order_monitors SET status='closed' WHERE id=?",
|
||||
(mid,),
|
||||
)
|
||||
stats["cancelled"] += 1
|
||||
else:
|
||||
logger.info(
|
||||
"pending monitor=%s order=%s kept (cancel not confirmed)",
|
||||
mid, vt_oid,
|
||||
)
|
||||
elif not vt_oid:
|
||||
conn.execute(
|
||||
"UPDATE trade_order_monitors SET status='closed' WHERE id=?",
|
||||
(mid,),
|
||||
)
|
||||
stats["closed"] += 1
|
||||
|
||||
if any(stats.values()):
|
||||
conn.commit()
|
||||
|
||||
+96
-8
@@ -25,6 +25,7 @@ from vnpy_bridge import (
|
||||
ctp_list_active_orders,
|
||||
ctp_list_positions,
|
||||
ctp_status,
|
||||
ctp_account_margin_used,
|
||||
execute_order,
|
||||
get_bridge,
|
||||
)
|
||||
@@ -564,6 +565,13 @@ def reconcile_monitors_without_position(conn, mode: str, *, grace_sec: int = 120
|
||||
"""持仓已平时:关闭监控并撤销残留止盈止损挂单(新开仓 grace_sec 内不清理)。"""
|
||||
if not ctp_status(mode).get("connected"):
|
||||
return 0
|
||||
try:
|
||||
bridge = get_bridge()
|
||||
since_connect = time.time() - float(getattr(bridge, "_last_connect_ok_ts", 0) or 0)
|
||||
if since_connect < 90:
|
||||
return 0
|
||||
except Exception:
|
||||
pass
|
||||
positions = ctp_list_positions(mode, refresh_if_empty=False, refresh_margin=False)
|
||||
position_keys: set[tuple[str, str]] = set()
|
||||
for p in positions:
|
||||
@@ -573,14 +581,9 @@ def reconcile_monitors_without_position(conn, mode: str, *, grace_sec: int = 120
|
||||
direction = p.get("direction") or "long"
|
||||
position_keys.add((sym, direction))
|
||||
|
||||
if not position_keys:
|
||||
try:
|
||||
acc = get_bridge().get_account()
|
||||
margin_used = float(acc.get("balance") or 0) - float(acc.get("available") or 0)
|
||||
if margin_used > 500:
|
||||
return 0
|
||||
except Exception:
|
||||
return 0
|
||||
margin_used = ctp_account_margin_used(mode) or 0.0
|
||||
if not position_keys and margin_used > 300:
|
||||
return 0
|
||||
|
||||
now_ts = time.time()
|
||||
|
||||
@@ -669,6 +672,91 @@ def _execute_local_close(
|
||||
logger.debug("SL/TP notify failed: %s", exc)
|
||||
|
||||
|
||||
def check_sl_tp_on_tick(
|
||||
conn,
|
||||
mode: str,
|
||||
exchange: str,
|
||||
symbol: str,
|
||||
mark: float,
|
||||
*,
|
||||
capital: float = 0.0,
|
||||
notify_fn: Callable[[str], None] | None = None,
|
||||
be_tick_mult: int = 2,
|
||||
) -> int:
|
||||
"""EVENT_TICK 触发:仅检查与 tick 品种匹配的 active 监控。"""
|
||||
ensure_monitor_order_columns(conn)
|
||||
if not ctp_status(mode).get("connected") or not is_trading_session():
|
||||
return 0
|
||||
if mark <= 0:
|
||||
return 0
|
||||
sym_l = (symbol or "").lower()
|
||||
ex_u = (exchange or "").upper()
|
||||
closed = 0
|
||||
rows = [dict(r) for r in conn.execute(
|
||||
"SELECT * FROM trade_order_monitors WHERE status='active'"
|
||||
).fetchall()]
|
||||
for mon in rows:
|
||||
mid = int(mon.get("id") or 0)
|
||||
ms = (mon.get("symbol") or "").strip()
|
||||
direction = (mon.get("direction") or "long").strip().lower()
|
||||
try:
|
||||
vnpy_sym, ex2 = ths_to_vnpy_symbol(ms)
|
||||
if sym_l != vnpy_sym.lower():
|
||||
continue
|
||||
if ex_u and ex2 and ex_u != ex2.upper():
|
||||
continue
|
||||
except Exception:
|
||||
if sym_l != ms.lower():
|
||||
continue
|
||||
|
||||
sl = mon.get("stop_loss")
|
||||
tp = mon.get("take_profit")
|
||||
try:
|
||||
sl_f = float(sl) if sl is not None else None
|
||||
tp_f = float(tp) if tp is not None else None
|
||||
except (TypeError, ValueError):
|
||||
sl_f, tp_f = None, None
|
||||
if sl_f is None and tp_f is None:
|
||||
continue
|
||||
|
||||
positions = ctp_list_positions(mode)
|
||||
if not _find_position(positions, ms, direction):
|
||||
continue
|
||||
|
||||
tick = _tick_size(ms)
|
||||
if mon.get("trailing_be"):
|
||||
mon = _update_trailing_stop_loss(conn, mon, mark, be_tick_mult=be_tick_mult)
|
||||
try:
|
||||
sl_f = float(mon["stop_loss"]) if mon.get("stop_loss") is not None else sl_f
|
||||
except (TypeError, ValueError):
|
||||
pass
|
||||
|
||||
reason = None
|
||||
if tp_f is not None and _tp_triggered(direction, tp_f, mark, tick):
|
||||
reason = "take_profit"
|
||||
elif sl_f is not None and _sl_triggered(direction, sl_f, mark, tick):
|
||||
reason = "stop_loss"
|
||||
if not reason:
|
||||
continue
|
||||
if mid > 0 and not _can_close_now(mid):
|
||||
continue
|
||||
if not _try_acquire_close_symbol(ms, direction):
|
||||
continue
|
||||
try:
|
||||
_execute_local_close(
|
||||
conn, mon, mode=mode, mark=mark, reason=reason,
|
||||
capital=capital, notify_fn=notify_fn,
|
||||
)
|
||||
if mid > 0:
|
||||
_mark_close_attempt(mid)
|
||||
closed += 1
|
||||
except Exception as exc:
|
||||
logger.warning("SL/TP tick close failed monitor=%s: %s", mid, exc)
|
||||
finally:
|
||||
_release_close_symbol(ms, direction)
|
||||
return closed
|
||||
|
||||
|
||||
def check_monitors_locally(
|
||||
conn,
|
||||
mode: str,
|
||||
|
||||
@@ -2,7 +2,9 @@
|
||||
/* 持仓监控页 — 与 split-grid(关键位监控)同宽,全端自适应 */
|
||||
.trade-page{width:100%}
|
||||
.trade-split{margin-bottom:1.25rem}
|
||||
.trade-split .card{min-height:480px}
|
||||
.trade-split .card{min-height:320px}
|
||||
.trade-split .trade-card#order{margin-bottom:.75rem}
|
||||
.sync-badge{font-size:.72rem;font-weight:400;margin-left:.35rem}
|
||||
.trade-top-bar{
|
||||
display:flex;flex-wrap:wrap;gap:.65rem 1rem;
|
||||
align-items:center;justify-content:space-between;
|
||||
|
||||
+53
-43
@@ -6,6 +6,8 @@
|
||||
var sizingMode = window.TRADE_SIZING_MODE || 'fixed';
|
||||
if (sizingMode === 'risk') sizingMode = 'amount';
|
||||
var list = document.getElementById('position-live-list');
|
||||
var orderList = document.getElementById('order-live-list');
|
||||
var syncBadge = document.getElementById('sync-badge');
|
||||
var recommendList = document.getElementById('recommend-list');
|
||||
var symInput = document.getElementById('trade-symbol');
|
||||
var dirSelect = document.getElementById('trade-direction');
|
||||
@@ -44,7 +46,7 @@
|
||||
var REC_COLSPAN = 18;
|
||||
var marketNavEnabled = !!window.MARKET_NAV_ENABLED;
|
||||
var productCategories = window.PRODUCT_CATEGORIES || [];
|
||||
var POS_CACHE_KEY = 'qihuo_trading_live_v3';
|
||||
var POS_CACHE_KEY = 'qihuo_trading_live_v4';
|
||||
|
||||
function runWhenReady(fn) {
|
||||
if (document.readyState === 'loading') {
|
||||
@@ -156,8 +158,21 @@
|
||||
return !!(msg && (msg.indexOf('不可达') >= 0 || msg.indexOf('Connection refused') >= 0 || msg.indexOf('timed out') >= 0));
|
||||
}
|
||||
|
||||
function applyActiveOrders(orders) {
|
||||
if (!orderList) return;
|
||||
orders = orders || [];
|
||||
if (!orders.length) {
|
||||
orderList.innerHTML = '<div class="empty-hint">暂无委托。</div>';
|
||||
return;
|
||||
}
|
||||
orderList.innerHTML = orders.map(buildPendingOrderCard).join('');
|
||||
bindPendingDismiss(orderList);
|
||||
bindCancelOpenButtons(orderList);
|
||||
bindCancelOrderButtons(orderList);
|
||||
}
|
||||
|
||||
function applyPositionsData(data) {
|
||||
if (!list || !data) return;
|
||||
if (!data) return;
|
||||
var cap = document.getElementById('cap-display');
|
||||
if (cap && data.capital != null) cap.textContent = Number(data.capital).toFixed(2);
|
||||
var connected = data.ctp_status && data.ctp_status.connected;
|
||||
@@ -168,6 +183,16 @@
|
||||
ctpConnecting = !!connecting;
|
||||
isTradingSession = !!data.trading_session;
|
||||
syncCtpBadgeFromStatus(data.ctp_status || { connected: connected, connecting: connecting });
|
||||
if (syncBadge) {
|
||||
if (data.sync_label && connected) {
|
||||
syncBadge.hidden = false;
|
||||
syncBadge.textContent = data.sync_label;
|
||||
syncBadge.className = 'sync-badge ' + (data.sync_state === 'syncing' ? 'text-accent' : 'text-muted');
|
||||
} else {
|
||||
syncBadge.hidden = true;
|
||||
syncBadge.textContent = '';
|
||||
}
|
||||
}
|
||||
if (!connected && !connecting && data.ctp_status && data.ctp_status.last_error) {
|
||||
showCtpError(data.ctp_status.last_error);
|
||||
if (isCtpLoginBanError(data.ctp_status.last_error)) {
|
||||
@@ -181,10 +206,14 @@
|
||||
riskBadge.textContent = data.risk_status.status_label || '';
|
||||
riskBadge.className = 'badge ' + (data.risk_status.can_trade ? 'profit' : 'loss');
|
||||
}
|
||||
var rows = data.rows || [];
|
||||
applyActiveOrders(data.active_orders || []);
|
||||
if (!list) return;
|
||||
var rows = (data.rows || []).filter(function (row) {
|
||||
return row.order_state !== 'pending';
|
||||
});
|
||||
var seenKeys = {};
|
||||
rows = rows.filter(function (row) {
|
||||
var k = row.key || ((row.symbol_code || '') + ':' + (row.direction || ''));
|
||||
var k = row.key || row.position_key || ((row.symbol_code || '') + ':' + (row.direction || ''));
|
||||
if (seenKeys[k]) return false;
|
||||
seenKeys[k] = true;
|
||||
return true;
|
||||
@@ -210,36 +239,7 @@
|
||||
tryAutoCtpReconnect();
|
||||
return;
|
||||
}
|
||||
var pendingOnly = data.pending_orders || [];
|
||||
if (pendingOnly.length) {
|
||||
list.innerHTML = '<div class="empty-hint" style="margin-bottom:.75rem">暂无持仓</div>' +
|
||||
pendingOnly.map(function (p) {
|
||||
var cancelAllowed = p.cancel_allowed !== false && isTradingSession;
|
||||
var actionBtn = '';
|
||||
if (p.monitor_id) {
|
||||
actionBtn = '<button type="button" class="pos-dismiss-btn' +
|
||||
(cancelAllowed ? '' : ' is-session-off') + '"' +
|
||||
(cancelAllowed ? '' : ' disabled title="不在交易时间段"') +
|
||||
' data-monitor-id="' + p.monitor_id + '" data-pending-cancel="1">撤单</button>';
|
||||
} else if (p.order_id && p.source === 'ctp') {
|
||||
actionBtn = '<button type="button" class="pos-dismiss-btn' +
|
||||
(cancelAllowed ? '' : ' is-session-off') + '"' +
|
||||
(cancelAllowed ? '' : ' disabled title="不在交易时间段"') +
|
||||
' data-cancel-order="' + encodeURIComponent(p.order_id) + '">撤单</button>';
|
||||
}
|
||||
return (
|
||||
'<div class="pos-pending-item ' +
|
||||
(p.order_kind === 'stop_loss' ? 'sl' : (p.order_kind === 'take_profit' ? 'tp' : 'ctp')) +
|
||||
'"><span>' + (p.label || '挂单') + ' · ' + (p.symbol || p.symbol_code) + '</span>' +
|
||||
'<span class="pos-pending-right"><strong>' + fmtNum(p.price) + '</strong> · ' +
|
||||
(p.lots || 1) + ' 手' + actionBtn + '</span></div>'
|
||||
);
|
||||
}).join('');
|
||||
bindPendingDismiss(list);
|
||||
bindCancelOrderButtons(list);
|
||||
} else {
|
||||
list.innerHTML = '<div class="empty-hint">暂无持仓。</div>';
|
||||
}
|
||||
list.innerHTML = '<div class="empty-hint">暂无持仓。</div>';
|
||||
return;
|
||||
}
|
||||
if (!connected) {
|
||||
@@ -839,22 +839,32 @@
|
||||
? row.pending_timeout_min
|
||||
: (row.auto_cancel_sec != null ? Math.max(1, Math.ceil(row.auto_cancel_sec / 60)) : 5);
|
||||
var cancelAllowed = row.cancel_allowed !== false && isTradingSession;
|
||||
var cancelBtn = row.can_cancel_order ?
|
||||
'<button type="button" class="pos-close-btn' + (cancelAllowed ? '' : ' is-session-off') + '"' +
|
||||
(cancelAllowed ? '' : ' disabled title="不在交易时间段"') +
|
||||
' data-cancel-open="' + row.monitor_id + '">撤单</button>' : '';
|
||||
var cancelBtn = '';
|
||||
if (row.can_cancel_order) {
|
||||
if (row.monitor_id) {
|
||||
cancelBtn = '<button type="button" class="pos-close-btn' + (cancelAllowed ? '' : ' is-session-off') + '"' +
|
||||
(cancelAllowed ? '' : ' disabled title="不在交易时间段"') +
|
||||
' data-cancel-open="' + row.monitor_id + '">撤单</button>';
|
||||
} else if (row.order_id || row.vt_order_id) {
|
||||
cancelBtn = '<button type="button" class="pos-close-btn' + (cancelAllowed ? '' : ' is-session-off') + '"' +
|
||||
(cancelAllowed ? '' : ' disabled title="不在交易时间段"') +
|
||||
' data-cancel-order="' + encodeURIComponent(row.vt_order_id || row.order_id) + '">撤单</button>';
|
||||
}
|
||||
}
|
||||
var pendingLabel = row.source_label || '挂单中';
|
||||
var isCloseOrder = pendingLabel.indexOf('平仓') >= 0;
|
||||
var metaLine =
|
||||
'状态 <strong class="text-accent">挂单中</strong>' +
|
||||
'状态 <strong class="text-accent">' + pendingLabel + '</strong>' +
|
||||
' · 委托价 <strong>' + fmtNum(orderPx) + '</strong>' +
|
||||
(row.rr_ratio != null ? ' · 盈亏比 <strong>' + row.rr_ratio + ':1</strong>' : '') +
|
||||
' · ' + slTpStatusHtml(row) +
|
||||
' · 移动保本 ' + trailingStatusHtml(row) +
|
||||
' · <span class="text-muted">约 ' + remainMin + ' 分钟内未成交自动撤单</span>';
|
||||
(row.stop_loss != null || row.take_profit != null ? ' · ' + slTpStatusHtml(row) : '') +
|
||||
(row.trailing_be ? ' · 移动保本 ' + trailingStatusHtml(row) : '') +
|
||||
(!isCloseOrder ? ' · <span class="text-muted">约 ' + remainMin + ' 分钟内未成交自动撤单</span>' : '');
|
||||
return (
|
||||
'<div class="pos-card is-pending">' +
|
||||
'<div class="pos-card-head"><div><div class="title">' + posSymbolTitleHtml(row,
|
||||
' <span class="badge dir">' + dirBadge + '</span>' +
|
||||
' <span class="badge pending">挂单中</span>') + '</div>' +
|
||||
' <span class="badge pending">' + (isCloseOrder ? '平仓委托' : '挂单中') + '</span>') + '</div>' +
|
||||
'<div class="text-muted pos-symbol-sub">' + posSymbolSubHtml(row) + '</div></div>' +
|
||||
'<div class="pos-card-actions">' + cancelBtn + '</div></div>' +
|
||||
'<div class="pos-card-meta pos-card-meta-line">' + metaLine + '</div>' +
|
||||
|
||||
+10
-2
@@ -105,11 +105,19 @@
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="card trade-card" id="active-orders">
|
||||
<h2>当前委托 <span class="sync-badge text-muted" id="sync-badge" hidden></span></h2>
|
||||
<p class="hint pos-hint">委托以 CTP 柜台为准;未成交可撤单,超时自动撤开仓单。</p>
|
||||
<div class="card-body card-scroll" id="order-live-list">
|
||||
<div class="empty-hint" id="order-placeholder">加载委托…</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="card trade-card" id="positions">
|
||||
<h2>当前持仓</h2>
|
||||
<p class="hint pos-hint">开仓委托先显示「挂单中」,柜台成交后写入监控;超过 <strong>{{ pending_order_timeout_min }}</strong> 分钟未成交自动撤单,可手动撤单。</p>
|
||||
<p class="hint pos-hint">持仓以 CTP 柜台为准;止盈止损为程序本地监控,触发后市价平仓。</p>
|
||||
<div class="card-body card-scroll" id="position-live-list">
|
||||
<div class="empty-hint" id="position-placeholder">加载本地持仓…</div>
|
||||
<div class="empty-hint" id="position-placeholder">加载持仓…</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
+190
-9
@@ -73,6 +73,7 @@ def _load_persisted_last_error() -> str:
|
||||
return (get_setting(CTP_LAST_ERROR_KEY, "") or "").strip()
|
||||
|
||||
_position_refresh_callback: Optional[Callable[[], None]] = None
|
||||
_tick_sl_tp_callback: Optional[Callable[[str, str, float], None]] = None
|
||||
_position_refresh_debounce_lock = threading.Lock()
|
||||
_position_refresh_debounce_ts: float = 0.0
|
||||
|
||||
@@ -82,6 +83,12 @@ def set_position_refresh_callback(fn: Optional[Callable[[], None]]) -> None:
|
||||
_position_refresh_callback = fn
|
||||
|
||||
|
||||
def set_tick_sl_tp_callback(fn: Optional[Callable[[str, str, float], None]]) -> None:
|
||||
"""注册 tick 回调:exchange, symbol, last_price → 本地 SL/TP 触发。"""
|
||||
global _tick_sl_tp_callback
|
||||
_tick_sl_tp_callback = fn
|
||||
|
||||
|
||||
def _fire_position_refresh_callback() -> None:
|
||||
fn = _position_refresh_callback
|
||||
if not fn:
|
||||
@@ -223,6 +230,8 @@ class CtpBridge:
|
||||
self._last_connect_ok_ts: float = 0.0
|
||||
self._tick_hooked = False
|
||||
self._position_hooked = False
|
||||
self._order_hooked = False
|
||||
self._trade_hooked = False
|
||||
self._bar_generators: dict[str, Any] = {}
|
||||
self._bars_1m: dict[str, deque] = {}
|
||||
self._init_engine()
|
||||
@@ -238,6 +247,8 @@ class CtpBridge:
|
||||
self._engine = MainEngine(self._ee)
|
||||
self._engine.add_gateway(CtpGateway)
|
||||
self._ensure_position_event_hook()
|
||||
self._ensure_order_event_hook()
|
||||
self._ensure_trade_event_hook()
|
||||
except ImportError:
|
||||
self._last_error = "未安装 vnpy / vnpy_ctp,请 pip install vnpy vnpy_ctp"
|
||||
except Exception as exc:
|
||||
@@ -253,17 +264,29 @@ class CtpBridge:
|
||||
|
||||
def _on_position(event) -> None:
|
||||
try:
|
||||
from ctp_trading_state import trading_state
|
||||
|
||||
pos = event.data
|
||||
vol = int(getattr(pos, "volume", 0) or 0)
|
||||
if vol <= 0:
|
||||
return
|
||||
row = self._position_row_from_vnpy(pos)
|
||||
if row:
|
||||
trading_state.upsert_position(row, notify=False)
|
||||
sym = getattr(pos, "symbol", "") or ""
|
||||
d = "long" if _is_long_direction(getattr(pos, "direction", None)) else "short"
|
||||
for attr in ("margin", "use_margin", "UseMargin"):
|
||||
raw = float(getattr(pos, attr, 0) or 0)
|
||||
if raw > 0:
|
||||
self._position_margins[self._position_margin_key(sym, d)] = raw
|
||||
break
|
||||
vol = int(getattr(pos, "volume", 0) or 0)
|
||||
if vol <= 0:
|
||||
exchange = getattr(pos, "exchange", None)
|
||||
ex_name = str(exchange.value if hasattr(exchange, "value") else exchange or "")
|
||||
from ctp_trading_state import position_key
|
||||
|
||||
trading_state.remove_position(
|
||||
position_key(ex_name, sym, d), notify=False,
|
||||
)
|
||||
else:
|
||||
for attr in ("margin", "use_margin", "UseMargin"):
|
||||
raw = float(getattr(pos, attr, 0) or 0)
|
||||
if raw > 0:
|
||||
self._position_margins[self._position_margin_key(sym, d)] = raw
|
||||
break
|
||||
except Exception as exc:
|
||||
logger.debug("position margin cache: %s", exc)
|
||||
_fire_position_refresh_callback_debounced()
|
||||
@@ -271,6 +294,137 @@ class CtpBridge:
|
||||
self._ee.register(EVENT_POSITION, _on_position)
|
||||
self._position_hooked = True
|
||||
|
||||
def _ensure_order_event_hook(self) -> None:
|
||||
if self._order_hooked or not self._ee:
|
||||
return
|
||||
try:
|
||||
from vnpy.trader.event import EVENT_ORDER
|
||||
except ImportError:
|
||||
return
|
||||
|
||||
def _on_order(event) -> None:
|
||||
try:
|
||||
from ctp_trading_state import trading_state
|
||||
|
||||
order = event.data
|
||||
row = self._order_row_from_vnpy(order)
|
||||
if not row:
|
||||
return
|
||||
status_s = str(row.get("status") or "")
|
||||
terminal = any(
|
||||
x in status_s
|
||||
for x in ("ALLTRADED", "CANCELLED", "REJECTED", "全部成交", "已撤销", "拒单")
|
||||
)
|
||||
oid = str(row.get("order_id") or row.get("vt_order_id") or "")
|
||||
if terminal or int(row.get("lots") or 0) <= 0:
|
||||
trading_state.remove_order(oid, notify=False)
|
||||
else:
|
||||
trading_state.upsert_order(row, notify=False)
|
||||
except Exception as exc:
|
||||
logger.debug("order event: %s", exc)
|
||||
_fire_position_refresh_callback_debounced(min_interval=0.2)
|
||||
|
||||
self._ee.register(EVENT_ORDER, _on_order)
|
||||
self._order_hooked = True
|
||||
|
||||
def _ensure_trade_event_hook(self) -> None:
|
||||
if self._trade_hooked or not self._ee:
|
||||
return
|
||||
try:
|
||||
from vnpy.trader.event import EVENT_TRADE
|
||||
except ImportError:
|
||||
return
|
||||
|
||||
def _on_trade(event) -> None:
|
||||
try:
|
||||
trade = event.data
|
||||
row = self._trade_row_from_vnpy(trade)
|
||||
if row and row.get("offset") == "open":
|
||||
sym = row.get("symbol") or ""
|
||||
pd = row.get("position_direction") or "long"
|
||||
dt = row.get("datetime") or ""
|
||||
if sym and dt:
|
||||
self._position_open_times[self._position_margin_key(sym, pd)] = dt
|
||||
except Exception as exc:
|
||||
logger.debug("trade event: %s", exc)
|
||||
_fire_position_refresh_callback_debounced(min_interval=0.2)
|
||||
|
||||
self._ee.register(EVENT_TRADE, _on_trade)
|
||||
self._trade_hooked = True
|
||||
|
||||
def _order_row_from_vnpy(self, order: Any) -> Optional[dict[str, Any]]:
|
||||
try:
|
||||
status = getattr(order, "status", None)
|
||||
status_s = str(status)
|
||||
vol = int(getattr(order, "volume", 0) or 0)
|
||||
traded = int(getattr(order, "traded", 0) or 0)
|
||||
remain = max(0, vol - traded)
|
||||
direction = getattr(order, "direction", None)
|
||||
d = "long"
|
||||
if direction is not None and str(direction).endswith("SHORT"):
|
||||
d = "short"
|
||||
offset = getattr(order, "offset", None)
|
||||
sym = getattr(order, "symbol", "") or ""
|
||||
exchange = getattr(order, "exchange", None)
|
||||
ex_name = str(exchange.value if hasattr(exchange, "value") else exchange or "")
|
||||
vt_oid = str(getattr(order, "vt_orderid", "") or "")
|
||||
order_id = str(getattr(order, "orderid", "") or "")
|
||||
return {
|
||||
"symbol": sym,
|
||||
"exchange": ex_name,
|
||||
"direction": d,
|
||||
"lots": remain,
|
||||
"price": float(getattr(order, "price", 0) or 0),
|
||||
"offset": str(offset or ""),
|
||||
"order_id": vt_oid or order_id,
|
||||
"vt_order_id": vt_oid,
|
||||
"status": status_s,
|
||||
}
|
||||
except Exception as exc:
|
||||
logger.debug("order_row_from_vnpy: %s", exc)
|
||||
return None
|
||||
|
||||
def _position_row_from_vnpy(self, pos: Any) -> Optional[dict[str, Any]]:
|
||||
try:
|
||||
vol = int(getattr(pos, "volume", 0) or 0)
|
||||
d = "long" if _is_long_direction(getattr(pos, "direction", None)) else "short"
|
||||
sym = getattr(pos, "symbol", "") or ""
|
||||
exchange = getattr(pos, "exchange", None)
|
||||
ex_name = str(exchange.value if hasattr(exchange, "value") else exchange or "")
|
||||
price = float(getattr(pos, "price", 0) or 0)
|
||||
yd = int(getattr(pos, "yd_volume", 0) or 0)
|
||||
td = max(0, vol - yd)
|
||||
margin = self.estimate_position_margin(sym, ex_name, d, vol, price, pos=pos)
|
||||
open_time = self._lookup_position_open_time(sym, d) or None
|
||||
return {
|
||||
"symbol": sym,
|
||||
"exchange": ex_name,
|
||||
"direction": d,
|
||||
"lots": vol,
|
||||
"avg_price": price,
|
||||
"pnl": float(getattr(pos, "pnl", 0) or 0),
|
||||
"frozen": int(getattr(pos, "frozen", 0) or 0),
|
||||
"margin": margin,
|
||||
"open_time": open_time,
|
||||
"yd_volume": yd,
|
||||
"td_volume": td,
|
||||
}
|
||||
except Exception as exc:
|
||||
logger.debug("position_row_from_vnpy: %s", exc)
|
||||
return None
|
||||
|
||||
def calibrate_trading_state(self) -> None:
|
||||
"""全量校准内存簿(读 vnpy 缓存,不 query 柜台)。"""
|
||||
try:
|
||||
from ctp_trading_state import trading_state
|
||||
|
||||
with _ctp_td_lock:
|
||||
orders = self.list_active_orders()
|
||||
positions = self._collect_positions()
|
||||
trading_state.calibrate_from_lists(orders, positions)
|
||||
except Exception as exc:
|
||||
logger.debug("calibrate trading state: %s", exc)
|
||||
|
||||
def available(self) -> bool:
|
||||
return self._engine is not None
|
||||
|
||||
@@ -336,6 +490,12 @@ class CtpBridge:
|
||||
except Exception as exc:
|
||||
logger.debug("gateway close: %s", exc)
|
||||
self._connected_mode = None
|
||||
try:
|
||||
from ctp_trading_state import trading_state
|
||||
|
||||
trading_state.clear()
|
||||
except Exception:
|
||||
pass
|
||||
time.sleep(0.6)
|
||||
|
||||
def _login_rejected(self, ctp_logs: list[str]) -> bool:
|
||||
@@ -463,6 +623,10 @@ class CtpBridge:
|
||||
mode, self._td_logged_in(),
|
||||
len(self._engine.get_all_accounts() or []))
|
||||
self._schedule_fee_sync(mode)
|
||||
try:
|
||||
self.calibrate_trading_state()
|
||||
except Exception as exc:
|
||||
logger.debug("post-connect calibrate: %s", exc)
|
||||
_fire_position_refresh_burst()
|
||||
return
|
||||
finally:
|
||||
@@ -844,6 +1008,20 @@ class CtpBridge:
|
||||
sym = (getattr(tick, "symbol", "") or "").lower()
|
||||
te = getattr(tick, "exchange", None)
|
||||
ex_s = str(te.value if hasattr(te, "value") else te or "").upper()
|
||||
price = self._price_from_tick(tick)
|
||||
if price and price > 0:
|
||||
try:
|
||||
from ctp_trading_state import trading_state
|
||||
|
||||
trading_state.set_tick_price(ex_s, sym, price)
|
||||
except Exception:
|
||||
pass
|
||||
fn = _tick_sl_tp_callback
|
||||
if fn:
|
||||
try:
|
||||
fn(ex_s, sym, float(price))
|
||||
except Exception as exc:
|
||||
logger.debug("tick sl/tp callback: %s", exc)
|
||||
key = self._tick_key(sym, ex_s)
|
||||
bg = self._bar_generators.get(key)
|
||||
if not bg:
|
||||
@@ -1329,6 +1507,8 @@ class CtpBridge:
|
||||
sym = getattr(order, "symbol", "") or ""
|
||||
exchange = getattr(order, "exchange", None)
|
||||
ex_name = str(exchange.value if hasattr(exchange, "value") else exchange or "")
|
||||
vt_oid = str(getattr(order, "vt_orderid", "") or "")
|
||||
order_id = str(getattr(order, "orderid", "") or "")
|
||||
out.append({
|
||||
"symbol": sym,
|
||||
"exchange": ex_name,
|
||||
@@ -1336,7 +1516,8 @@ class CtpBridge:
|
||||
"lots": remain,
|
||||
"price": float(getattr(order, "price", 0) or 0),
|
||||
"offset": offset_s,
|
||||
"order_id": str(getattr(order, "orderid", "") or ""),
|
||||
"order_id": vt_oid or order_id,
|
||||
"vt_order_id": vt_oid,
|
||||
"status": status_s,
|
||||
})
|
||||
return out
|
||||
|
||||
Reference in New Issue
Block a user