wangbo/ComfyUI-Manager
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI-Manager.git synced 2026-07-28 19:17:34 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Chore: Replace print to logging in cnr_utils.py

Browse Source
This commit is contained in:
craftingmod
2026-06-30 04:47:56 +09:00
parent c352b16bb1
commit edcfb6a35a
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
glob/cnr_utils.py
+8 -6
View File
@@ -6,6 +6,7 @@ import time
from dataclasses import dataclass from dataclasses import dataclass
from typing import List from typing import List
import logging
import manager_core import manager_core
import manager_util import manager_util
import requests import requests
@@ -22,7 +23,7 @@ async def get_cnr_data(cache_mode=True, dont_wait=True):
try: try:
return await _get_cnr_data(cache_mode, dont_wait) return await _get_cnr_data(cache_mode, dont_wait)
except asyncio.TimeoutError: except asyncio.TimeoutError:
print("A timeout occurred during the fetch process from ComfyRegistry.") logging.error(f"[ComfyUI-Manager] A timeout occurred during the fetch process from ComfyRegistry.")
return await _get_cnr_data(cache_mode=True, dont_wait=True) # timeout fallback return await _get_cnr_data(cache_mode=True, dont_wait=True) # timeout fallback
async def _get_cnr_data(cache_mode=True, dont_wait=True): async def _get_cnr_data(cache_mode=True, dont_wait=True):
@@ -78,12 +79,12 @@ async def _get_cnr_data(cache_mode=True, dont_wait=True):
full_nodes[x['id']] = x full_nodes[x['id']] = x
if page % 5 == 0: if page % 5 == 0:
print(f"FETCH ComfyRegistry Data: {page}/{sub_json_obj['totalPages']}") logging.info(f"[ComfyUI-Manager] FETCH ComfyRegistry Data: {page}/{sub_json_obj['totalPages']}")
page += 1 page += 1
time.sleep(0.5) time.sleep(0.5)
print("FETCH ComfyRegistry Data [DONE]") logging.info(f"[ComfyUI-Manager] FETCH ComfyRegistry Data [DONE]")
for v in full_nodes.values(): for v in full_nodes.values():
if 'latest_version' not in v: if 'latest_version' not in v:
@@ -94,12 +95,13 @@ async def _get_cnr_data(cache_mode=True, dont_wait=True):
if cache_mode: if cache_mode:
is_cache_loading = True is_cache_loading = True
cache_state = manager_util.get_cache_state(uri) cache_state = manager_util.get_cache_state(uri)
if dont_wait: if dont_wait:
if cache_state == 'not-cached': if cache_state == 'not-cached':
return {} return {}
else: else:
print("[ComfyUI-Manager] The ComfyRegistry cache update is still in progress, so an outdated cache is being used.") logging.warning(f"[ComfyUI-Manager] The ComfyRegistry cache update is still in progress, so an outdated cache is being used.")
with open(manager_util.get_cache_path(uri), 'r', encoding="UTF-8", errors="ignore") as json_file: with open(manager_util.get_cache_path(uri), 'r', encoding="UTF-8", errors="ignore") as json_file:
return json.load(json_file)['nodes'] return json.load(json_file)['nodes']
@@ -113,7 +115,7 @@ async def _get_cnr_data(cache_mode=True, dont_wait=True):
return json_obj['nodes'] return json_obj['nodes']
except: except:
res = {} res = {}
print("Cannot connect to comfyregistry.") logging.error(f"[ComfyUI-Manager] Cannot connect to comfyregistry.")
finally: finally:
if cache_mode: if cache_mode:
is_cache_loading = False is_cache_loading = False
@@ -237,7 +239,7 @@ def generate_cnr_id(fullpath, cnr_id):
with open(cnr_id_path, "w") as f: with open(cnr_id_path, "w") as f:
return f.write(cnr_id) return f.write(cnr_id)
except: except:
print(f"[ComfyUI Manager] unable to create file: {cnr_id_path}") logging.error(f"[ComfyUI-Manager] unable to create file: {cnr_id_path}")
def read_cnr_id(fullpath): def read_cnr_id(fullpath):