gjm
6143bb9ab2
支持直接审查 PDF 与位图图纸，全链路离线（不连 CAD、不连 8080 端口）。 渲染在内存内完成，复用既有墨迹热力图 + 自适应瓦片规划。 - 新增 dwg_review/sources.py - detect_source / plan_file / collect 统一入口 - PdfReader：pypdfium2 惰性导入，页序即图框序，处理 /Rotate 与 crop 边距 - ImageReader：世界单位 = 源像素，snap_box 整数对齐保证窗口与像素严格一致 - ink_grid：纯 PIL 灰度阈值 → 占用网格（不引入 numpy） - ReviewContext.create(connect_cad=False)：离线时 client/tools 均为 None - 全链路 ctx.tools is None 守卫：review 只读子循环 / inspect_region / annotator / checks 的窗口类检查 / 绘图区最大化 - checklist.build_system_prompt(allow_tools=False, offline=True)：离线提示词 - __main__：位置参数 source，按扩展名自动判定源类型并决定是否连 CAD - config 新增 8 个常量（PDF_MAX_PAGES / 文本最小字数 / 各源缩放上限 / 墨迹阈值等） - tiling.content_predicate 与 heatmap.cell_box 提升为公开函数 验证： - pytest 457 passed - verify_pdf_source.py：PDF 23/23、图片 9/9（不需 CAD），含实测瓦片字高像素与 旋转映射自检（文本区墨迹 0.311 vs 忽略 /Rotate 反向对照 0.145） - 真实 LLM 端到端：20 页 HVAC 图纸前 3 页跑通，结论具体可核验且未编造 handle 已知局限：页边线在墨迹网格里会把整页连成单簇，瓦片常退化为整页象限（CAD 侧 靠 drop_oversized 丢图框块，离线侧暂无等价手段），留待后续打磨。
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
"""CLI 入口（`dwg_review.__main__`）参数解析与来源判定单测。
|
|
|
|
只测**纯逻辑**（不真连 CAD、不真调 LLM）：位置参数 → (source, source_path) 的判定，
|
|
以及 CAD-only 选项在离线源下被拒绝。
|
|
"""
|
|
|
|
from PIL import Image
|
|
|
|
from dwg_review.__main__ import _resolve_source_arg, build_parser
|
|
|
|
|
|
def test_parser_accepts_positional_source():
|
|
args = build_parser().parse_args(["a.pdf"])
|
|
assert args.source == "a.pdf"
|
|
|
|
|
|
def test_parser_source_optional():
|
|
assert build_parser().parse_args([]).source is None
|
|
|
|
|
|
def test_parser_keeps_existing_flags():
|
|
args = build_parser().parse_args(["a.pdf", "--plan-only", "--frames", "3"])
|
|
assert args.plan_only is True and args.frames == 3
|
|
|
|
|
|
def test_resolve_no_source_is_cad_mode():
|
|
assert _resolve_source_arg(None) == (None, None)
|
|
|
|
|
|
def test_resolve_pdf(tmp_path):
|
|
path = tmp_path / "d.pdf"
|
|
path.write_bytes(b"%PDF-1.4\n")
|
|
resolved, err = _resolve_source_arg(str(path))
|
|
assert err is None
|
|
assert resolved[0] == "pdf"
|
|
assert resolved[1].endswith("d.pdf")
|
|
|
|
|
|
def test_resolve_image(tmp_path):
|
|
path = tmp_path / "d.png"
|
|
Image.new("RGB", (10, 10)).save(path)
|
|
resolved, err = _resolve_source_arg(str(path))
|
|
assert err is None and resolved[0] == "image"
|
|
|
|
|
|
def test_resolve_dwg_falls_back_to_cad_mode(capsys):
|
|
"""DWG 参数无法在 Python 侧打开 → 提示后按 CAD 在线模式运行（source=None）。"""
|
|
resolved, err = _resolve_source_arg("x.dwg")
|
|
assert resolved is None and err is None
|
|
assert "CAD 在线模式" in capsys.readouterr().out
|
|
|
|
|
|
def test_resolve_missing_file_errors(tmp_path):
|
|
resolved, err = _resolve_source_arg(str(tmp_path / "nope.pdf"))
|
|
assert resolved is None and "文件不存在" in err
|
|
|
|
|
|
def test_resolve_unknown_extension_errors():
|
|
resolved, err = _resolve_source_arg("x.svg")
|
|
assert resolved is None and "无法识别" in err
|