347 lines
14 KiB
Python
347 lines
14 KiB
Python
|
# version: v6
|
|||
|
|
"""
|
|||
|
|
mt5_routes.py — MT5 Connection Management API
|
|||
|
|
================================================
|
|||
|
|
Endpoints for the consumer Settings → Providers panel.
|
|||
|
|
|
|||
|
|
GET /api/mt5/config → read MT5 connection settings
|
|||
|
|
PATCH /api/mt5/config → save MT5 connection settings
|
|||
|
|
POST /api/mt5/connect → connect to MT5 with current config
|
|||
|
|
POST /api/mt5/disconnect → disconnect from MT5
|
|||
|
|
|
|||
|
|
Settings are stored in user_config.json under providers.accounts.mt5_default.
|
|||
|
|
This module does NOT introduce any calculation triggers.
|
|||
|
|
|
|||
|
|
v2 — PATCH /config now disconnects and reinitializes the MT5 provider after
|
|||
|
|
saving, so the next /connect picks up the new terminal_path / server /
|
|||
|
|
login immediately. Previously the provider cached the old path in memory
|
|||
|
|
and stale connects kept firing until the backend was restarted — e.g.
|
|||
|
|
user couldn't revert an invalid GoDo path back to auto-detect.
|
|||
|
|
|
|||
|
|
v3 — Added `use_tick_data` to the GET/PATCH config payload. When True, the
|
|||
|
|
backend tick loop polls MT5 every 0.2s instead of 1.0s, giving the chart
|
|||
|
|
a live MT5-style "tick feel" on the last candle. Default OFF. Toggle
|
|||
|
|
surfaces in Settings → PROVIDERS.
|
|||
|
|
|
|||
|
|
v4 — PATCH /config no longer disconnects + reinits the MT5 provider on every
|
|||
|
|
change. Reinit now triggers ONLY when a connection-relevant field
|
|||
|
|
(connection_method / terminal_path / server / login / password) is
|
|||
|
|
touched. Flipping `use_tick_data` (and any other future cosmetic field)
|
|||
|
|
leaves the live connection alone. Without this, toggling USE TICK DATA
|
|||
|
|
in the UI caused a ~5s outage with a flood of 503 responses while the
|
|||
|
|
reconnect loop healed the connection.
|
|||
|
|
|
|||
|
|
v5 — POST /browse-terminal opens a native Windows file-picker so users with
|
|||
|
|
multiple MT5 installations can select the correct terminal64.exe without
|
|||
|
|
manually typing (or pasting) a long path. Runs tkinter.filedialog on a
|
|||
|
|
worker thread, returns the chosen absolute path.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
from fastapi import APIRouter, Request, HTTPException
|
|||
|
|
|
|||
|
|
log = logging.getLogger("mt5_routes")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def create_mt5_router(cfg_manager, app) -> APIRouter:
|
|||
|
|
router = APIRouter(prefix="/api/mt5", tags=["mt5"])
|
|||
|
|
|
|||
|
|
def _get_mt5_config() -> dict:
|
|||
|
|
"""Read MT5 config from user_config providers section."""
|
|||
|
|
config = cfg_manager.get_config()
|
|||
|
|
providers = config.get("providers", {})
|
|||
|
|
accounts = providers.get("accounts", {})
|
|||
|
|
mt5_acct = accounts.get("mt5_default", {})
|
|||
|
|
return {
|
|||
|
|
"connection_method": mt5_acct.get("connection_method", "path"),
|
|||
|
|
"terminal_path": mt5_acct.get("terminal_path") or "",
|
|||
|
|
"server": mt5_acct.get("server", ""),
|
|||
|
|
"login": mt5_acct.get("login", ""),
|
|||
|
|
"password": "", # Never send password back
|
|||
|
|
"enabled": mt5_acct.get("enabled", True),
|
|||
|
|
# v3: USE TICK DATA — when ON, tick loop polls MT5 5×/sec instead
|
|||
|
|
# of 1×/sec. Default OFF; user opts in via Settings → PROVIDERS.
|
|||
|
|
"use_tick_data": bool(mt5_acct.get("use_tick_data", False)),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# ── GET /api/mt5/config ──
|
|||
|
|
@router.get("/config")
|
|||
|
|
async def get_mt5_config():
|
|||
|
|
return _get_mt5_config()
|
|||
|
|
|
|||
|
|
# ── PATCH /api/mt5/config ──
|
|||
|
|
@router.patch("/config")
|
|||
|
|
async def patch_mt5_config(request: Request):
|
|||
|
|
import asyncio
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
patch = await request.json()
|
|||
|
|
except Exception:
|
|||
|
|
raise HTTPException(400, "Invalid JSON")
|
|||
|
|
|
|||
|
|
# Build the nested update for providers.accounts.mt5_default
|
|||
|
|
mt5_update = {}
|
|||
|
|
for key in ["connection_method", "terminal_path", "server", "login", "password",
|
|||
|
|
"use_tick_data"]: # v3
|
|||
|
|
if key in patch:
|
|||
|
|
mt5_update[key] = patch[key]
|
|||
|
|
|
|||
|
|
if mt5_update:
|
|||
|
|
cfg_manager.update({
|
|||
|
|
"providers": {
|
|||
|
|
"accounts": {
|
|||
|
|
"mt5_default": mt5_update
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
log.info(f"MT5 config updated: {list(mt5_update.keys())}")
|
|||
|
|
|
|||
|
|
# v4: reinit only when a connection-relevant field changed.
|
|||
|
|
# use_tick_data is read live by focus_tick_loop on every iteration
|
|||
|
|
# — flipping it must NOT bounce the MT5 connection (5s 503 outage).
|
|||
|
|
CONNECTION_FIELDS = {"connection_method", "terminal_path",
|
|||
|
|
"server", "login", "password"}
|
|||
|
|
if CONNECTION_FIELDS.intersection(mt5_update.keys()):
|
|||
|
|
try:
|
|||
|
|
prov = cfg_manager.get_provider("mt5_default")
|
|||
|
|
if prov is not None:
|
|||
|
|
try:
|
|||
|
|
if getattr(prov, "connected", False):
|
|||
|
|
await asyncio.to_thread(prov.disconnect)
|
|||
|
|
except Exception as e:
|
|||
|
|
log.warning(f"MT5 disconnect during reinit failed: {e}")
|
|||
|
|
await asyncio.to_thread(cfg_manager.init_providers)
|
|||
|
|
log.info("MT5 provider reinitialized with new config")
|
|||
|
|
except Exception as e:
|
|||
|
|
log.warning(f"MT5 provider reinit failed: {e}")
|
|||
|
|
|
|||
|
|
return {"config": _get_mt5_config()}
|
|||
|
|
|
|||
|
|
# ── POST /api/mt5/browse-terminal ──
|
|||
|
|
@router.post("/browse-terminal")
|
|||
|
|
async def browse_terminal():
|
|||
|
|
"""v5: Open a native OS file picker for the user to select terminal64.exe.
|
|||
|
|
Runs tkinter.filedialog on a worker thread (FastAPI's async loop can't
|
|||
|
|
block on a dialog). Returns {"path": "..."} on pick, {"path": null,
|
|||
|
|
"cancelled": true} on cancel. Windows-only for now — defaults to the
|
|||
|
|
most likely MT5 install dir if nothing is configured yet."""
|
|||
|
|
import asyncio
|
|||
|
|
import os
|
|||
|
|
import platform
|
|||
|
|
|
|||
|
|
if platform.system() != "Windows":
|
|||
|
|
raise HTTPException(501, "Browse dialog is Windows-only for now.")
|
|||
|
|
|
|||
|
|
# Pick a sensible initial directory: current configured path → its dir,
|
|||
|
|
# or %ProgramFiles% as a fallback.
|
|||
|
|
current = _get_mt5_config().get("terminal_path") or ""
|
|||
|
|
if current and os.path.isfile(current):
|
|||
|
|
initial_dir = os.path.dirname(current)
|
|||
|
|
elif current and os.path.isdir(current):
|
|||
|
|
initial_dir = current
|
|||
|
|
else:
|
|||
|
|
pf = os.environ.get("ProgramFiles", r"C:\Program Files")
|
|||
|
|
initial_dir = pf
|
|||
|
|
|
|||
|
|
def _pick() -> dict:
|
|||
|
|
# v6: switched tkinter → native comdlg32 via ctypes.
|
|||
|
|
# tkinter is not bundled in the PyInstaller build (strips by
|
|||
|
|
# default unless explicitly hidden-imported), so the installed
|
|||
|
|
# .exe would fail with "No module named 'tkinter'". comdlg32.dll
|
|||
|
|
# is core Windows — always available, no dependencies, no build
|
|||
|
|
# change needed. Still runs on a worker thread because the native
|
|||
|
|
# call blocks until the user picks or cancels.
|
|||
|
|
try:
|
|||
|
|
import ctypes
|
|||
|
|
from ctypes import wintypes, Structure, byref, sizeof, c_void_p
|
|||
|
|
|
|||
|
|
class OPENFILENAMEW(Structure):
|
|||
|
|
_fields_ = [
|
|||
|
|
("lStructSize", wintypes.DWORD),
|
|||
|
|
("hwndOwner", wintypes.HWND),
|
|||
|
|
("hInstance", wintypes.HINSTANCE),
|
|||
|
|
("lpstrFilter", wintypes.LPCWSTR),
|
|||
|
|
("lpstrCustomFilter",wintypes.LPWSTR),
|
|||
|
|
("nMaxCustFilter", wintypes.DWORD),
|
|||
|
|
("nFilterIndex", wintypes.DWORD),
|
|||
|
|
("lpstrFile", wintypes.LPWSTR),
|
|||
|
|
("nMaxFile", wintypes.DWORD),
|
|||
|
|
("lpstrFileTitle", wintypes.LPWSTR),
|
|||
|
|
("nMaxFileTitle", wintypes.DWORD),
|
|||
|
|
("lpstrInitialDir", wintypes.LPCWSTR),
|
|||
|
|
("lpstrTitle", wintypes.LPCWSTR),
|
|||
|
|
("Flags", wintypes.DWORD),
|
|||
|
|
("nFileOffset", wintypes.WORD),
|
|||
|
|
("nFileExtension", wintypes.WORD),
|
|||
|
|
("lpstrDefExt", wintypes.LPCWSTR),
|
|||
|
|
("lCustData", wintypes.LPARAM),
|
|||
|
|
("lpfnHook", c_void_p),
|
|||
|
|
("lpTemplateName", wintypes.LPCWSTR),
|
|||
|
|
("pvReserved", c_void_p),
|
|||
|
|
("dwReserved", wintypes.DWORD),
|
|||
|
|
("FlagsEx", wintypes.DWORD),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
OFN_FILEMUSTEXIST = 0x00001000
|
|||
|
|
OFN_PATHMUSTEXIST = 0x00000800
|
|||
|
|
OFN_EXPLORER = 0x00080000
|
|||
|
|
OFN_NOCHANGEDIR = 0x00000008
|
|||
|
|
|
|||
|
|
buf = ctypes.create_unicode_buffer(4096)
|
|||
|
|
ofn = OPENFILENAMEW()
|
|||
|
|
ctypes.memset(byref(ofn), 0, sizeof(ofn))
|
|||
|
|
ofn.lStructSize = sizeof(OPENFILENAMEW)
|
|||
|
|
ofn.lpstrFile = ctypes.cast(buf, wintypes.LPWSTR)
|
|||
|
|
ofn.nMaxFile = 4096
|
|||
|
|
# Filter string: pairs of "label\0pattern\0" terminated with an extra \0.
|
|||
|
|
ofn.lpstrFilter = (
|
|||
|
|
"MT5 terminal\0terminal64.exe;terminal.exe\0"
|
|||
|
|
"Executables\0*.exe\0"
|
|||
|
|
"All files\0*.*\0\0"
|
|||
|
|
)
|
|||
|
|
ofn.lpstrInitialDir = initial_dir
|
|||
|
|
ofn.lpstrTitle = "Select MT5 terminal (terminal64.exe)"
|
|||
|
|
ofn.Flags = (OFN_FILEMUSTEXIST | OFN_PATHMUSTEXIST
|
|||
|
|
| OFN_EXPLORER | OFN_NOCHANGEDIR)
|
|||
|
|
|
|||
|
|
ok = ctypes.windll.comdlg32.GetOpenFileNameW(byref(ofn))
|
|||
|
|
if not ok:
|
|||
|
|
# User cancelled OR dialog failed. CommDlgExtendedError
|
|||
|
|
# returns 0 for cancel, non-zero for actual errors.
|
|||
|
|
err = ctypes.windll.comdlg32.CommDlgExtendedError()
|
|||
|
|
if err != 0:
|
|||
|
|
return {"path": None, "error": f"comdlg32 error {err:#x}"}
|
|||
|
|
return {"path": None, "cancelled": True}
|
|||
|
|
|
|||
|
|
picked = buf.value
|
|||
|
|
if not picked:
|
|||
|
|
return {"path": None, "cancelled": True}
|
|||
|
|
return {"path": os.path.normpath(picked), "cancelled": False}
|
|||
|
|
except Exception as e:
|
|||
|
|
log.error(f"browse-terminal dialog failed: {e}")
|
|||
|
|
return {"path": None, "error": str(e)}
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
result = await asyncio.to_thread(_pick)
|
|||
|
|
except Exception as e:
|
|||
|
|
raise HTTPException(500, f"Dialog launch failed: {e}")
|
|||
|
|
if result.get("error"):
|
|||
|
|
raise HTTPException(500, result["error"])
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ── POST /api/mt5/connect ──
|
|||
|
|
@router.post("/connect")
|
|||
|
|
async def connect_mt5(request: Request):
|
|||
|
|
import asyncio
|
|||
|
|
|
|||
|
|
provider = cfg_manager.get_provider("mt5_default")
|
|||
|
|
if provider is None:
|
|||
|
|
# Try to init providers first
|
|||
|
|
try:
|
|||
|
|
await asyncio.to_thread(cfg_manager.init_providers)
|
|||
|
|
provider = cfg_manager.get_provider("mt5_default")
|
|||
|
|
except Exception as e:
|
|||
|
|
raise HTTPException(500, f"Failed to initialize MT5 provider: {e}")
|
|||
|
|
|
|||
|
|
if provider is None:
|
|||
|
|
raise HTTPException(500, "MT5 provider not available")
|
|||
|
|
|
|||
|
|
# If already connected, return current state
|
|||
|
|
if provider.connected:
|
|||
|
|
account = None
|
|||
|
|
try:
|
|||
|
|
import MetaTrader5 as mt5
|
|||
|
|
info = mt5.account_info()
|
|||
|
|
if info:
|
|||
|
|
account = {
|
|||
|
|
"login": info.login,
|
|||
|
|
"server": info.server,
|
|||
|
|
"balance": info.balance,
|
|||
|
|
"equity": info.equity,
|
|||
|
|
"currency": info.currency,
|
|||
|
|
}
|
|||
|
|
except Exception:
|
|||
|
|
pass
|
|||
|
|
return {"connected": True, "account": account}
|
|||
|
|
|
|||
|
|
# Try to connect
|
|||
|
|
try:
|
|||
|
|
connected = await asyncio.wait_for(
|
|||
|
|
asyncio.to_thread(provider.connect),
|
|||
|
|
timeout=15.0,
|
|||
|
|
)
|
|||
|
|
except asyncio.TimeoutError:
|
|||
|
|
return {"connected": False, "error": "Connection timed out (15s)"}
|
|||
|
|
except Exception as e:
|
|||
|
|
return {"connected": False, "error": str(e)}
|
|||
|
|
|
|||
|
|
if not connected:
|
|||
|
|
return {"connected": False, "error": "MT5 connection failed — check terminal path or credentials"}
|
|||
|
|
|
|||
|
|
# Update app state
|
|||
|
|
app.state.provider = provider
|
|||
|
|
|
|||
|
|
# Resolve symbols
|
|||
|
|
try:
|
|||
|
|
universe = cfg_manager.get_active_universe()
|
|||
|
|
await asyncio.to_thread(provider.resolve_universe, universe)
|
|||
|
|
except Exception:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
# Get account info
|
|||
|
|
account = None
|
|||
|
|
try:
|
|||
|
|
import MetaTrader5 as mt5
|
|||
|
|
info = mt5.account_info()
|
|||
|
|
if info:
|
|||
|
|
account = {
|
|||
|
|
"login": info.login,
|
|||
|
|
"server": info.server,
|
|||
|
|
"balance": info.balance,
|
|||
|
|
"equity": info.equity,
|
|||
|
|
"currency": info.currency,
|
|||
|
|
}
|
|||
|
|
except Exception:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
log.info("MT5 connected via settings panel")
|
|||
|
|
return {"connected": True, "account": account}
|
|||
|
|
|
|||
|
|
# ── POST /api/mt5/disconnect ──
|
|||
|
|
@router.post("/disconnect")
|
|||
|
|
async def disconnect_mt5():
|
|||
|
|
import asyncio
|
|||
|
|
provider = cfg_manager.get_provider("mt5_default")
|
|||
|
|
if provider and provider.connected:
|
|||
|
|
try:
|
|||
|
|
await asyncio.to_thread(provider.disconnect)
|
|||
|
|
log.info("MT5 disconnected via settings panel")
|
|||
|
|
except Exception as e:
|
|||
|
|
log.warning(f"MT5 disconnect error: {e}")
|
|||
|
|
return {"connected": False}
|
|||
|
|
|
|||
|
|
# ── GET /api/mt5/status ──
|
|||
|
|
@router.get("/status")
|
|||
|
|
async def get_mt5_status():
|
|||
|
|
"""Quick status check — used by settings panel."""
|
|||
|
|
provider = cfg_manager.get_provider("mt5_default")
|
|||
|
|
connected = provider.connected if provider else False
|
|||
|
|
account = None
|
|||
|
|
if connected:
|
|||
|
|
try:
|
|||
|
|
import MetaTrader5 as mt5
|
|||
|
|
info = mt5.account_info()
|
|||
|
|
if info:
|
|||
|
|
account = {
|
|||
|
|
"login": info.login,
|
|||
|
|
"server": info.server,
|
|||
|
|
"balance": info.balance,
|
|||
|
|
"equity": info.equity,
|
|||
|
|
}
|
|||
|
|
except Exception:
|
|||
|
|
pass
|
|||
|
|
return {"connected": connected, "account": account}
|
|||
|
|
|
|||
|
|
return router
|