Skip to content

Commit 1462234

Browse files
[2.7] bpo-8765: Deprecate writing unicode to binary streams in Py3k mode. (GH-11127)
1 parent 77b80c9 commit 1462234

File tree

4 files changed

+24
-3
lines changed

4 files changed

+24
-3
lines changed

Lib/test/test_fileio.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from UserList import UserList
1313

1414
from test.test_support import TESTFN, check_warnings, run_unittest, make_bad_fd
15-
from test.test_support import py3k_bytes as bytes, cpython_only
15+
from test.test_support import py3k_bytes as bytes, cpython_only, check_py3k_warnings
1616
from test.script_helper import run_python
1717

1818
from _io import FileIO as _FileIO
@@ -101,6 +101,10 @@ def test_none_args(self):
101101
self.assertEqual(self.f.readline(None), b"hi\n")
102102
self.assertEqual(self.f.readlines(None), [b"bye\n", b"abc"])
103103

104+
def testWriteUnicode(self):
105+
with check_py3k_warnings():
106+
self.f.write(u'')
107+
104108
def testRepr(self):
105109
self.assertEqual(repr(self.f), "<_io.FileIO name=%r mode='%s'>"
106110
% (self.f.name, self.f.mode))
@@ -210,7 +214,7 @@ def testErrnoOnClose(self, f):
210214

211215
@ClosedFDRaises
212216
def testErrnoOnClosedWrite(self, f):
213-
f.write('a')
217+
f.write(b'a')
214218

215219
@ClosedFDRaises
216220
def testErrnoOnClosedSeek(self, f):
Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
The write() method of buffered and unbuffered binary streams in the io
2+
module emits now a DeprecationWarning in Py3k mode for unicode argument.

Modules/_io/bufferedio.c

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1812,6 +1812,13 @@ bufferedwriter_write(buffered *self, PyObject *args)
18121812
if (!PyArg_ParseTuple(args, "s*:write", &buf)) {
18131813
return NULL;
18141814
}
1815+
if (PyUnicode_Check(PyTuple_GET_ITEM(args, 0)) &&
1816+
PyErr_WarnPy3k("write() argument must be string or buffer, "
1817+
"not 'unicode'", 1) < 0)
1818+
{
1819+
PyBuffer_Release(&buf);
1820+
return NULL;
1821+
}
18151822

18161823
if (IS_CLOSED(self)) {
18171824
PyErr_SetString(PyExc_ValueError, "write to closed file");

Modules/_io/fileio.c

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -716,8 +716,16 @@ fileio_write(fileio *self, PyObject *args)
716716
if (!self->writable)
717717
return err_mode("writing");
718718

719-
if (!PyArg_ParseTuple(args, "s*", &pbuf))
719+
if (!PyArg_ParseTuple(args, "s*:write", &pbuf)) {
720720
return NULL;
721+
}
722+
if (PyUnicode_Check(PyTuple_GET_ITEM(args, 0)) &&
723+
PyErr_WarnPy3k("write() argument must be string or buffer, "
724+
"not 'unicode'", 1) < 0)
725+
{
726+
PyBuffer_Release(&pbuf);
727+
return NULL;
728+
}
721729

722730
if (_PyVerify_fd(self->fd)) {
723731
Py_BEGIN_ALLOW_THREADS

0 commit comments

Comments
 (0)