Use Pillow's Image context manager to close image file

This commit is contained in:
Jon Dufresne 2019-08-15 19:46:12 -07:00
parent 47cd262b3e
commit 8f30ec9407

View File

@ -42,12 +42,8 @@ def get_image_size(filename: str) -> Tuple[int, int]:
size = None
if size is None and Image: # fallback to Pillow
im = Image.open(filename)
size = im.size
try:
im.fp.close()
except Exception:
pass
with Image.open(filename) as im:
size = im.size
return size
except Exception: