Fetch module state by class in _io.FileIO.readinto

python · kumaraditya303 · May 11, 2023 · May 10, 2023 · May 10, 2023 · May 10, 2023
commit 8196135de1acd75b2ebadd70123a97f300f57836
diff --git a/Modules/_io/clinic/fileio.c.h b/Modules/_io/clinic/fileio.c.h
diff --git a/Modules/_io/fileio.c b/Modules/_io/fileio.c
@@ -628,23 +628,24 @@ _io_FileIO_seekable_impl(fileio *self)

 /*[clinic input]
 _io.FileIO.readinto
+    cls: defining_class
    buffer: Py_buffer(accept={rwbuffer})
    /

 Same as RawIOBase.readinto().
 [clinic start generated code]*/

 static PyObject *
-_io_FileIO_readinto_impl(fileio *self, Py_buffer *buffer)
-/*[clinic end generated code: output=b01a5a22c8415cb4 input=4721d7b68b154eaf]*/
+_io_FileIO_readinto_impl(fileio *self, PyTypeObject *cls, Py_buffer *buffer)
+/*[clinic end generated code: output=97f0f3d69534db34 input=fd20323e18ce1ec8]*/
 {
    Py_ssize_t n;
    int err;

    if (self->fd < 0)
        return err_closed();
    if (!self->readable) {
-        _PyIO_State *state = IO_STATE();
+        _PyIO_State *state = get_io_state_by_cls(cls);
        return err_mode(state, "reading");
    }