新增: STEP批量转GLB脚本(网页3D轻量化构建期工具, 待真实数据就位后启用)

occt-service/batch-convert-glb.py

@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+"""
+批量转换：把 catalog\step\ 下所有 STEP 转成 glTF/GLB（网页 3D 轻量化用）。
+
+用法：
+  python batch-convert-glb.py <step目录> <gltf输出目录> [deflection]
+
+说明：
+  - 走本地 OCCT 服务的 /convert 接口（需先启动 occt_service.py）
+  - 同名 .step → .glb，断点续传（已存在且非空的 .glb 跳过）
+  - 单线程、逐文件，避免给服务造成压力
+  - 默认 deflection=0.5（屏幕预览够用）
+"""
+import os
+import sys
+import time
+import urllib.request
+
+OCCT_URL = os.environ.get("OCCT_URL", "http://localhost:8090")
+
+
+def convert_one(step_path, glb_path, deflection):
+    with open(step_path, "rb") as f:
+        data = f.read()
+    req = urllib.request.Request(
+        "%s/convert?deflection=%s" % (OCCT_URL, deflection),
+        data=data, method="POST",
+        headers={"Content-Type": "application/octet-stream"},
+    )
+    resp = urllib.request.urlopen(req, timeout=120)
+    out = resp.read()
+    if not out or out[:4] != b"glTF":
+        raise RuntimeError("转换失败(非法 glb): %s" % step_path)
+    # 原子写：先写临时再改名
+    tmp = glb_path + ".tmp"
+    with open(tmp, "wb") as f:
+        f.write(out)
+    os.replace(tmp, glb_path)
+
+
+def main():
+    if len(sys.argv) < 3:
+        print("用法: python batch-convert-glb.py <step目录> <gltf输出目录> [deflection]")
+        sys.exit(1)
+    step_dir = sys.argv[1]
+    out_dir = sys.argv[2]
+    deflection = float(sys.argv[3]) if len(sys.argv) > 3 else 0.5
+    os.makedirs(out_dir, exist_ok=True)
+
+    steps = []
+    for root, _, files in os.walk(step_dir):
+        for fn in files:
+            if fn.lower().endswith((".step", ".stp")):
+                steps.append(os.path.join(root, fn))
+    steps.sort()
+
+    ok = 0
+    skip = 0
+    fail = 0
+    t0 = time.time()
+    for i, sp in enumerate(steps):
+        base = os.path.splitext(os.path.basename(sp))[0]
+        gp = os.path.join(out_dir, base + ".glb")
+        if os.path.exists(gp) and os.path.getsize(gp) > 0:
+            skip += 1
+            continue
+        try:
+            convert_one(sp, gp, deflection)
+            ok += 1
+            if (ok + skip) % 50 == 0:
+                print("[%d/%d] ok=%d skip=%d fail=%d  %.1fs" % (i + 1, len(steps), ok, skip, fail, time.time() - t0))
+        except Exception as e:
+            fail += 1
+            print("[FAIL] %s: %s" % (os.path.basename(sp), e))
+
+    print("完成: 总 %d, 转换 %d, 跳过 %d, 失败 %d, 耗时 %.1fs" % (len(steps), ok, skip, fail, time.time() - t0))
+    if fail:
+        sys.exit(2)
+
+
+if __name__ == "__main__":
+    main()