Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 6 additions & 4 deletions iconvcodec.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,8 +39,6 @@ def _iconv_encode_impl(encoder, msg, errors, bufsize=None):
out1, len1 = _iconv_encode_impl(
encoder, msg[inlen:].decode()[1:].encode(), errors
)
else:
raise ValueError("unsupported error handling")
return outres + out1, inlen + len1 + 1
raise

Expand Down Expand Up @@ -74,8 +72,6 @@ def _iconv_decode_impl(decoder, msg, errors, bufsize=None):
out1, len1 = _iconv_decode_impl(decoder, msg[inlen + 1 :], errors)
elif errors == "ignore":
out1, len1 = _iconv_decode_impl(decoder, msg[inlen + 1 :], errors)
else:
raise ValueError("unsupported error handling")
return outres.decode() + out1, inlen + len1 + 1


Expand All @@ -93,9 +89,15 @@ def codec_factory(encoding):
def encode(inp, errors="strict"):
msg = inp.encode()

if errors not in ("strict", "replace", "ignore"):
raise ValueError("unsupported error handling")

return _iconv_encode_impl(encoder, msg, errors)

def decode(msg, errors="strict"):
if errors not in ("strict", "replace", "ignore"):
raise ValueError("unsupported error handling")

return _iconv_decode_impl(decoder, msg, errors)

return encode, decode
Expand Down
7 changes: 7 additions & 0 deletions test_iconvcodec.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,3 +45,10 @@ def test_incremental_decode(self):

self.assertEqual(first, "")
self.assertEqual(second, "\u0141")

def test_invalid_errors(self):
with self.assertRaisesRegex(ValueError, "unsupported error handling"):
"™".encode("ASCII//TRANSLIT", "WRONG")

with self.assertRaisesRegex(ValueError, "unsupported error handling"):
b"a".decode("ASCII//TRANSLIT", "WRONG")