diff --git a/comfyui_manager/__init__.py b/comfyui_manager/__init__.py index a99aecb5..1dd0791e 100644 --- a/comfyui_manager/__init__.py +++ b/comfyui_manager/__init__.py @@ -1,6 +1,7 @@ import os import logging + def prestartup(): from . import prestartup_script # noqa: F401 logging.info('[PRE] ComfyUI-Manager') @@ -15,3 +16,18 @@ def start(): if os.environ.get('ENABLE_LEGACY_COMFYUI_MANAGER_FRONT', 'false') == 'true': import nodes nodes.EXTENSION_WEB_DIRS['comfyui-manager-legacy'] = os.path.join(os.path.dirname(__file__), 'js') + + +def should_be_disabled(fullpath:str) -> bool: + """ + 1. Disables the legacy ComfyUI-Manager. + 2. The blocklist can be expanded later based on policies. + """ + + # In cases where installation is done via a zip archive, the directory name may not be comfyui-manager, and it may not contain a git repository. + # It is assumed that any installed legacy ComfyUI-Manager will have at least 'comfyui-manager' in its directory name. + dir_name = os.path.basename(fullpath).lower() + if 'comfyui-manager' in dir_name: + return True + else: + return False