mirror of https://github.com/python/cpython
bpo-24234: Implement bytes.__bytes__ (GH-27901)
This commit is contained in:
parent
6082bb5add
commit
24b63c695a
|
@ -169,6 +169,14 @@ Other Language Changes
|
|||
(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
|
||||
===========
|
||||
|
||||
|
|
|
@ -981,6 +981,18 @@ class BaseBytesTest:
|
|||
class BytesTest(BaseBytesTest, unittest.TestCase):
|
||||
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):
|
||||
b = b'python'
|
||||
msg = "byte indices must be integers or slices"
|
||||
|
|
|
@ -1543,11 +1543,11 @@ class ProtocolTests(BaseTestCase):
|
|||
|
||||
def test_supports_bytes(self):
|
||||
|
||||
# Note: bytes itself doesn't have __bytes__.
|
||||
class B:
|
||||
def __bytes__(self):
|
||||
return b''
|
||||
|
||||
self.assertIsSubclass(bytes, typing.SupportsBytes)
|
||||
self.assertIsSubclass(B, typing.SupportsBytes)
|
||||
self.assertNotIsSubclass(str, typing.SupportsBytes)
|
||||
|
||||
|
|
|
@ -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.
|
|
@ -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 RIGHTSTRIP 1
|
||||
#define BOTHSTRIP 2
|
||||
|
@ -2474,6 +2493,7 @@ bytes_getnewargs(PyBytesObject *v, PyObject *Py_UNUSED(ignored))
|
|||
static PyMethodDef
|
||||
bytes_methods[] = {
|
||||
{"__getnewargs__", (PyCFunction)bytes_getnewargs, METH_NOARGS},
|
||||
BYTES___BYTES___METHODDEF
|
||||
{"capitalize", stringlib_capitalize, METH_NOARGS,
|
||||
_Py_capitalize__doc__},
|
||||
STRINGLIB_CENTER_METHODDEF
|
||||
|
|
|
@ -2,6 +2,24 @@
|
|||
preserve
|
||||
[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__,
|
||||
"split($self, /, sep=None, maxsplit=-1)\n"
|
||||
"--\n"
|
||||
|
@ -878,4 +896,4 @@ skip_optional_pos:
|
|||
exit:
|
||||
return return_value;
|
||||
}
|
||||
/*[clinic end generated code: output=b3f0ec2753246b9c input=a9049054013a1b77]*/
|
||||
/*[clinic end generated code: output=d706344859f40122 input=a9049054013a1b77]*/
|
||||
|
|
|
@ -2179,7 +2179,6 @@ unsupported_special_methods = set("""
|
|||
__abs__
|
||||
__add__
|
||||
__and__
|
||||
__bytes__
|
||||
__call__
|
||||
__delitem__
|
||||
__divmod__
|
||||
|
|
Loading…
Reference in New Issue