1580158
/
sd-webui-infinite-image-browsing
mirror of https://github.com/zanllp/sd-webui-infinite-image-browsing.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #823 from zanllp/feat/handle-missing-filetype 

add fallback for uninstalled filetype handlers
pull/825/head
zanllp 2025-06-09 00:52:09 +08:00 committed by GitHub
parent edb69ea13e 87f261ce72
commit 22d51a4e6f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/iib/tool.py
View File

@ -14,7 +14,6 @@ import zipfile
from PIL import Image from PIL import Image
import shutil import shutil
# import magic # import magic
import filetype
sd_img_dirs = [ sd_img_dirs = [
"outdir_txt2img_samples", "outdir_txt2img_samples",
@ -203,8 +202,14 @@ def convert_to_bytes(file_size_str):
def is_video_simple(filepath): def is_video_simple(filepath):
kind = filetype.guess(filepath) try:
return kind and kind.mime.startswith('video/') import filetype
kind = filetype.guess(filepath)
# print(f"File type guessed: {kind}")
return kind and kind.mime.startswith('video/')
except:
# 如果 filetype 模块不可用，使用简单的文件扩展名检查
return isinstance(get_video_type(filepath), str)
def get_video_type(file_path): def get_video_type(file_path):
video_extensions = ['.mp4', '.m4v', '.avi', '.mkv', '.mov', '.wmv', '.flv', '.ts'] video_extensions = ['.mp4', '.m4v', '.avi', '.mkv', '.mov', '.wmv', '.flv', '.ts']