106 lines
3.7 KiB
Python
106 lines
3.7 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
r"""平台依赖检查（矩阵 102 依赖矩阵部分）。
|
|||
|
|
|
|||
|
|
检测当前环境：python 版本、平台/架构、requirements.txt 全部依赖可导入性与版本下限。
|
|||
|
|
用法：.venv\Scripts\python.exe scripts/platform-deps-check.py （退出码 0=通过）
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import importlib
|
|||
|
|
import platform
|
|||
|
|
import re
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|||
|
|
|
|||
|
|
_MIN_PYTHON = (3, 11)
|
|||
|
|
_REQUIRE_PATTERN = re.compile(r"^([A-Za-z0-9_\-\.]+)(?:\[[^\]]*\])?(>=|==|~=|<=)?([0-9][0-9.]*)?")
|
|||
|
|
|
|||
|
|
# 平台敏感依赖：wheel 覆盖性标注（文档见 docs/architecture/platform-matrix.md）
|
|||
|
|
_PLATFORM_SENSITIVE = {"ortools", "cryptography", "uvicorn"}
|
|||
|
|
|
|||
|
|
# pip 包名 → 实际导入模块名（不一致映射）
|
|||
|
|
_IMPORT_NAME_MAP = {
|
|||
|
|
"python-dotenv": "dotenv",
|
|||
|
|
"python-docx": "docx",
|
|||
|
|
"python-multipart": "multipart",
|
|||
|
|
"pyyaml": "yaml",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _check_python() -> list[str]:
|
|||
|
|
problems = []
|
|||
|
|
if sys.version_info < _MIN_PYTHON:
|
|||
|
|
problems.append(f"python {sys.version_info.major}.{sys.version_info.minor} < 3.11")
|
|||
|
|
return problems
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _check_platform() -> dict:
|
|||
|
|
machine = platform.machine().lower()
|
|||
|
|
arch = "aarch64" if machine in ("aarch64", "arm64") else "x86_64" if machine in ("amd64", "x86_64") else machine
|
|||
|
|
system = platform.system()
|
|||
|
|
return {"system": system, "arch": arch, "platform": platform.platform()}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _parse_requirements() -> list[tuple[str, str | None, str | None]]:
|
|||
|
|
req = ROOT / "requirements.txt"
|
|||
|
|
out: list[tuple[str, str | None, str | None]] = []
|
|||
|
|
if not req.exists():
|
|||
|
|
return out
|
|||
|
|
for line in req.read_text(encoding="utf-8").splitlines():
|
|||
|
|
line = line.split("#", 1)[0].strip()
|
|||
|
|
if not line:
|
|||
|
|
continue
|
|||
|
|
m = _REQUIRE_PATTERN.match(line)
|
|||
|
|
if m:
|
|||
|
|
out.append((m.group(1), m.group(2), m.group(3)))
|
|||
|
|
return out
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _check_deps() -> tuple[list[str], list[dict]]:
|
|||
|
|
problems: list[str] = []
|
|||
|
|
detail: list[dict] = []
|
|||
|
|
for name, op, ver in _parse_requirements():
|
|||
|
|
import_name = _IMPORT_NAME_MAP.get(name, name.replace("-", "_"))
|
|||
|
|
try:
|
|||
|
|
mod = importlib.import_module(import_name)
|
|||
|
|
except ImportError:
|
|||
|
|
try:
|
|||
|
|
mod = importlib.import_module(name)
|
|||
|
|
except ImportError:
|
|||
|
|
problems.append(f"依赖缺失: {name}")
|
|||
|
|
detail.append({"name": name, "ok": False, "sensitive": name in _PLATFORM_SENSITIVE})
|
|||
|
|
continue
|
|||
|
|
version = getattr(mod, "__version__", "?")
|
|||
|
|
ok = True
|
|||
|
|
if ver and op == ">=":
|
|||
|
|
try:
|
|||
|
|
import packaging.version
|
|||
|
|
ok = packaging.version.parse(str(version)) >= packaging.version.parse(ver)
|
|||
|
|
except Exception:
|
|||
|
|
ok = True # 无 packaging 或版本解析失败时不阻断
|
|||
|
|
detail.append({"name": name, "version": str(version), "ok": ok, "sensitive": name in _PLATFORM_SENSITIVE})
|
|||
|
|
if not ok:
|
|||
|
|
problems.append(f"版本不满足: {name} {version} < {ver}")
|
|||
|
|
return problems, detail
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main() -> int:
|
|||
|
|
problems = _check_python()
|
|||
|
|
platform_info = _check_platform()
|
|||
|
|
dep_problems, detail = _check_deps()
|
|||
|
|
print(f"python: {sys.version.split()[0]} | system: {platform_info['system']} | arch: {platform_info['arch']}")
|
|||
|
|
print(f"deps: {sum(1 for d in detail if d['ok'])}/{len(detail)} ok | sensitive: "
|
|||
|
|
f"{[d['name'] for d in detail if d['sensitive'] and d['ok']]}")
|
|||
|
|
all_problems = problems + dep_problems
|
|||
|
|
for p in all_problems:
|
|||
|
|
print("FAIL:", p)
|
|||
|
|
if not all_problems:
|
|||
|
|
print("platform-deps-check: PASS")
|
|||
|
|
return 1 if all_problems else 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
sys.exit(main())
|