Upload 云端_定时版本检测引擎.py

云端_定时版本检测引擎.py

@@ -106,23 +106,29 @@ async def precache_github_zip(repo_url: str, token: Optional[str], item_id: str,
     client = None
     response = None
     resources = None
+    stage = "storage_identity"
     try:
         storage_repo = archive_repository(db.DATASET_REPO_ID)
         from proxy_access import read_proxy_item
+        stage = "resource_read"
         item = await run_owned_thread(read_proxy_item, db, item_id)
+        stage = "source_validation"
         current_token = select_github_token(item)
         if repository_parts(repo_url) != repository_parts(item['link']) or token != current_token:
             return False
         owner, repo = repository_parts(repo_url)
+        stage = "commit_resolution"
         ref = commit_ref(version_hash)
         sha = commit_sha(ref) if len(ref) == 40 else await fetch_github_commit(repo_url, token, ref)
         if not sha:
             return False
+        stage = "archive_identity"
         signature = source_signature(item, token)
         path_in_repo = archive_path(item_id, sha, signature)
         hf_token = os.environ.get("HF_TOKEN")
         if not hf_token:
             return False
+        stage = "storage_permission"
         storage = await run_owned_thread(HfApi(token=hf_token).repo_info, repo_id=storage_repo, repo_type="dataset")
         if getattr(storage, 'private', None) is not True:
             logger.warning("ZIP预缓存要求私有Dataset，跳过发布")
@@ -130,6 +136,7 @@ async def precache_github_zip(repo_url: str, token: Optional[str], item_id: str,
         headers = {"Accept": "application/vnd.github.v3+json", "User-Agent": "ComfyUI-Hub", "Accept-Encoding": "identity"}
         if token:
             headers["Authorization"] = f"Bearer {token}"
+        stage = "archive_request"
         client = httpx.AsyncClient(follow_redirects=False, trust_env=False, transport=ValidatedAsyncHTTPTransport())
         resources = ProxyResources(client)
         # 共用逐跳公网/DNS/TLS与跨origin凭据隔离；归档重定向仍需验证。
@@ -138,6 +145,7 @@ async def precache_github_zip(repo_url: str, token: Optional[str], item_id: str,
             f"https://api.github.com/repos/{owner}/{repo}/zipball/{sha}", headers=headers, timeout=300))
         if response.status_code != 200:
             return False
+        stage = "archive_download"
         fd, temp_path = tempfile.mkstemp(suffix=".zip", prefix="ranking-precache-")
         size = 0
         with os.fdopen(fd, 'wb') as output:
@@ -146,11 +154,16 @@ async def precache_github_zip(repo_url: str, token: Optional[str], item_id: str,
                 if size > ZIP_MAX_SIZE:
                     return False
                 output.write(chunk)
+        stage = "archive_close"
         await resources.close()
+        stage = "archive_validation"
         await run_owned_thread(validate_zip_archive, temp_path, size)
+        stage = "hf_upload"
         uploaded = await run_owned_thread(upload_file, path_or_fileobj=temp_path, path_in_repo=path_in_repo,
             repo_id=storage_repo, repo_type="dataset", token=hf_token)
+        stage = "hf_commit"
         hf_commit = commit_sha(getattr(uploaded, 'oid', None))
+        stage = "source_recheck"
         latest_item = await run_owned_thread(read_proxy_item, db, item_id)
         if source_signature(latest_item, select_github_token(latest_item)) != signature:
             return False
@@ -161,9 +174,10 @@ async def precache_github_zip(repo_url: str, token: Optional[str], item_id: str,
             if db.DATASET_REPO_ID != storage_repo:
                 return False
             return publish_archive_metadata(versions, item_id, ref, metadata)
+        stage = "metadata_publish"
         return bool(await run_owned_thread(db.atomic_update, "versions.json", update, default_data={}))
     except Exception as error:
-        logger.warning("预缓存失败: error_type=%s", type(error).__name__)
+        logger.warning("预缓存失败: stage=%s | error_type=%s", stage, type(error).__name__)
         return False
     finally:
         try: