TBNilles/ComfyUI-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improved: is_local_mode - use ipaddress module instead of string match

Browse Source 
refactor: get_config() - ensure lowercase option when returning dict

https://github.com/ltdrdata/ComfyUI-Manager/issues/1546
This commit is contained in:
Dr.Lt.Data
2025-02-15 10:02:25 +09:00
parent a6816d53d7
commit 12351bada7
3 changed files with 18 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
glob/manager_server.py
View File

@@ -55,8 +55,14 @@ def handle_stream(stream, prefix):
from comfy.cli_args import args
import latent_preview
def is_loopback(address):
import ipaddress
try:
return ipaddress.ip_address(address).is_loopback
except ValueError:
return False
is_local_mode = args.listen.startswith('127.') or args.listen.startswith('local.')
is_local_mode = is_loopback(args.listen)
model_dir_name_map = {
@@ -85,11 +91,11 @@ def is_allowed_security_level(level):
return False
elif level == 'high':
if is_local_mode:
return core.get_config()['security_level'].lower() in ['weak', 'normal-']
return core.get_config()['security_level'] in ['weak', 'normal-']
else:
return core.get_config()['security_level'].lower() == 'weak'
return core.get_config()['security_level'] == 'weak'
elif level == 'middle':
return core.get_config()['security_level'].lower() in ['weak', 'normal', 'normal-']
return core.get_config()['security_level'] in ['weak', 'normal', 'normal-']
else:
return True