DemonGatanjieuandClaude Opus 5.5
d379581853
"Doctor" said little about what it does: check the opened workflow's models and put back the ones this computer has. The rail, page, home card, help, tour, README, the unreleased changelog and the architecture docs now say Model Check (模型检查); the help paragraph now describes the check as it works today. Code, CSS and locale keys keep "doctor"; released changelog entries, audits and decisions keep the old name. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
import os
|
|
from pathlib import Path
|
|
|
|
|
|
# These components are normally referenced by stable ecosystem filenames.
|
|
# Renaming them breaks existing workflows without providing meaningful value.
|
|
PHYSICAL_RENAME_PROTECTED_TYPES = frozenset({
|
|
"vae",
|
|
"vae_approx",
|
|
"clip",
|
|
"text_encoders",
|
|
"clip_vision",
|
|
})
|
|
|
|
# This is a recovery-confidence policy, not a scanner-support list. The scan
|
|
# wizard may hash these folders, but Model Check must not redirect one of these
|
|
# components from size, filename, metadata labels, or Civitai availability.
|
|
HASH_ONLY_RECOVERY_TYPES = frozenset({
|
|
"vae",
|
|
"vae_approx",
|
|
"clip",
|
|
"text_encoders",
|
|
"clip_vision",
|
|
})
|
|
|
|
|
|
def is_physical_rename_protected(folder_type=None, folder_path=None):
|
|
"""Return True when a model category must keep its physical filename."""
|
|
normalized_type = str(folder_type or "").strip().lower()
|
|
if normalized_type in PHYSICAL_RENAME_PROTECTED_TYPES:
|
|
return True
|
|
|
|
if not folder_path:
|
|
return False
|
|
|
|
try:
|
|
parts = {
|
|
os.path.normcase(part).lower()
|
|
for part in Path(os.path.abspath(folder_path)).parts
|
|
}
|
|
except (OSError, TypeError, ValueError):
|
|
return False
|
|
return bool(parts.intersection(PHYSICAL_RENAME_PROTECTED_TYPES))
|
|
|
|
|
|
def requires_hash_for_model_recovery(folder_types):
|
|
"""Return True when every requested category requires a hash match."""
|
|
normalized = {
|
|
str(folder_type or "").strip().lower()
|
|
for folder_type in (folder_types or ())
|
|
if str(folder_type or "").strip()
|
|
}
|
|
return bool(normalized) and normalized.issubset(HASH_ONLY_RECOVERY_TYPES)
|