pax/booru-viewer
1
0
Fork 0
Code Issues Releases 18 Activity

Fix ugoira conversion — filter non-image files, skip bad frames

Browse Source
This commit is contained in:
pax 2026-04-04 19:43:47 -05:00
parent 2f029da916
commit ce51bfd98d
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
booru_viewer/core/cache.py
View File

@ -53,15 +53,20 @@ def _ext_from_url(url: str) -> str:
def _convert_ugoira_to_gif(zip_path: Path) -> Path: def _convert_ugoira_to_gif(zip_path: Path) -> Path:
"""Convert a Pixiv ugoira zip (numbered JPEG/PNG frames) to an animated GIF.""" """Convert a Pixiv ugoira zip (numbered JPEG/PNG frames) to an animated GIF."""
import io
gif_path = zip_path.with_suffix(".gif") gif_path = zip_path.with_suffix(".gif")
if gif_path.exists(): if gif_path.exists():
return gif_path return gif_path
_IMG_EXTS = {".jpg", ".jpeg", ".png", ".bmp", ".webp"}
with zipfile.ZipFile(zip_path, "r") as zf: with zipfile.ZipFile(zip_path, "r") as zf:
names = sorted(zf.namelist()) names = sorted(n for n in zf.namelist() if Path(n).suffix.lower() in _IMG_EXTS)
frames = [] frames = []
for name in names: for name in names:
try:
data = zf.read(name) data = zf.read(name)
frames.append(Image.open(__import__("io").BytesIO(data)).convert("RGBA")) frames.append(Image.open(io.BytesIO(data)).convert("RGBA"))
except Exception:
continue
if not frames: if not frames:
raise ValueError("Zip contains no image frames") raise ValueError("Zip contains no image frames")
frames[0].save( frames[0].save(