handle PIL error empty thumbnail image, #425

2023-02-13 11:55:47 +07:00 · 2023-02-13 11:55:47 +07:00 · ef685ecb42
parent df1018e70c
commit ef685ecb42
1 changed files with 5 additions and 1 deletions
--- a/tubearchivist/home/src/download/thumbnails.py
+++ b/tubearchivist/home/src/download/thumbnails.py
@ -43,7 +43,11 @@ class ThumbManagerBase:
                response = requests.get(url, stream=True, timeout=5)
                if response.ok:
                    try:
-                        return Image.open(response.raw)
+                        img = Image.open(response.raw)
+                        if isinstance(img, Image.Image):
+                            return img
+                        return self.get_fallback()
+
                    except UnidentifiedImageError:
                        print(f"failed to open thumbnail: {url}")
                        return self.get_fallback()