93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
|
"""编译完跑一遍：把 build/ 下的伴生包 .pyd 挂到 sys.modules，
|
|||
|
|
再用真实接口（openpyxl.load_workbook）写读一个最小 Excel，确认链路可用。
|
|||
|
|
|
|||
|
|
用法：
|
|||
|
|
python tools/verify_companions.py
|
|||
|
|
|
|||
|
|
注意：本脚本直接 import mil.core.companion_loader，**不**经过 mil/__init__.py，
|
|||
|
|
避免在无 PySide6 的开发机上因 UI 桥接而炸（生产宿主环境必然有 PySide6）。
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import argparse
|
|||
|
|
import importlib.util
|
|||
|
|
import sys
|
|||
|
|
import tempfile
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
ROOT = Path(__file__).resolve().parent.parent
|
|||
|
|
sys.path.insert(0, str(ROOT))
|
|||
|
|
|
|||
|
|
# 不走 mil/__init__.py，单独加载 companion_loader
|
|||
|
|
spec = importlib.util.spec_from_file_location(
|
|||
|
|
"mil_companion_loader", ROOT / "mil" / "core" / "companion_loader.py"
|
|||
|
|
)
|
|||
|
|
companion_loader = importlib.util.module_from_spec(spec)
|
|||
|
|
spec.loader.exec_module(companion_loader)
|
|||
|
|
load_companions = companion_loader.load_companions
|
|||
|
|
|
|||
|
|
# 伴生包清单：与 mil/__init__.py 中的 COMPANION_PACKAGES 保持一致
|
|||
|
|
COMPANION_PACKAGES: list[str] = [
|
|||
|
|
"openpyxl",
|
|||
|
|
"et_xmlfile",
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main() -> int:
|
|||
|
|
parser = argparse.ArgumentParser(description="验证伴生包 .pyd 加载链路")
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--fallback",
|
|||
|
|
action="store_true",
|
|||
|
|
help="未发现产物时回退到 site-packages 已装版本（开发自测用）",
|
|||
|
|
)
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
loaded = load_companions(COMPANION_PACKAGES)
|
|||
|
|
print(f"[verify] 加载器返回: {loaded}")
|
|||
|
|
|
|||
|
|
if not loaded and not args.fallback:
|
|||
|
|
print(
|
|||
|
|
"[verify] FAIL: 没有挂载到任何伴生包（是否忘了 python build_pyd.py？）\n"
|
|||
|
|
" 调试期可加 --fallback 走 site-packages 自测加载器逻辑。",
|
|||
|
|
file=sys.stderr,
|
|||
|
|
)
|
|||
|
|
return 1
|
|||
|
|
|
|||
|
|
if not loaded and args.fallback:
|
|||
|
|
# 开发自测：build/mil_runtime 都没有时，从 sys.path 兜底
|
|||
|
|
import importlib
|
|||
|
|
for pkg in COMPANION_PACKAGES:
|
|||
|
|
importlib.import_module(pkg)
|
|||
|
|
loaded.append(pkg)
|
|||
|
|
print(f"[verify] fallback 后挂载: {loaded}")
|
|||
|
|
|
|||
|
|
# 用 openpyxl 真接口验证：写一个最小 xlsx，立刻读回来
|
|||
|
|
import openpyxl # noqa: WPS433 延迟导入以验证挂载生效
|
|||
|
|
print(f"[verify] openpyxl 来自: {openpyxl.__file__}")
|
|||
|
|
|
|||
|
|
with tempfile.TemporaryDirectory() as td:
|
|||
|
|
xlsx_path = Path(td) / "smoke.xlsx"
|
|||
|
|
|
|||
|
|
wb = openpyxl.Workbook()
|
|||
|
|
ws = wb.active
|
|||
|
|
ws["A1"] = "time"
|
|||
|
|
ws["B1"] = "sig1"
|
|||
|
|
ws["A2"] = 0.0
|
|||
|
|
ws["B2"] = 1
|
|||
|
|
ws["A3"] = 1.0
|
|||
|
|
ws["B3"] = 2
|
|||
|
|
wb.save(xlsx_path)
|
|||
|
|
|
|||
|
|
wb2 = openpyxl.load_workbook(xlsx_path)
|
|||
|
|
ws2 = wb2.active
|
|||
|
|
assert ws2["A1"].value == "time", ws2["A1"].value
|
|||
|
|
assert ws2["B1"].value == "sig1", ws2["B1"].value
|
|||
|
|
assert float(ws2["A3"].value) == 1.0, ws2["A3"].value
|
|||
|
|
assert int(ws2["B3"].value) == 2, ws2["B3"].value
|
|||
|
|
|
|||
|
|
print("[verify] PASS: openpyxl 写读闭环 ✓")
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
sys.exit(main())
|