#4841: Fix FileIO constructor to honor closefd when called repeatedly
Patch by Victor Stinner.
This commit is contained in:
parent
8e6287f50d
commit
877effc298
|
@ -593,6 +593,19 @@ class IOTest(unittest.TestCase):
|
|||
self.assertEqual(rawio.read(2), None)
|
||||
self.assertEqual(rawio.read(2), b"")
|
||||
|
||||
def test_fileio_closefd(self):
|
||||
# Issue #4841
|
||||
with self.open(__file__, 'rb') as f1, \
|
||||
self.open(__file__, 'rb') as f2:
|
||||
fileio = self.FileIO(f1.fileno(), closefd=False)
|
||||
# .__init__() must not close f1
|
||||
fileio.__init__(f2.fileno(), closefd=False)
|
||||
f1.readline()
|
||||
# .close() must not close f2
|
||||
fileio.close()
|
||||
f2.readline()
|
||||
|
||||
|
||||
class CIOTest(IOTest):
|
||||
|
||||
def test_IOBase_finalize(self):
|
||||
|
|
|
@ -198,10 +198,14 @@ fileio_init(PyObject *oself, PyObject *args, PyObject *kwds)
|
|||
|
||||
assert(PyFileIO_Check(oself));
|
||||
if (self->fd >= 0) {
|
||||
if (self->closefd) {
|
||||
/* Have to close the existing file first. */
|
||||
if (internal_close(self) < 0)
|
||||
return -1;
|
||||
}
|
||||
else
|
||||
self->fd = -1;
|
||||
}
|
||||
|
||||
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|si:fileio",
|
||||
kwlist, &nameobj, &mode, &closefd))
|
||||
|
|
Loading…
Reference in New Issue