Skip to content

Commit

Permalink
Translate encoder error codes to strings
Browse files Browse the repository at this point in the history
When decoding, we use raise_oserror() to convert codec error codes to
strings.  Adapt that code to be used when encoding as well.  Add a new
helper function that returns the exception so we can still raise
`from exc`.

Ideally we'd replace the other two callers of raise_oserror() with
`raise _get_oserror()`, but the former is part of the public API.
  • Loading branch information
bgilbert committed Dec 10, 2023
1 parent 1a98590 commit 5426d7b
Showing 1 changed file with 9 additions and 6 deletions.
15 changes: 9 additions & 6 deletions src/PIL/ImageFile.py
Expand Up @@ -63,15 +63,19 @@
# Helpers


def raise_oserror(error):
def _get_oserror(error, *, encoder=False):
try:
msg = Image.core.getcodecstatus(error)
except AttributeError:
msg = ERRORS.get(error)
if not msg:
msg = f"decoder error {error}"
msg += " when reading image file"
raise OSError(msg)
msg = f"{'encoder' if encoder else 'decoder'} error {error}"
msg += f" when {'writing' if encoder else 'reading'} image file"
return OSError(msg)


def raise_oserror(error):
raise _get_oserror(error)


def _tilesort(t):
Expand Down Expand Up @@ -551,8 +555,7 @@ def _encode_tile(im, fp, tile: list[_Tile], bufsize, fh, exc=None):
# slight speedup: compress to real file object
errcode = encoder.encode_to_file(fh, bufsize)
if errcode < 0:
msg = f"encoder error {errcode} when writing image file"
raise OSError(msg) from exc
raise _get_oserror(errcode, encoder=True) from exc
finally:
encoder.cleanup()

Expand Down

0 comments on commit 5426d7b

Please sign in to comment.