dualea/Experts/Advisors/DualEA/ML/snapshot_export_onnx.py

"""Export an audited universal model through its single native ONNX exporter.

Legacy pickle/XGBoost/status-label artifacts are intentionally unsupported.
The exporter consumes the real-history model_data.json training audit bundle.
"""
from __future__ import annotations
import argparse
import json
from pathlib import Path
import subprocess
import sys

EXPORT_SCRIPT = Path(__file__).resolve().with_name("Export-MarketModel.ps1")
DEFAULT_OUTPUT = Path(__file__).resolve().parent / "artifacts" / "universal"


def build_command(artifacts: str | Path, output_directory: str | Path) -> list[str]:
    return ["powershell.exe", "-NoProfile", "-ExecutionPolicy", "Bypass",
            "-File", str(EXPORT_SCRIPT), "-InputDirectory",
            str(Path(artifacts).expanduser().resolve()), "-OutputDirectory",
            str(Path(output_directory).expanduser().resolve())]


def main(argv: list[str] | None = None) -> int:
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--artifacts", required=True, help="MQL training output directory containing model_data.json")
    parser.add_argument("--output-directory", default=str(DEFAULT_OUTPUT))
    parser.add_argument("--dry-run", action="store_true", help="Print exact argv as JSON without exporting")
    args = parser.parse_args(argv)
    command = build_command(args.artifacts, args.output_directory)
    if args.dry_run:
        print(json.dumps(command, indent=2))
        return 0
    if not EXPORT_SCRIPT.is_file():
        parser.error(f"Native ONNX exporter is missing: {EXPORT_SCRIPT}")
    model_data = Path(args.artifacts).expanduser().resolve() / "model_data.json"
    if not model_data.is_file():
        parser.error(f"Audited trained model_data.json is required: {model_data}")
    if sys.platform != "win32":
        parser.error("This entry point requires Windows PowerShell; no legacy Python export is performed")
    try:
        return subprocess.run(command, check=False).returncode
    except OSError as error:
        parser.exit(1, f"Could not launch the native ONNX exporter: {error}\n")


if __name__ == "__main__":
    raise SystemExit(main())