feat(step1): Universal EA Schema Layer — replaces param_manifest.yaml
NEW FILES: ea/__init__.py - EA module package ea/schema.py - ParameterDef + ParameterSchema (replaces YAML manifest) ea/set_parser.py - Parse ANY MT5 .set file → ParameterSchema ea/registry.py - EAProfile + EARegistry (ea_registry.yaml) ea_registry.yaml - LEGSTECH_EA_V2 pre-registered with automation overrides MODIFIED: mt5/ini_builder.py - Accepts ParameterSchema OR legacy manifest_path mt5/runner.py - Accepts optional EAProfile for EA-agnostic validation config.yaml - Removed hardcoded ea: block; added ea_registry path KEY DESIGN: - .set file IS the manifest: value|min|max|step auto-detects float/int/bool/enum/fixed - automation_overrides in EAProfile forces headless-safe values (InpShowPanel=0) - Phase A optimization INI: Optimization=2 (genetic), ranges from schema - Phase B single backtest: Optimization=0 (unchanged behavior) - LEGSTECH advanced mode: fully backwards compatible via legacy manifest fallback - runner.py still works for legacy callers (no EAProfile needed) TESTED: - 49/49 LEGSTECH params parsed correctly - InpShowPanel=0 automation override applied - INI output correct for both Phase A (Optimization=2) and Phase B (Optimization=0) - All imports pass
This commit is contained in:
@@ -0,0 +1,278 @@
|
||||
"""
|
||||
ea/set_parser.py
|
||||
Parse any MT5 .set file into a ParameterSchema.
|
||||
|
||||
Handles both .set formats:
|
||||
value|min|max|step (single pipe — most common)
|
||||
value||min||max||step||Y/N (double pipe — some MT5 builds)
|
||||
|
||||
Fixed detection:
|
||||
min == max → type="fixed" (e.g. InpMagicNumber=202402|202402|202402|1)
|
||||
min == 0 AND max == 0 → type="fixed" (zeroed range = "don't optimize")
|
||||
No range at all → type="fixed"
|
||||
|
||||
Type detection (non-fixed only):
|
||||
min==0, max==1, step==1 → bool
|
||||
"." in step string → float
|
||||
max - min <= 8, step==1 → enum (small discrete integer set)
|
||||
otherwise → int
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import re
|
||||
from pathlib import Path
|
||||
from typing import Any, Optional
|
||||
|
||||
from loguru import logger
|
||||
|
||||
from ea.schema import ParameterDef, ParameterSchema
|
||||
|
||||
|
||||
# ── Tester section keys to skip (not EA inputs) ───────────────────────────────
|
||||
|
||||
_TESTER_KEYS = {
|
||||
"expert", "symbol", "period", "optimization", "model",
|
||||
"fromdate", "todate", "forwardmode", "report", "replacereport",
|
||||
"shutdownterminal", "deposit", "currency", "leverage",
|
||||
"optimizationmode", "forwarddate", "optimizationiterations",
|
||||
}
|
||||
|
||||
# EA params that should always be fixed even if they have a range
|
||||
_FORCE_FIXED_PATTERNS = [
|
||||
"testermode", "testeri", "testerinit", "showpanel",
|
||||
"magicnumber", "magic",
|
||||
]
|
||||
|
||||
|
||||
class SetParser:
|
||||
"""
|
||||
Parses a MT5 .set file into a ParameterSchema.
|
||||
|
||||
Usage:
|
||||
parser = SetParser()
|
||||
schema = parser.parse(
|
||||
path=Path("C:/MT5 Set files/LEGSTECH_EA_V2.set"),
|
||||
ea_name="LEGSTECH_EA_V2",
|
||||
default_optimize=False, # user chooses via UI
|
||||
)
|
||||
"""
|
||||
|
||||
def parse(
|
||||
self,
|
||||
path: Path,
|
||||
ea_name: str,
|
||||
default_optimize: bool = False,
|
||||
force_optimize: Optional[set[str]] = None,
|
||||
force_fixed: Optional[set[str]] = None,
|
||||
) -> ParameterSchema:
|
||||
"""
|
||||
Parse a .set file and return a ParameterSchema.
|
||||
|
||||
Args:
|
||||
path: Path to the .set file.
|
||||
ea_name: Display name for the EA.
|
||||
default_optimize: Whether to mark all optimizable params as optimize=True by default.
|
||||
If False (default), the user selects via UI.
|
||||
force_optimize: Set of param names that are always optimize=True regardless.
|
||||
force_fixed: Set of param names that are always type="fixed".
|
||||
"""
|
||||
force_optimize = force_optimize or set()
|
||||
force_fixed = force_fixed or set()
|
||||
|
||||
path = Path(path)
|
||||
if not path.exists():
|
||||
raise FileNotFoundError(f".set file not found: {path}")
|
||||
|
||||
try:
|
||||
text = path.read_text(encoding="utf-16")
|
||||
except UnicodeError:
|
||||
text = path.read_text(encoding="utf-8", errors="replace")
|
||||
|
||||
parameters: dict[str, ParameterDef] = {}
|
||||
current_section = ""
|
||||
|
||||
for raw_line in text.splitlines():
|
||||
line = raw_line.strip()
|
||||
if not line or line.startswith(";"):
|
||||
continue
|
||||
|
||||
# Section header
|
||||
if line.startswith("[") and line.endswith("]"):
|
||||
current_section = line[1:-1].lower()
|
||||
continue
|
||||
|
||||
# Skip lines without "="
|
||||
if "=" not in line:
|
||||
continue
|
||||
|
||||
name, _, rest = line.partition("=")
|
||||
name = name.strip()
|
||||
rest = rest.strip()
|
||||
|
||||
# Skip tester-section metadata keys
|
||||
if name.lower() in _TESTER_KEYS:
|
||||
continue
|
||||
|
||||
# Parse the value + optional range
|
||||
param = self._parse_param(name, rest)
|
||||
if param is None:
|
||||
logger.debug(f"SetParser: skipped unrecognised line: {line!r}")
|
||||
continue
|
||||
|
||||
# Apply force-fixed overrides
|
||||
if name in force_fixed or self._is_force_fixed(name):
|
||||
param.type = "fixed"
|
||||
param.optimize = False
|
||||
elif name in force_optimize:
|
||||
param.optimize = True
|
||||
elif default_optimize and param.type != "fixed":
|
||||
param.optimize = True
|
||||
|
||||
parameters[name] = param
|
||||
|
||||
if not parameters:
|
||||
raise ValueError(f"No EA input parameters found in .set file: {path}")
|
||||
|
||||
schema = ParameterSchema(ea_name=ea_name, source_set=path, parameters=parameters)
|
||||
logger.info(f"SetParser: parsed {schema.summary()} from {path.name}")
|
||||
return schema
|
||||
|
||||
# ── Internal ─────────────────────────────────────────────────────────────
|
||||
|
||||
def _parse_param(self, name: str, rest: str) -> Optional[ParameterDef]:
|
||||
"""
|
||||
Parse a single parameter line.
|
||||
rest is everything after the first "=" on the line.
|
||||
"""
|
||||
# Normalise: double-pipe "||" → single "|"
|
||||
rest = re.sub(r"\|\|", "|", rest)
|
||||
|
||||
# Strip trailing Y/N optimize flag if present
|
||||
yn_match = re.search(r"\|([YN])$", rest, re.IGNORECASE)
|
||||
if yn_match:
|
||||
rest = rest[:yn_match.start()]
|
||||
|
||||
parts = [p.strip() for p in rest.split("|")]
|
||||
|
||||
if len(parts) == 1:
|
||||
# No range info → fixed
|
||||
value = self._cast_value(parts[0])
|
||||
return ParameterDef(
|
||||
name=name, default=value, type="fixed",
|
||||
min=None, max=None, step=None, optimize=False,
|
||||
)
|
||||
|
||||
if len(parts) < 4:
|
||||
# Incomplete range — treat as fixed
|
||||
value = self._cast_value(parts[0])
|
||||
return ParameterDef(
|
||||
name=name, default=value, type="fixed",
|
||||
min=None, max=None, step=None, optimize=False,
|
||||
)
|
||||
|
||||
raw_val, raw_min, raw_max, raw_step = parts[0], parts[1], parts[2], parts[3]
|
||||
|
||||
try:
|
||||
default_f = float(raw_val)
|
||||
min_f = float(raw_min)
|
||||
max_f = float(raw_max)
|
||||
step_f = float(raw_step)
|
||||
except ValueError:
|
||||
value = self._cast_value(raw_val)
|
||||
return ParameterDef(
|
||||
name=name, default=value, type="fixed",
|
||||
min=None, max=None, step=None, optimize=False,
|
||||
)
|
||||
|
||||
# Fixed detection
|
||||
is_fixed = (
|
||||
abs(min_f - max_f) < 1e-9 # min == max
|
||||
or (abs(min_f) < 1e-9 and abs(max_f) < 1e-9) # both zero (zeroed range)
|
||||
)
|
||||
if is_fixed:
|
||||
return ParameterDef(
|
||||
name=name,
|
||||
default=self._typed_default(raw_val, raw_step),
|
||||
type="fixed",
|
||||
min=min_f, max=max_f, step=step_f,
|
||||
optimize=False,
|
||||
)
|
||||
|
||||
# Type detection
|
||||
ptype = self._detect_type(min_f, max_f, step_f, raw_step, raw_val)
|
||||
default = self._typed_cast(ptype, default_f, raw_val)
|
||||
|
||||
return ParameterDef(
|
||||
name=name,
|
||||
default=default,
|
||||
type=ptype,
|
||||
min=min_f,
|
||||
max=max_f,
|
||||
step=step_f,
|
||||
optimize=False, # user sets this via UI; can be overridden by caller
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
def _detect_type(min_f: float, max_f: float, step_f: float,
|
||||
raw_step: str, raw_val: str) -> str:
|
||||
"""Infer parameter type from its range."""
|
||||
# Bool: exactly 0–1 with step 1
|
||||
if abs(min_f) < 1e-9 and abs(max_f - 1.0) < 1e-9 and abs(step_f - 1.0) < 1e-9:
|
||||
return "bool"
|
||||
|
||||
# Float: step has decimal component
|
||||
if "." in raw_step and not raw_step.endswith(".0") and float(raw_step) % 1 != 0:
|
||||
return "float"
|
||||
# Also float if default value has meaningful decimal
|
||||
if "." in raw_val and float(raw_val) % 1 != 0:
|
||||
return "float"
|
||||
|
||||
# Enum: small integer set (≤ 8 distinct values, step 1)
|
||||
n_values = int(round((max_f - min_f) / step_f)) + 1 if step_f > 0 else 1
|
||||
if abs(step_f - 1.0) < 1e-9 and n_values <= 8:
|
||||
return "enum"
|
||||
|
||||
return "int"
|
||||
|
||||
@staticmethod
|
||||
def _typed_cast(ptype: str, value_f: float, raw: str) -> Any:
|
||||
if ptype == "bool":
|
||||
return value_f != 0 or raw.lower() in ("true", "1")
|
||||
if ptype == "int":
|
||||
return int(round(value_f))
|
||||
if ptype == "enum":
|
||||
return int(round(value_f))
|
||||
return value_f # float
|
||||
|
||||
@staticmethod
|
||||
def _typed_default(raw: str, raw_step: str) -> Any:
|
||||
"""Cast a fixed-param value without range context."""
|
||||
lower = raw.lower()
|
||||
if lower in ("true", "false"):
|
||||
return lower == "true"
|
||||
try:
|
||||
f = float(raw)
|
||||
# Return int if it's a whole number and step is integer-like
|
||||
if "." not in raw_step or raw_step.endswith(".0"):
|
||||
if f == int(f):
|
||||
return int(f)
|
||||
return f
|
||||
except ValueError:
|
||||
return raw
|
||||
|
||||
@staticmethod
|
||||
def _cast_value(raw: str) -> Any:
|
||||
lower = raw.lower()
|
||||
if lower in ("true", "false"):
|
||||
return lower == "true"
|
||||
try:
|
||||
f = float(raw)
|
||||
return int(f) if f == int(f) and "." not in raw else f
|
||||
except ValueError:
|
||||
return raw
|
||||
|
||||
@staticmethod
|
||||
def _is_force_fixed(name: str) -> bool:
|
||||
"""Return True for params that are always fixed regardless of their range."""
|
||||
lower = name.lower()
|
||||
return any(pat in lower for pat in _FORCE_FIXED_PATTERNS)
|
||||
Reference in New Issue
Block a user