[cleanup] Remove completed TODO comments and fix ruff issues

- Removed completed TODO comments about code quality checks and client_id handling
- Updated comments to reflect implemented features
- Fixed ruff linting errors:
  - Removed duplicate constant definitions
  - Added missing locale import
  - Fixed unused imports
  - Moved is_local_mode logic to security_utils module
  - Added model_dir_name_map import to model_utils

All ruff checks now pass successfully.
This commit is contained in:
bymyself
2025-06-08 01:35:22 -07:00
parent c8882dcb7c
commit 7f1ebbe081
6 changed files with 27 additions and 77 deletions

View File

@@ -3,7 +3,7 @@ import git
import logging
import traceback
from comfyui_manager.common import context, manager_util
from comfyui_manager.common import context
import folder_paths
from comfy.cli_args import args
import latent_preview
@@ -125,17 +125,18 @@ def initialize_environment():
context.comfy_path = os.path.dirname(folder_paths.__file__)
core.js_path = os.path.join(context.comfy_path, "web", "extensions")
local_db_model = os.path.join(manager_util.comfyui_manager_path, "model-list.json")
local_db_alter = os.path.join(manager_util.comfyui_manager_path, "alter-list.json")
local_db_custom_node_list = os.path.join(
manager_util.comfyui_manager_path, "custom-node-list.json"
)
local_db_extension_node_mappings = os.path.join(
manager_util.comfyui_manager_path, "extension-node-map.json"
)
# Legacy database paths - kept for potential future use
# local_db_model = os.path.join(manager_util.comfyui_manager_path, "model-list.json")
# local_db_alter = os.path.join(manager_util.comfyui_manager_path, "alter-list.json")
# local_db_custom_node_list = os.path.join(
# manager_util.comfyui_manager_path, "custom-node-list.json"
# )
# local_db_extension_node_mappings = os.path.join(
# manager_util.comfyui_manager_path, "extension-node-map.json"
# )
set_preview_method(core.get_config()["preview_method"])
environment_utils.print_comfyui_version()
print_comfyui_version()
setup_environment()
core.check_invalid_nodes()

View File

@@ -3,6 +3,7 @@ import logging
import folder_paths
from comfyui_manager.glob import manager_core as core
from comfyui_manager.glob.constants import model_dir_name_map
def get_model_dir(data, show_log=False):

View File

@@ -1,7 +1,18 @@
from comfyui_manager.glob import manager_core as core
from comfy.cli_args import args
def is_loopback(address):
import ipaddress
try:
return ipaddress.ip_address(address).is_loopback
except ValueError:
return False
def is_allowed_security_level(level):
is_local_mode = is_loopback(args.listen)
if level == "block":
return False
elif level == "high":