Upload 2 files

router_messages.py

@@ -59,6 +59,7 @@ class AdminScriptRequest(BaseModel):
 # 🔒 P0安全修复：脚本白名单（仅允许执行指定的脚本）
 # 警告：添加新脚本前请确保其安全性
 ALLOWED_SCRIPTS = {
+    "临时_版本同步测试.py",
     "密码迁移.py",          # 用户密码哈希化迁移
     "测试脚本.py",          # 接口测试工具
     "迁移_余额合并.py",      # 一次性余额合并迁移（执行后可移除）
@@ -103,6 +104,9 @@ async def run_admin_script(request: Request, req: AdminScriptRequest, current_us
         }
     
     try:
+        if script_name == "临时_版本同步测试.py":
+            from 临时_版本同步测试 import run
+            return run()
         # 清理敏感环境变量，防止泄露到子进程
         sensitive_prefixes = ('GITHUB', 'JWT', 'HF_', 'DATABASE', 'ALIPAY', 'PASSWORD',
                               'MAKE_', 'TWILIO_', 'ALIYUN_', 'TENCENT_', 'NCA_')

临时_版本同步测试.py

@@ -0,0 +1,109 @@
+"""管理员临时诊断：当前进程只启动一轮，后续调用查询原结果。"""
+import asyncio
+import logging
+
+MAX_ITEMS = 5
+logger = logging.getLogger("ComfyUI-Ranking.VersionSyncTest")
+_result = dict(state="ready", selected=0, checked=0, updated=0,
+               attempted=0, cached=0, already_cached=0, failed=0)
+
+
+def _summary():
+    return "VERSION_SYNC_TEST | " + " | ".join(f"{key}={value}" for key, value in _result.items())
+
+
+def run():
+    from resource_work import start_resource_background
+    from script_execution import ScriptExecutionBusy
+
+    if _result["state"] == "ready":
+        _result["state"] = "running"
+        try:
+            accepted = start_resource_background(_execute)
+        except BaseException:
+            _result["state"] = "ready"
+            raise
+        if not accepted:
+            _result["state"] = "ready"
+            raise ScriptExecutionBusy()
+    failed = _result["state"] in ("failed", "cancelled")
+    return {"status": "error" if failed else "success",
+            "return_code": 1 if failed else 0, "output": _summary()}
+
+
+async def _execute():
+    stage = "source_read"
+    try:
+        import 数据库连接 as db
+        from owned_thread import run_owned_thread
+        from github_source import repository_parts, select_github_token
+        from github_archive import cached_archive
+        from github_versions import refresh_resource_version
+        from 云端_定时版本检测引擎 import precache_github_zip
+
+        logger.info("VERSION_SYNC_TEST_STARTED | limit=%s", MAX_ITEMS)
+        items = await run_owned_thread(db.load_data, "items.json", default_data=[], strict=True, skip_cache=True)
+        versions = await run_owned_thread(db.load_data, "versions.json", default_data={}, strict=True, skip_cache=True)
+        if not isinstance(items, list) or not isinstance(versions, dict):
+            raise ValueError("invalid version source")
+        pending, available = [], []
+        for item in items:
+            if not isinstance(item, dict) or item.get("type") not in ("tool", "skill"):
+                continue
+            if not isinstance(item.get("id"), str):
+                continue
+            try:
+                repository_parts(item.get("link"))
+                token = select_github_token(item)
+                archive = cached_archive(item, token, versions.get(item["id"]), repo_id=db.DATASET_REPO_ID)
+            except ValueError:
+                continue
+            if len(available) < MAX_ITEMS:
+                available.append(item)
+            if archive is None:
+                pending.append(item)
+                if len(pending) == MAX_ITEMS:
+                    break
+        selected = pending or available
+        _result["selected"] = len(selected)
+        for index, item in enumerate(selected, 1):
+            stage = "version_check"
+            try:
+                refreshed = await refresh_resource_version(db, item["id"])
+                if not refreshed:
+                    _result["failed"] += 1
+                    logger.warning("VERSION_SYNC_TEST_ITEM | item=%s | stage=version_check | result=unconfirmed", index)
+                    continue
+                _result["checked"] += 1
+                _result["updated"] += bool(refreshed["changed"])
+                stage = "cache_read"
+                current_versions = await run_owned_thread(db.load_data, "versions.json", default_data={}, strict=True, skip_cache=True)
+                current, token = refreshed["item"], refreshed["token"]
+                if cached_archive(current, token, current_versions.get(item["id"]), repo_id=db.DATASET_REPO_ID):
+                    _result["already_cached"] += 1
+                    continue
+                stage = "precache"
+                _result["attempted"] += 1
+                if await precache_github_zip(current["link"], token, item["id"], refreshed["commit"]):
+                    _result["cached"] += 1
+                    logger.info("VERSION_SYNC_TEST_ITEM | item=%s | stage=precache | result=success", index)
+                else:
+                    _result["failed"] += 1
+                    logger.warning("VERSION_SYNC_TEST_ITEM | item=%s | stage=precache | result=failed", index)
+            except Exception as error:
+                _result["failed"] += 1
+                logger.warning("VERSION_SYNC_TEST_ITEM | item=%s | stage=%s | error_type=%s", index, stage, type(error).__name__)
+        _result["state"] = "failed" if _result["failed"] else "completed"
+    except asyncio.CancelledError:
+        _result["state"] = "cancelled"
+        raise
+    except Exception as error:
+        _result["state"] = "failed"
+        _result["failed"] += 1
+        logger.warning("VERSION_SYNC_TEST_FAILED | stage=%s | error_type=%s", stage, type(error).__name__)
+    finally:
+        logger.info(_summary())
+
+
+if __name__ == "__main__":
+    raise SystemExit("Use the authenticated administrator script endpoint.")