bpo-24234: Implement bytes.__bytes__ (GH-27901)

This commit is contained in:
Dong-hee Na 2021-08-23 10:01:51 +00:00 committed by GitHub
parent 6082bb5add
commit 24b63c695a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
7 changed files with 63 additions and 3 deletions

View File

@ -169,6 +169,14 @@ Other Language Changes
(Contributed by Serhiy Storchaka in :issue:`12022`.) (Contributed by Serhiy Storchaka in :issue:`12022`.)
Other CPython Implementation Changes
====================================
* Special methods :meth:`complex.__complex__` and :meth:`bytes.__bytes__` are implemented to
support :class:`typing.SupportsComplex` and :class:`typing.SupportsBytes` protocols.
(Contributed by Mark Dickinson and Dong-hee Na in :issue:`24234`.)
New Modules New Modules
=========== ===========

View File

@ -981,6 +981,18 @@ class BaseBytesTest:
class BytesTest(BaseBytesTest, unittest.TestCase): class BytesTest(BaseBytesTest, unittest.TestCase):
type2test = bytes type2test = bytes
def test__bytes__(self):
foo = b'foo'
self.assertEqual(foo.__bytes__(), foo)
self.assertEqual(type(foo.__bytes__()), self.type2test)
class bytes_subclass(bytes):
pass
bar = bytes_subclass(b'bar')
self.assertEqual(bar.__bytes__(), bar)
self.assertEqual(type(bar.__bytes__()), self.type2test)
def test_getitem_error(self): def test_getitem_error(self):
b = b'python' b = b'python'
msg = "byte indices must be integers or slices" msg = "byte indices must be integers or slices"

View File

@ -1543,11 +1543,11 @@ class ProtocolTests(BaseTestCase):
def test_supports_bytes(self): def test_supports_bytes(self):
# Note: bytes itself doesn't have __bytes__.
class B: class B:
def __bytes__(self): def __bytes__(self):
return b'' return b''
self.assertIsSubclass(bytes, typing.SupportsBytes)
self.assertIsSubclass(B, typing.SupportsBytes) self.assertIsSubclass(B, typing.SupportsBytes)
self.assertNotIsSubclass(str, typing.SupportsBytes) self.assertNotIsSubclass(str, typing.SupportsBytes)

View File

@ -0,0 +1,3 @@
Implement the :meth:`__bytes__` special method on the :class:`bytes` type,
so a bytes object ``b`` passes an ``isinstance(b, typing.SupportsBytes)``
check.

View File

@ -1687,6 +1687,25 @@ static PyBufferProcs bytes_as_buffer = {
}; };
/*[clinic input]
bytes.__bytes__
Convert this value to exact type bytes.
[clinic start generated code]*/
static PyObject *
bytes___bytes___impl(PyBytesObject *self)
/*[clinic end generated code: output=63a306a9bc0caac5 input=34ec5ddba98bd6bb]*/
{
if (PyBytes_CheckExact(self)) {
Py_INCREF(self);
return (PyObject *)self;
}
else {
return PyBytes_FromString(self->ob_sval);
}
}
#define LEFTSTRIP 0 #define LEFTSTRIP 0
#define RIGHTSTRIP 1 #define RIGHTSTRIP 1
#define BOTHSTRIP 2 #define BOTHSTRIP 2
@ -2474,6 +2493,7 @@ bytes_getnewargs(PyBytesObject *v, PyObject *Py_UNUSED(ignored))
static PyMethodDef static PyMethodDef
bytes_methods[] = { bytes_methods[] = {
{"__getnewargs__", (PyCFunction)bytes_getnewargs, METH_NOARGS}, {"__getnewargs__", (PyCFunction)bytes_getnewargs, METH_NOARGS},
BYTES___BYTES___METHODDEF
{"capitalize", stringlib_capitalize, METH_NOARGS, {"capitalize", stringlib_capitalize, METH_NOARGS,
_Py_capitalize__doc__}, _Py_capitalize__doc__},
STRINGLIB_CENTER_METHODDEF STRINGLIB_CENTER_METHODDEF

View File

@ -2,6 +2,24 @@
preserve preserve
[clinic start generated code]*/ [clinic start generated code]*/
PyDoc_STRVAR(bytes___bytes____doc__,
"__bytes__($self, /)\n"
"--\n"
"\n"
"Convert this value to exact type bytes.");
#define BYTES___BYTES___METHODDEF \
{"__bytes__", (PyCFunction)bytes___bytes__, METH_NOARGS, bytes___bytes____doc__},
static PyObject *
bytes___bytes___impl(PyBytesObject *self);
static PyObject *
bytes___bytes__(PyBytesObject *self, PyObject *Py_UNUSED(ignored))
{
return bytes___bytes___impl(self);
}
PyDoc_STRVAR(bytes_split__doc__, PyDoc_STRVAR(bytes_split__doc__,
"split($self, /, sep=None, maxsplit=-1)\n" "split($self, /, sep=None, maxsplit=-1)\n"
"--\n" "--\n"
@ -878,4 +896,4 @@ skip_optional_pos:
exit: exit:
return return_value; return return_value;
} }
/*[clinic end generated code: output=b3f0ec2753246b9c input=a9049054013a1b77]*/ /*[clinic end generated code: output=d706344859f40122 input=a9049054013a1b77]*/

View File

@ -2179,7 +2179,6 @@ unsupported_special_methods = set("""
__abs__ __abs__
__add__ __add__
__and__ __and__
__bytes__
__call__ __call__
__delitem__ __delitem__
__divmod__ __divmod__