python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-04-12 21:34:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

Close file after reading in ImagePalette.load()

Browse Source 
Fixes some "ResourceWarning: unclosed file ..." when running tests with
warnings enabled.
This commit is contained in:
Jon Dufresne 2016-11-11 16:59:32 -08:00
parent 76d156bef0
commit 6bc8fdd342
1 changed files with 17 additions and 17 deletions

34
PIL/ImagePalette.py
View File

@ -197,23 +197,23 @@ def load(filename):
# FIXME: supports GIMP gradients only
fp = open(filename, "rb")
with open(filename, "rb") as fp:
for paletteHandler in [
GimpPaletteFile.GimpPaletteFile,
GimpGradientFile.GimpGradientFile,
PaletteFile.PaletteFile
]:
try:
fp.seek(0)
lut = paletteHandler(fp).getpalette()
if lut:
break
except (SyntaxError, ValueError):
# import traceback
# traceback.print_exc()
pass
else:
raise IOError("cannot load palette")
for paletteHandler in [
GimpPaletteFile.GimpPaletteFile,
GimpGradientFile.GimpGradientFile,
PaletteFile.PaletteFile
]:
try:
fp.seek(0)
lut = paletteHandler(fp).getpalette()
if lut:
break
except (SyntaxError, ValueError):
# import traceback
# traceback.print_exc()
pass
else:
raise IOError("cannot load palette")
return lut # data, rawmode