wangbo/ComfyUI-Manager
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI-Manager.git synced 2025-12-19 19:33:03 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[fix] Remove unused imports to fix linting errors

Browse Source
This commit is contained in:
bymyself 2025-06-17 13:08:52 -07:00
parent 03ecda3cfe
commit 14298b0859
2 changed files with 12 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfyui_manager/glob/constants.py
View File

@ -1,4 +1,3 @@
from comfy.cli_args import args
SECURITY_MESSAGE_MIDDLE_OR_BELOW = "ERROR: To use this action, a security_level of `middle or below` is required. Please contact the administrator.\nReference: https://github.com/ltdrdata/ComfyUI-Manager#security-policy"
SECURITY_MESSAGE_NORMAL_MINUS = "ERROR: To use this feature, you must either set '--listen' to a local IP and set the security level to 'normal-' or lower, or set the security level to 'middle' or 'weak'. Please contact the administrator.\nReference: https://github.com/ltdrdata/ComfyUI-Manager#security-policy"
@ -15,9 +14,6 @@ def is_loopback(address):
return False
is_local_mode = is_loopback(args.listen)
model_dir_name_map = {
"checkpoints": "checkpoints",
"checkpoint": "checkpoints",

23
comfyui_manager/glob/utils/security_utils.py
View File

@ -1,5 +1,6 @@
from comfyui_manager.glob import manager_core as core
from comfy.cli_args import args
from comfyui_manager.data_models import SecurityLevel, RiskLevel, ManagerDatabaseSource
def is_loopback(address):
@ -13,23 +14,23 @@ def is_loopback(address):
def is_allowed_security_level(level):
is_local_mode = is_loopback(args.listen)
if level == "block":
if level == RiskLevel.block.value:
return False
elif level == "high":
elif level == RiskLevel.high.value:
if is_local_mode:
return core.get_config()["security_level"] in ["weak", "normal-"]
return core.get_config()["security_level"] in [SecurityLevel.weak.value, SecurityLevel.normal_.value]
else:
return core.get_config()["security_level"] == "weak"
elif level == "middle":
return core.get_config()["security_level"] in ["weak", "normal", "normal-"]
return core.get_config()["security_level"] == SecurityLevel.weak.value
elif level == RiskLevel.middle.value:
return core.get_config()["security_level"] in [SecurityLevel.weak.value, SecurityLevel.normal.value, SecurityLevel.normal_.value]
else:
return True
async def get_risky_level(files, pip_packages):
json_data1 = await core.get_data_by_mode("local", "custom-node-list.json")
json_data1 = await core.get_data_by_mode(ManagerDatabaseSource.local.value, "custom-node-list.json")
json_data2 = await core.get_data_by_mode(
"cache",
ManagerDatabaseSource.cache.value,
"custom-node-list.json",
channel_url="https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main",
)
@ -40,7 +41,7 @@ async def get_risky_level(files, pip_packages):
for x in files:
if x not in all_urls:
return "high"
return RiskLevel.high.value
all_pip_packages = set()
for x in json_data1["custom_nodes"] + json_data2["custom_nodes"]:
@ -48,6 +49,6 @@ async def get_risky_level(files, pip_packages):
for p in pip_packages:
if p not in all_pip_packages:
return "block"
return RiskLevel.block.value
return "middle"
return RiskLevel.middle.value