`ZstdFile` raises `AttributeError` instead of `io.UnsupportedOperation` when iterating over a non-readable file

[lkk7]

Bug report

Bug description:

All other compression modules raise io.UnsupportedOperation when iterating over a non-readable file:

next(iter(lzma.LZMAFile(io.BytesIO(), "w")))   # io.UnsupportedOperation: File not open for reading
next(iter(bz2.BZ2File(io.BytesIO(), "w")))     # io.UnsupportedOperation: ...
next(iter(gzip.GzipFile(fileobj=io.BytesIO(), mode="w")))  # ...

But ZstdFile.__next__ doesn't call _check_can_read(), so it dereferences self._buffer, which is None in write mode:

import io
from compression.zstd import ZstdFile

next(iter(ZstdFile(io.BytesIO(), "w")))

This results in:

AttributeError: 'NoneType' object has no attribute 'readline'

CPython versions tested on:

3.14, 3.15, CPython main branch

Operating systems tested on:

macOS, but it does not depend on OS

Linked PRs

• gh-153037: Fix AttributeError when iterating over a non-readable ZstdFile #153045

[lkk7]

I have a patch ready and working, I believe, so I'll push that soon.

[lkk7]

@emmatyping tagging you because I see you're the zstd expert and I'd be very happy to submit a patch if you agree this should be fixed.

[emmatyping]

I agree consistency here would be good. Thank you for working on a patch @lkk7!