kyugao/sfgrid
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

new ui

Browse Source
This commit is contained in:
WIN-ALB39P9B6BU\Docker
2026-06-12 16:25:41 +08:00
parent ef4c1cca32
commit 2d8a0c3bca
23 changed files with 2904 additions and 525 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/ui/flet/app_v2.py
+863
View File
@@ -0,0 +1,863 @@
"""
Flet UI v2 — 重构版。
分层架构：
DataStore — 纯数据，无 UI 依赖
GridPanel — 网格策略持仓表格 + 选中 + 操作栏
DrawerPanel — 右侧抽屉叠加层 + 四 Tab
TradeDialogs— 启动/暂停/网格配置弹窗
QmtApp — 生命周期协调
"""
import asyncio
import time
import threading
import flet as ft
from core.qmt_real import RealQmtV, qmtv
from core.logger import LogLevel, PrintLog
from core.sfgrid.model import SFGridTradeTarget, STRATEGY_TYPE_GRID
from core.sfgrid.sfgrid_strategy import SFGridStrategy
from core.eventbus import event_bus, MarketDataUpdate, EventMarketActiveSwitch
from core.sfgrid.bus_events import EventTradeTargetUpdate
# ── 工具函数 ──
_ORDER_STATUS = {48: '未报', 49: '待报', 50: '已报', 51: '已报待撤', 52: '部成待撤',
53: '部撤', 54: '已撤', 55: '部成', 56: '已成', 57: '废单'}
def _fmt_time(t) -> str:
if not t: return ''
import datetime
try:
ts = int(t)
if ts > 1e12: ts //= 1000
return datetime.datetime.fromtimestamp(ts).strftime('%H:%M:%S')
except (ValueError, OSError):
return str(t)
def _direction(ot: int) -> str:
return '买' if ot == 23 else '卖' if ot == 24 else str(ot)
def _plain(code: str) -> str:
return code.split('.')[0] if '.' in code else code
def _text(s, color=None, bold=False, size=None) -> ft.Text:
"""快捷构建 Text"""
return ft.Text(str(s), color=color, weight=ft.FontWeight.BOLD if bold else None, size=size)
def _datatable(cols: list[str], rows: list[list[str]], col_widths: list = None) -> ft.DataTable:
"""构建 DataTable（不包装 ListView）"""
data_rows = []
for r in rows:
cells = []
for i, c in enumerate(r):
w = (col_widths or [None] * len(r))[i] if i < len(r) else None
cells.append(ft.DataCell(ft.Text(str(c), overflow=ft.TextOverflow.ELLIPSIS, max_lines=1, width=w)))
data_rows.append(ft.DataRow(cells=cells))
if not data_rows:
data_rows.append(ft.DataRow(cells=[ft.DataCell(ft.Text("")) for _ in cols]))
return ft.DataTable(columns=[ft.DataColumn(ft.Text(h)) for h in cols], rows=data_rows,
width=float('inf'), heading_row_height=36, data_row_min_height=32)
# ══════════════════════════════════════════════════════════════
# DataStore — 纯数据层
# ══════════════════════════════════════════════════════════════
class _DataStore:
"""持仓、行情、订单数据管理，不依赖任何 UI"""
def __init__(self):
self.tradeTargets: dict[int, SFGridTradeTarget] = {}
self.stockCodeIdMap: dict[str, int] = {}
self.marketPrices: dict[int, float] = {} # tid → lastPrice
self.preClose: dict[int, float] = {} # tid → 昨收
self.avgPrices: dict[int, float] = {} # tid → avgPrice
self.orders: list = []
self.trades: list = []
self.strategyCtrl: dict[int, SFGridStrategy] = {}
# 市场监控
self.monitorPrice: float = 10.0
self.marketLog: dict[str, dict] = {} # stock_code → {name, price, time}
self.listeningStocks: list = []
# ── 初始化 ──
def load_from_qmt(self):
"""从 QMT 加载持仓并导入 DB"""
positions = qmtv.getAllPositions()
PrintLog(LogLevel.INFO, f'[Data] 持仓: {len(positions)} 个')
for code, pos in positions.items():
existing = SFGridTradeTarget.get_or_none(SFGridTradeTarget.stock_code == code)
if existing is None:
name = getattr(pos, 'instrument_name', '') or qmtv.getInstrumentName(code)
SFGridTradeTarget.create(
stock_code=code, stock_name=name,
current_position=int(pos.volume),
init_price=float(getattr(pos, 'avg_price', 0) or 0),
grid_index=0, enabled=False,
grid_start_price=float(getattr(pos, 'avg_price', 0) or 0) or 10.0,
grid_size=1.0, grid_volume=200, grid_upper_count=1, grid_lower_count=10,
)
# 映射 (先建立 stockCodeIdMap)
results = list(SFGridTradeTarget.select())
for t in results:
pos = positions.get(t.stock_code)
t.current_position = 0 if pos is None else int(pos.volume)
tid = t.get_id()
self.tradeTargets[tid] = t
self.stockCodeIdMap[t.stock_code] = tid
if pos is not None:
self.avgPrices[tid] = float(getattr(pos, 'avg_price', 0) or 0)
# 昨收（依赖 stockCodeIdMap）
self._load_preclose(positions)
def _load_preclose(self, positions: dict):
try:
from xtquant import xtdata
for stock_code in positions:
if '.' not in stock_code:
full_code = f'{stock_code}.SH' if stock_code.startswith(('6','5','9')) else f'{stock_code}.SZ'
else:
full_code = stock_code
detail = xtdata.get_instrument_detail(full_code)
if detail:
pc = detail.get('PreClose', 0) if isinstance(detail, dict) else getattr(detail, 'PreClose', 0)
if pc > 0 and stock_code in self.stockCodeIdMap:
self.preClose[self.stockCodeIdMap[stock_code]] = float(pc)
PrintLog(LogLevel.INFO, f'[Data] 昨收: {len(self.preClose)} 个')
except Exception as e:
PrintLog(LogLevel.DEBUG, f'[Data] 昨收异常: {e}')
def init_strategies(self):
for tid, t in self.tradeTargets.items():
if t.strategy_type == STRATEGY_TYPE_GRID and t.enabled:
self.strategyCtrl[tid] = SFGridStrategy(t)
# ── 行情 ──
def apply_tick(self, data: dict) -> int:
"""应用行情数据，返回更新的标的数"""
count = 0
for stock_code, tick in data.items():
plain = _plain(stock_code)
tid = self.stockCodeIdMap.get(plain)
lp = tick.get('lastPrice', 0)
if tid is not None and tid in self.tradeTargets:
self.marketPrices[tid] = lp
self.tradeTargets[tid].market_price = lp
count += 1
elif lp == self.monitorPrice or stock_code in self.listeningStocks:
if stock_code not in self.listeningStocks:
self.listeningStocks.append(stock_code)
self.marketLog[stock_code] = {
'stock_name': qmtv.getInstrumentName(stock_code),
'last_price': lp,
'time': time.strftime("%H:%M:%S"),
}
return count
def pull_prices(self):
"""主动拉取缺失的市价"""
for tid, t in self.tradeTargets.items():
if tid not in self.marketPrices or self.marketPrices[tid] == 0:
price = qmtv.getLastPrice(t.stock_code)
if price > 0:
self.marketPrices[tid] = price
t.market_price = price
# ── 持仓 ──
def refresh_positions(self):
positions = qmtv.getAllPositions()
for t in self.tradeTargets.values():
pos = positions.get(t.stock_code)
t.current_position = 0 if pos is None else int(pos.volume)
# ── 委托 / 成交 ──
def refresh_orders(self):
try:
self.orders = list(qmtv.queryTodayOrders())
except Exception:
pass
def refresh_trades(self):
try:
self.trades = list(qmtv.queryTodayTrades())
except Exception:
pass
def active_orders(self) -> list:
"""返回过滤+去重后的活跃委托行"""
_CANCELED = {54, 57}
o_map = {}
for o in self.orders:
oid = str(getattr(o, 'order_id', ''))
if not oid: continue
o_map[oid] = o
rows = []
for o in o_map.values():
st = getattr(o, 'order_status', 0)
if st in _CANCELED: continue
tv = getattr(o, 'traded_volume', 0) or 0
ov = getattr(o, 'order_volume', 0) or 0
rows.append([
_fmt_time(getattr(o, 'order_time', 0)),
_plain(getattr(o, 'stock_code', '')),
getattr(o, 'instrument_name', '') or '',
_direction(getattr(o, 'order_type', 0)),
f"{getattr(o, 'price', 0):.3f}",
f"{tv}/{ov}",
f"{getattr(o, 'traded_price', 0):.3f}" if getattr(o, 'traded_price', 0) > 0 else '-',
_ORDER_STATUS.get(st, '未知'),
])
return rows
def active_trades(self) -> list:
"""返回去重后的成交行"""
t_map = {}
for t in self.trades:
tid = str(getattr(t, 'traded_id', ''))
if not tid: continue
t_map[tid] = t
rows = []
for t in t_map.values():
rows.append([
_fmt_time(getattr(t, 'traded_time', 0)),
_plain(getattr(t, 'stock_code', '')),
getattr(t, 'instrument_name', '') or '',
_direction(getattr(t, 'order_type', 0)),
f"{getattr(t, 'traded_price', 0):.3f}",
str(getattr(t, 'traded_volume', 0)),
f"{getattr(t, 'traded_amount', 0):.2f}",
f"{getattr(t, 'commission', 0):.2f}",
])
return rows
def pending_tags(self, stock_code: str) -> list:
"""某标的挂单方向标签"""
tags = []
for o in self.orders:
if _plain(getattr(o, 'stock_code', '')) != stock_code: continue
if getattr(o, 'order_status', 0) in {54, 57}: continue
ot = getattr(o, 'order_type', 0)
if ot == 23 and '多' not in tags: tags.append('多')
elif ot == 24 and '空' not in tags: tags.append('空')
return tags
# ══════════════════════════════════════════════════════════════
# GridPanel — 网格策略持仓表格
# ══════════════════════════════════════════════════════════════
class _GridPanel:
"""网格策略持仓面板 — 自定义 Row 表格，行内按钮直接可用"""
def __init__(self, data: _DataStore, dialogs=None):
self._data = data
self._dialogs = dialogs # _TradeDialogs
self._col = None
@property
def controls(self):
return self._col.controls if self._col else []
def build(self) -> ft.Control:
self._col = ft.Column(spacing=0, scroll=ft.ScrollMode.AUTO, expand=True)
self._rebuild()
return self._col
def _rebuild(self):
# 列: (宽度/expand, ...) expand为比重，无则固定宽
_C = [(40, 0), (0, 2), (80, 2), (0, 2), (55, 1), (65, 1), (75, 1), (100, 0)]
H = ["ID", "股票", "市场价", "网格基准", "持仓", "成本", "状态", "操作"]
header = []
for h, (w, e) in zip(H, _C):
if e > 0:
header.append(ft.Container(_text(h, bold=True), padding=4, expand=e))
else:
header.append(ft.Container(_text(h, bold=True), width=w, padding=4))
rows = [ft.Row(header, spacing=0), ft.Divider(height=1, color='#e0e0e0')]
for tid, t in self._data.tradeTargets.items():
if t.strategy_type != STRATEGY_TYPE_GRID: continue
pg = t.getPriceGrid()
idx = t.grid_index
base = pg[idx] if 0 <= idx < len(pg) else 0
mp = self._data.marketPrices.get(tid, 0) or 0
pc = self._data.preClose.get(tid, 0) or 0
pcolor = '#CC0000' if mp > pc > 0 else '#009900' if mp < pc and mp > 0 else None
# 网格基准列
gcells = [_text(f'{base:.2f}', bold=True, )]
if mp > base > 0:
gcells.append(_text(' ▲', color='#CC0000', bold=True, ))
elif 0 < mp < base:
gcells.append(_text(' ▼', color='#009900', bold=True, ))
for tag in self._data.pending_tags(t.stock_code):
gcells.append(ft.Container(
_text(tag, color='white', bold=True, size=10),
bgcolor='#E67E22' if tag == '多' else '#3498DB',
border_radius=4, padding=ft.Padding(3, 1, 3, 1),
))
gcell = ft.Row(gcells, spacing=3) if len(gcells) > 1 else gcells[0]
# 操作按钮 — IconButton 图标居中，无点击偏移
if t.enabled:
btns = [
ft.IconButton(ft.Icons.PAUSE, icon_size=18, tooltip="暂停",
on_click=lambda e, tt=t: self._dialogs.confirm_stop(tt)),
]
else:
btns = [
ft.IconButton(ft.Icons.PLAY_ARROW, icon_size=18, tooltip="启动",
on_click=lambda e, tt=t: self._dialogs.confirm_start(tt)),
]
btns.append(ft.IconButton(ft.Icons.SETTINGS, icon_size=18, tooltip="设置",
icon_color='#BDBDBD' if t.enabled else None,
on_click=lambda e, tt=t: self._dialogs.open_config(tt)))
cells_text = [str(tid), f'{t.stock_code} {t.stock_name}',
f'{mp:.3f}', gcell,
str(t.current_position),
f'{self._data.avgPrices.get(tid, 0):.3f}',
'▶运行中' if t.enabled else '⏸已暂停',
btns]
row_cells = []
for i, (w, e) in enumerate(_C):
content = cells_text[i]
if i == 2: # 市场价用颜色
content = _text(cells_text[i], color=pcolor, bold=True)
elif isinstance(content, str):
content = _text(content)
elif isinstance(content, list):
content = ft.Row(content, spacing=2)
if e > 0:
row_cells.append(ft.Container(content, padding=4, expand=e))
else:
row_cells.append(ft.Container(content, width=w, padding=4 if i != 7 else 2))
row = ft.Row(row_cells, spacing=0)
rows.append(ft.Container(row, padding=ft.Padding(0, 2, 0, 2)))
rows.append(ft.Divider(height=1, color='#f0f0f0'))
self._col.controls = rows
# ══════════════════════════════════════════════════════════════
# DrawerPanel — 右侧抽屉面板
# ══════════════════════════════════════════════════════════════
class _DrawerPanel:
"""右侧叠加抽屉：市场监控 + 委托 + 成交 + 未分类"""
def __init__(self, data: _DataStore, dialogs=None):
self._data = data
self._dialogs = dialogs
self._tab_orders = ft.Tab(label="当前委托")
self._tab_trades = ft.Tab(label="当日成交")
self._order_table = None
self._trade_table = None
self._uncl_col = None
self._market_table = None
def build(self) -> ft.Control:
"""返回 overlay Container"""
self._uncl_col = ft.Column(spacing=0, scroll=ft.ScrollMode.AUTO, expand=True)
self._order_table = ft.ListView(expand=True)
self._trade_table = ft.ListView(expand=True)
self._market_table = ft.ListView(expand=True)
bar = ft.TabBar(tabs=[
ft.Tab(label="实时价格监控"),
self._tab_orders,
self._tab_trades,
ft.Tab(label="未分类持仓"),
])
view = ft.TabBarView(controls=[
self._build_market_tab(),
self._order_table,
self._trade_table,
self._uncl_col,
], expand=True)
panel = ft.Container(
ft.Column([
ft.Container(_text("监控面板", bold=True, size=14), padding=ft.Padding(10, 10, 10, 5)),
ft.Tabs(ft.Column([bar, view], expand=True), length=4, expand=True),
], expand=True),
width=700, bgcolor=ft.Colors.SURFACE,
)
backdrop = ft.Container(bgcolor='#44000000', expand=True)
return ft.Container(ft.Row([backdrop, panel], spacing=0), visible=False, expand=True)
def _build_market_tab(self) -> ft.Control:
price_input = ft.TextField(value=str(self._data.monitorPrice), width=80, height=32, text_size=13)
return ft.Column([
ft.Row([_text("监控配置", size=13), _text("价格", size=13), price_input,
ft.ElevatedButton("确认", on_click=lambda e: self._set_monitor(price_input.value), height=32)]),
ft.Container(content=self._market_table, expand=True),
], expand=True)
def _set_monitor(self, val: str):
try:
self._data.monitorPrice = float(val)
self._data.marketLog.clear()
self._data.listeningStocks.clear()
except ValueError:
pass
# ── 刷新 ──
def refresh_all(self):
self._refresh_grid()
self._refresh_orders()
self._refresh_trades()
self._refresh_market()
def _refresh_grid(self):
W = [35, 180, 80, 60, 70, 65] # ID, 股票, 市场价, 持仓, 成本, 操作
H = ["ID", "股票", "市场价", "持仓", "成本", "操作"]
rows = [
ft.Row([ft.Container(_text(h, bold=True), width=w, padding=4)
for h, w in zip(H, W)], spacing=0),
ft.Divider(height=1, color='#e0e0e0'),
]
for tid, t in self._data.tradeTargets.items():
if t.strategy_type == STRATEGY_TYPE_GRID: continue
mp = self._data.marketPrices.get(tid, 0) or 0
row = ft.Row([
ft.Container(_text(str(tid)), width=W[0], padding=4),
ft.Container(_text(f'{t.stock_code} {t.stock_name}'), width=W[1], padding=4),
ft.Container(_text(f'{mp:.3f}'), width=W[2], padding=4),
ft.Container(_text(str(t.current_position)), width=W[3], padding=4),
ft.Container(_text(f'{self._data.avgPrices.get(tid, 0):.3f}'), width=W[4], padding=4),
ft.Container(ft.IconButton(ft.Icons.SETTINGS, icon_size=18, tooltip="网格配置",
on_click=lambda e, tt=t: self._dialogs.open_config(tt)),
width=W[5], padding=0),
], spacing=0)
rows.append(ft.Container(row, padding=ft.Padding(0, 2, 0, 2)))
rows.append(ft.Divider(height=1, color='#f0f0f0'))
self._uncl_col.controls = rows
def _refresh_orders(self):
rows = self._data.active_orders()
self._tab_orders.label = f"当前委托 ({len(rows)})" if rows else "当前委托"
cols = ["时间", "代码", "名称", "方向", "委托价", "已成交/委托量", "均价", "状态"]
self._order_table.controls = [ft.ListView(
[_datatable(cols, rows, [65,55,70,35,60,80,55,50])], expand=True)]
def _refresh_trades(self):
rows = self._data.active_trades()
self._tab_trades.label = f"当日成交 ({len(rows)})" if rows else "当日成交"
cols = ["时间", "代码", "名称", "方向", "成交价", "成交量", "成交金额", "手续费"]
self._trade_table.controls = [ft.ListView(
[_datatable(cols, rows, [65,55,70,35,65,60,70,55])], expand=True)]
def _refresh_market(self):
rows = [[d['time'], f"{d['stock_name']}-{sc}", f"{d['last_price']:.3f}"]
for sc, d in self._data.marketLog.items()]
self._market_table.controls = [ft.ListView(
[_datatable(["时间", "股票名称", "最新价格"], rows)], expand=True)]
# ══════════════════════════════════════════════════════════════
# ActionBar — 选中行后的操作按钮栏
# ══════════════════════════════════════════════════════════════
# ══════════════════════════════════════════════════════════════
# SplashScreen
# ══════════════════════════════════════════════════════════════
class _SplashScreen:
def __init__(self, page: ft.Page):
self._bar = ft.ProgressBar(width=340, value=0, color='#0078d4')
self._status = ft.Text("正在初始化...", size=13)
self._page = page
page.add(ft.Container(
ft.Container(
ft.Column([
_text("神之一手", color='#0078d4', bold=True, size=22),
_text("交易系统", color='#666666', size=14),
ft.Container(height=20),
self._status,
ft.Container(height=8),
self._bar,
], alignment=ft.MainAxisAlignment.CENTER, horizontal_alignment=ft.CrossAxisAlignment.CENTER),
width=380, height=200,
bgcolor=ft.Colors.SURFACE, border_radius=12,
shadow=ft.BoxShadow(blur_radius=20, color='#20000000'),
alignment=ft.Alignment.CENTER,
),
alignment=ft.Alignment.CENTER, expand=True, bgcolor='#F5F5F5',
))
page.update()
def progress(self, text: str, pct: float):
self._status.value = text
self._bar.value = pct
self._page.update()
# ══════════════════════════════════════════════════════════════
# TradeDialogs
# ══════════════════════════════════════════════════════════════
class _TradeDialogs:
"""启动/暂停/网格配置弹窗"""
def __init__(self, page: ft.Page, data: _DataStore, on_done):
self._page = page
self._data = data
self._on_done = on_done
def confirm_start(self, target):
dlg = ft.AlertDialog(
title=_text("确认启动"),
content=_text(f"确定要启动交易吗？\n\n{target.stock_code} {target.stock_name}"),
actions=[
ft.TextButton("取消", on_click=lambda e: self._close()),
ft.TextButton("确定", on_click=lambda e, t=target: self._do_start(t)),
],
)
self._page.show_dialog(dlg)
def confirm_stop(self, target):
dlg = ft.AlertDialog(
title=_text("确认暂停"),
content=_text(f"确定要暂停交易吗？\n\n{target.stock_code} {target.stock_name}"),
actions=[
ft.TextButton("取消", on_click=lambda e: self._close()),
ft.TextButton("确定", on_click=lambda e, t=target: self._do_stop(t)),
],
)
self._page.show_dialog(dlg)
def open_config(self, target):
if target.enabled:
dlg = ft.AlertDialog(title=_text("提示"), content=_text("运行中的策略无法修改参数，请先暂停"))
self._page.show_dialog(dlg)
# 1.5 秒后自动关闭
import asyncio
async def _auto_close():
await asyncio.sleep(1.5)
self._page.pop_dialog()
self._page.update()
asyncio.ensure_future(_auto_close())
return
PrintLog(LogLevel.INFO, f'[Flet-v2] 打开网格配置: {target.stock_code}')
def _row(label, value, unit=''):
inp = ft.TextField(value=str(value), width=120, height=30, text_size=13,
content_padding=ft.Padding(6, 0, 6, 0))
return ft.Row([_text(label, size=12),
inp,
_text(unit, color='#888888', size=12) if unit else ft.Container(width=0)],
spacing=6, alignment=ft.MainAxisAlignment.START)
r1 = _row("基准价格:", target.grid_start_price, "元")
r2 = _row("网格大小:", target.grid_size, "元")
r3 = _row("网格交易量:", target.grid_volume, "手")
r4 = _row("上方网格数量:", target.grid_upper_count, "格")
r5 = _row("下方网格数量:", target.grid_lower_count, "格")
r6 = _row("当前网格层级:", target.grid_index, "")
all_rows = [r1, r2, r3, r4, r5, r6]
inputs = [r.controls[1] for r in all_rows] # index 1 is the TextField
preview = ft.Text("", size=11, italic=True)
def _refresh_preview(e=None):
try:
bp, gs = float(inputs[0].value), float(inputs[1].value)
up, lo = int(inputs[3].value), int(inputs[4].value)
parts = []
for i in range(up, 0, -1): parts.append(f"{bp + gs * i:.2f}(卖{up - i + 1})")
parts.append(f"→{bp:.2f}←(基准)")
for i in range(1, lo + 1):
p = bp - gs * i
if p > 0: parts.append(f"{p:.2f}(买{i})")
preview.value = " ".join(parts) or "—"
except ValueError:
preview.value = ""
preview.update()
# 实时计算：四个关键字段绑定 on_change
for idx in (0, 1, 3, 4):
inputs[idx].on_change = _refresh_preview
def _save(e):
try:
target.grid_start_price = float(inputs[0].value)
target.grid_size = float(inputs[1].value)
target.grid_volume = int(inputs[2].value)
target.grid_upper_count = int(inputs[3].value)
target.grid_lower_count = int(inputs[4].value)
target.grid_index = int(inputs[5].value)
target.save()
self._close()
self._on_done()
PrintLog(LogLevel.INFO, f'[Flet] 网格配置已保存: {target.targetName()}')
except ValueError:
pass
info = ft.Column([
ft.Row([_text("股票代码:", bold=True, size=12), _text(target.stock_code, size=12)]),
ft.Row([_text("股票名称:", bold=True, size=12), _text(target.stock_name, size=12)]),
], spacing=2)
self._page.show_dialog(ft.AlertDialog(
title=_text(f"网格配置 - {target.stock_code} ({target.stock_name})"),
content=ft.Column([
ft.Container(info, bgcolor='#F5F5F5', padding=10, border_radius=6),
ft.Divider(height=1, color='#e0e0e0'),
*all_rows,
preview,
], spacing=8, tight=True, height=420, scroll=ft.ScrollMode.AUTO),
actions=[ft.TextButton("取消", on_click=lambda e: self._close()),
ft.ElevatedButton("保存", on_click=_save)],
))
def toast(self, msg: str):
self._page.show_dialog(ft.AlertDialog(
title=_text("提示"), content=_text(msg),
actions=[ft.TextButton("确定", on_click=lambda e: self._close())],
))
def _do_start(self, target):
self._close()
target.enabled = True
target.save()
self._data.strategyCtrl[target.get_id()] = SFGridStrategy(target)
self._on_done()
PrintLog(LogLevel.INFO, f'[Flet] 启动交易: {target.targetName()}')
def _do_stop(self, target):
self._close()
target.enabled = False
target.save()
ctrl = self._data.strategyCtrl.pop(target.get_id(), None)
if ctrl: ctrl.enabledTrading(False)
self._on_done()
PrintLog(LogLevel.INFO, f'[Flet] 暂停交易: {target.targetName()}')
def _close(self):
self._page.pop_dialog()
self._page.update()
# ══════════════════════════════════════════════════════════════
# QmtApp v2
# ══════════════════════════════════════════════════════════════
class QmtApp:
"""Flet 版 QMT 交易界面 v2"""
def __init__(self, page: ft.Page):
self.page = page
self.page.title = "神之一手"
self.page.window.width = 1400
self.page.window.height = 800
self.page.padding = 0
self._data = _DataStore()
self._dialogs = _TradeDialogs(page, self._data, self._refresh_ui)
self._grid = _GridPanel(self._data, dialogs=self._dialogs)
self._drawer = _DrawerPanel(self._data, dialogs=self._dialogs)
self._prices_loaded = False
self._run_startup()
# ══════════════════════════════════════════════════════════
# 启动
# ══════════════════════════════════════════════════════════
def _run_startup(self):
splash = _SplashScreen(self.page)
asyncio.ensure_future(self._do_startup(splash))
async def _do_startup(self, splash: _SplashScreen):
await asyncio.sleep(0.05)
steps = [
("正在检查 QMT 环境...", 0.10, lambda: RealQmtV._discover_qmt_port() or True),
("正在初始化交易器...", 0.35, lambda: qmtv.init_qmtv()),
("正在连接 QMT...", 0.55, lambda: qmtv.connect() or True),
("正在加载持仓数据...", 0.75, lambda: self._data.load_from_qmt()),
("正在构建界面...", 0.85, lambda: None),
("正在初始化策略...", 0.92, lambda: self._data.init_strategies()),
]
for text, pct, action in steps:
splash.progress(text, pct)
try:
if action() is False:
self._show_error(f"启动失败: {text}"); return
except Exception as e:
self._show_error(f"启动异常: {text}\n{e}"); return
splash.progress("启动完成", 1.0)
await asyncio.sleep(0.3)
self.page.clean()
self._build_main_ui()
self.page.update()
self._data.pull_prices()
self._data.refresh_orders()
self._data.refresh_trades()
self._refresh_ui()
event_bus.subscribe(MarketDataUpdate, self._on_market_data)
event_bus.subscribe(EventMarketActiveSwitch, self._on_market_active)
event_bus.subscribe(EventTradeTargetUpdate, lambda t: self._refresh_ui())
threading.Thread(target=self._refresh_loop, daemon=True).start()
def _show_error(self, msg: str):
self.page.clean()
self.page.add(ft.Container(
ft.Column([
ft.Icon(ft.Icons.ERROR_OUTLINE, size=48, color=ft.Colors.RED),
_text(msg, size=16),
ft.ElevatedButton("重试", on_click=lambda e: self._retry()),
], alignment=ft.MainAxisAlignment.CENTER, horizontal_alignment=ft.CrossAxisAlignment.CENTER),
alignment=ft.Alignment.CENTER, expand=True,
))
self.page.update()
def _retry(self):
self.page.clean()
self._run_startup()
# ══════════════════════════════════════════════════════════
# UI 构建
# ══════════════════════════════════════════════════════════
def _build_main_ui(self):
self._sidebar_icon = _PanelIcon('sidebar', active=False,
on_click=lambda e: self._toggle_drawer())
self._overlay = self._drawer.build()
title = ft.Container(
ft.Row([
_text("网格策略持仓", bold=True, size=13),
ft.Row([
ft.IconButton(ft.Icons.REFRESH, tooltip="刷新", icon_size=18,
on_click=lambda e: self._manual_refresh()),
self._sidebar_icon,
], spacing=0),
], alignment=ft.MainAxisAlignment.SPACE_BETWEEN),
padding=ft.Padding(10, 10, 10, 5),
)
grid_body = ft.Container(content=self._grid.build(), expand=True,
padding=ft.Padding(10, 0, 10, 10))
self.page.add(ft.Column([
title, ft.Stack([grid_body, self._overlay], expand=True),
], expand=True))
# ══════════════════════════════════════════════════════════
# UI 刷新
# ══════════════════════════════════════════════════════════
def _refresh_ui(self):
self._grid._rebuild()
self._drawer.refresh_all()
self.page.update()
def _manual_refresh(self):
self._data.pull_prices()
self._data.refresh_positions()
self._data.refresh_orders()
self._data.refresh_trades()
self._refresh_ui()
def _toggle_drawer(self):
self._drawer_open = not getattr(self, '_drawer_open', False)
self._overlay.visible = self._drawer_open
self._sidebar_icon.set_active(self._drawer_open)
self.page.update()
def _hide_overlay(self):
self._drawer_open = False
self._overlay.visible = False
self._sidebar_icon.set_active(False)
self.page.update()
# ══════════════════════════════════════════════════════════
# 事件
# ══════════════════════════════════════════════════════════
def _on_market_data(self, data: dict):
count = self._data.apply_tick(data)
if not self._prices_loaded and count > 0:
self._prices_loaded = True
self._refresh_ui()
def _on_market_active(self, is_active: bool):
pass
# ══════════════════════════════════════════════════════════
# 后台刷新
# ══════════════════════════════════════════════════════════
def _refresh_loop(self):
cycle = 0
while True:
time.sleep(5)
cycle += 1
try:
self._data.pull_prices()
self._data.refresh_positions()
if cycle % 6 == 0:
self._data.refresh_orders()
self._data.refresh_trades()
self._refresh_ui()
except Exception:
pass
# ══════════════════════════════════════════════════════════════
# PanelIcon
# ══════════════════════════════════════════════════════════════
class _PanelIcon(ft.Container):
_SIZE = 22; _M = 3
_C = {'bg': '#f0f0f0', 'off': '#b0b0b0', 'on': '#808080', 'active': '#0078d4'}
def __init__(self, kind: str, active: bool = True, on_click=None):
self._kind = kind
self._active = active
super().__init__(content=self._rects(), width=self._SIZE, height=self._SIZE,
bgcolor=self._C['bg'], border_radius=3, ink=True,
on_click=on_click, padding=ft.Padding(self._M, self._M, self._M, self._M))
def _rects(self):
o, n, a = self._C['off'], self._C['on'], self._C['active']
bw, bh = 6, self._SIZE - self._M * 2 - 2
if self._kind == 'sidebar':
return ft.Row([
ft.Container(width=bw, height=bh, bgcolor=n if self._active else o, border_radius=1),
ft.Container(width=2),
ft.Container(width=bw, height=bh, bgcolor=a if self._active else o, border_radius=1),
], spacing=0)
return ft.Column([
ft.Container(width=bh, height=bw, bgcolor=n if self._active else o, border_radius=1),
ft.Container(height=2),
ft.Container(width=bh, height=bw, bgcolor=a if self._active else o, border_radius=1),
], spacing=0)
def set_active(self, active: bool):
self._active = active
self.content = self._rects()
# ══════════════════════════════════════════════════════════════
# 入口
# ══════════════════════════════════════════════════════════════
def main(page: ft.Page):
QmtApp(page)
def run():
ft.app(target=main)