xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-13 02:47:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

use a 'with' context

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin 2024-09-19 17:04:54 +09:00
parent cd71722b11
commit d66e844c1a
2 changed files with 17 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
folder_paths.py
View File

@ -66,6 +66,14 @@ class CacheHelper:
def clear(self): def clear(self):
self.cache.clear() self.cache.clear()
def __enter__(self):
self.active = True
return self
def __exit__(self, exc_type, exc_value, traceback):
self.active = False
self.clear()
cache_helper = CacheHelper() cache_helper = CacheHelper()
extension_mimetypes_cache = { extension_mimetypes_cache = {

20
server.py
View File

@ -552,17 +552,15 @@ class PromptServer():
@routes.get("/object_info") @routes.get("/object_info")
async def get_object_info(request): async def get_object_info(request):
folder_paths.cache_helper.clear() with folder_paths.cache_helper:
folder_paths.cache_helper.active = True out = {}
out = {} for x in nodes.NODE_CLASS_MAPPINGS:
for x in nodes.NODE_CLASS_MAPPINGS: try:
try: out[x] = node_info(x)
out[x] = node_info(x) except Exception as e:
except Exception as e: logging.error(f"[ERROR] An error occurred while retrieving information for the '{x}' node.")
logging.error(f"[ERROR] An error occurred while retrieving information for the '{x}' node.") logging.error(traceback.format_exc())
logging.error(traceback.format_exc()) return web.json_response(out)
folder_paths.cache_helper.active = False
return web.json_response(out)
@routes.get("/object_info/{node_class}") @routes.get("/object_info/{node_class}")
async def get_object_info_node(request): async def get_object_info_node(request):