Fix the situation that requiresments.txt contains ~=

This commit is contained in:
sko00o 2024-12-05 16:39:31 +08:00
parent 621ca13906
commit cbc592bc9a
No known key found for this signature in database
GPG Key ID: 240DEE2151D0CF19
2 changed files with 6 additions and 6 deletions

View File

@ -131,7 +131,7 @@ def clear_pip_cache():
def is_blacklisted(name):
name = name.strip()
pattern = r'([^<>!=]+)([<>!=]=?)([^ ]*)'
pattern = r'([^<>!~=]+)([<>!~=]=?)([^ ]*)'
match = re.search(pattern, name)
if match:
@ -146,7 +146,7 @@ def is_blacklisted(name):
if match is None:
if name in pips:
return True
elif match.group(2) in ['<=', '==', '<']:
elif match.group(2) in ['<=', '==', '~=', '<']:
if name in pips:
if StrictVersion(pips[name]) >= StrictVersion(match.group(3)):
return True
@ -160,7 +160,7 @@ def is_installed(name):
if name.startswith('#'):
return True
pattern = r'([^<>!=]+)([<>!=]=?)([0-9.a-zA-Z]*)'
pattern = r'([^<>!~=]+)([<>!~=]=?)([0-9.a-zA-Z]*)'
match = re.search(pattern, name)
if match:
@ -175,7 +175,7 @@ def is_installed(name):
if match is None:
if name in pips:
return True
elif match.group(2) in ['<=', '==', '<']:
elif match.group(2) in ['<=', '==', '~=', '<']:
if name in pips:
if StrictVersion(pips[name]) >= StrictVersion(match.group(3)):
print(f"[ComfyUI-Manager] skip black listed pip installation: '{name}'")

View File

@ -394,7 +394,7 @@ def is_installed(name):
if name.startswith('#'):
return True
pattern = r'([^<>!=]+)([<>!=]=?)([0-9.a-zA-Z]*)'
pattern = r'([^<>!~=]+)([<>!~=]=?)([0-9.a-zA-Z]*)'
match = re.search(pattern, name)
if match:
@ -409,7 +409,7 @@ def is_installed(name):
if match is None:
if name in pips:
return True
elif match.group(2) in ['<=', '==', '<']:
elif match.group(2) in ['<=', '==', '~=', '<']:
if name in pips:
if StrictVersion(pips[name]) >= StrictVersion(match.group(3)):
print(f"[ComfyUI-Manager] skip black listed pip installation: '{name}'")