Handle UnidentifiedImageError gracefully

newsbeam · Jun 21, 2020 · 768965a · 768965a
1 parent 116a9aa
commit 768965a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/nm_iconfinder/nm_iconfinder.py b/nm_iconfinder/nm_iconfinder.py
@@ -5,7 +5,7 @@
 from io import BytesIO
 
 from bs4 import BeautifulSoup
-from PIL import Image
+from PIL import Image, UnidentifiedImageError
 import requests
 
 TIMEOUT = 3  # seconds
@@ -40,7 +40,10 @@ def from_url(cls, url: str) -> Optional["Icon"]:
             return None
 
         with BytesIO(res.content) as bio:
-            img = Image.open(bio)
+            try:
+                img = Image.open(bio)
+            except UnidentifiedImageError:
+                return None
         width, height = img.size
         # Ignore non-square Icons
         if width != height:

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name='nm_iconfinder',
     packages=['nm_iconfinder'],
-    version='0.1',
+    version='0.1.1',
     license='ISC',
     description='Find icons of a website given a URL',
     author='newsmail.today',