open_pub/Uni-Lab-OS
1
0
Fork 0
mirror of https://github.com/deepmodeling/Uni-Lab-OS synced 2026-03-27 01:49:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update model_upload.py

Browse Source
This commit is contained in:
Junhan Chang
2026-03-25 08:55:21 +08:00
parent 090d5c5cb5
commit 18296d3cb2
1 changed files with 30 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
unilabos/app/model_upload.py
View File

@@ -24,9 +24,23 @@ _MESH_BASE_DIR = Path(__file__).parent.parent / "device_mesh"
# 支持的模型文件后缀 # 支持的模型文件后缀
_MODEL_EXTENSIONS = frozenset({ _MODEL_EXTENSIONS = frozenset({
".xacro", ".urdf", ".stl", ".dae", ".obj", ".xacro", ".urdf", ".stl", ".dae", ".obj",
".gltf", ".glb", ".yaml", ".yml", ".gltf", ".glb", ".fbx", ".yaml", ".yml",
}) })
# 需要 XOR 加密/解密的 mesh 文件后缀（反爬保护 — 方案 C）
_MESH_ENCRYPT_EXTENSIONS = frozenset({
".stl", ".dae", ".obj", ".fbx", ".gltf", ".glb",
})
# XOR 密钥 — 从环境变量读取，与前端 mesh-decrypt.ts 一致
_XOR_KEY = os.environ.get("UNILAB_MESH_XOR_KEY", "unilab3d-model-protection-key-v1").encode()
def _xor_transform(data: bytes, key: bytes = _XOR_KEY) -> bytes:
"""XOR 加密/解密（对称操作）。"""
key_len = len(key)
return bytes(b ^ key[i % key_len] for i, b in enumerate(data))
def upload_device_model( def upload_device_model(
http_client: "HTTPClient", http_client: "HTTPClient",
@@ -170,17 +184,27 @@ def download_model_from_oss(
def _put_upload(local_path: Path, upload_url: str) -> None: def _put_upload(local_path: Path, upload_url: str) -> None:
"""通过预签名 URL 上传文件到 OSS。""" """通过预签名 URL 上传文件到 OSS。对 mesh 文件自动 XOR 加密。"""
with open(local_path, "rb") as f: with open(local_path, "rb") as f:
resp = requests.put(upload_url, data=f, timeout=120) data = f.read()
resp.raise_for_status() # 对 mesh 文件 XOR 加密后上传（反爬保护 — 方案 C）
if local_path.suffix.lower() in _MESH_ENCRYPT_EXTENSIONS:
data = _xor_transform(data)
logger.debug(f"[模型上传] XOR 加密: {local_path.name}")
resp = requests.put(upload_url, data=data, timeout=120)
resp.raise_for_status()
logger.debug(f"[模型上传] 已上传: {local_path.name}") logger.debug(f"[模型上传] 已上传: {local_path.name}")
def _download_file(url: str, local_path: Path) -> None: def _download_file(url: str, local_path: Path) -> None:
"""下载单个文件到本地路径。""" """下载单个文件到本地路径。对 mesh 文件自动 XOR 解密。"""
local_path.parent.mkdir(parents=True, exist_ok=True) local_path.parent.mkdir(parents=True, exist_ok=True)
resp = requests.get(url, timeout=60) resp = requests.get(url, timeout=60)
resp.raise_for_status() resp.raise_for_status()
local_path.write_bytes(resp.content) data = resp.content
# 从 OSS 下载的 mesh 文件是加密的，需要 XOR 解密后再存本地
if local_path.suffix.lower() in _MESH_ENCRYPT_EXTENSIONS:
data = _xor_transform(data)
logger.debug(f"[模型下载] XOR 解密: {local_path.name}")
local_path.write_bytes(data)
logger.debug(f"[模型下载] 已下载: {local_path}") logger.debug(f"[模型下载] 已下载: {local_path}")