wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-20 05:18:15 +08:00
Code Issues Actions 25 Packages Projects Releases Wiki Activity

Merge de4feda3f6 into 1d1099bea0

Browse Source
This commit is contained in:
sean-kim05 2026-07-17 12:59:40 -07:00 committed by GitHub
commit b8f0397c8a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
folder_paths.py
View File

@ -236,7 +236,7 @@ def filter_files_content_types(files: list[str], content_types: List[Literal["im
global extension_mimetypes_cache global extension_mimetypes_cache
result = [] result = []
for file in files: for file in files:
extension = file.split('.')[-1] extension = file.split('.')[-1].lower()
if extension not in extension_mimetypes_cache: if extension not in extension_mimetypes_cache:
mime_type, _ = mimetypes.guess_type(file, strict=False) mime_type, _ = mimetypes.guess_type(file, strict=False)
if not mime_type: if not mime_type:

8
tests-unit/folder_paths_test/filter_by_content_types_test.py
View File

@ -51,6 +51,14 @@ def test_categorizes_all_uniquely(mock_dir, file_extensions, patched_mimetype_ca
assert len(filtered_files) == len(extensions) assert len(filtered_files) == len(extensions)
def test_categorizes_uppercase_extensions(patched_mimetype_cache):
# Extensions must be matched case-insensitively (like filter_files_extensions),
# so an uppercase 3D-model extension isn't dropped just because the mimetype
# cache keys and the OS mimetypes DB are lowercase.
files = ["sample_model.FBX", "sample_model.fbx"]
assert filter_files_content_types(files, ["model"]) == ["sample_model.FBX", "sample_model.fbx"]
def test_handles_bad_extensions(): def test_handles_bad_extensions():
files = ["file1.txt", "file2.py", "file3.example", "file4.pdf", "file5.ini", "file6.doc", "file7.md"] files = ["file1.txt", "file2.py", "file3.example", "file4.pdf", "file5.ini", "file6.doc", "file7.md"]
assert filter_files_content_types(files, ["image", "audio", "video"]) == [] assert filter_files_content_types(files, ["image", "audio", "video"]) == []