mirror of https://github.com/python/cpython
Issue #25766: Special method __bytes__() now works in str subclasses.
This commit is contained in:
commit
3182db356c
|
@ -793,6 +793,12 @@ class BytesTest(BaseBytesTest, unittest.TestCase):
|
|||
def __index__(self):
|
||||
return 42
|
||||
self.assertEqual(bytes(A()), b'a')
|
||||
# Issue #25766
|
||||
class A(str):
|
||||
def __bytes__(self):
|
||||
return b'abc'
|
||||
self.assertEqual(bytes(A('\u20ac')), b'abc')
|
||||
self.assertEqual(bytes(A('\u20ac'), 'iso8859-15'), b'\xa4')
|
||||
# Issue #24731
|
||||
class A:
|
||||
def __bytes__(self):
|
||||
|
|
|
@ -10,6 +10,8 @@ Release date: tba
|
|||
Core and Builtins
|
||||
-----------------
|
||||
|
||||
- Issue #25766: Special method __bytes__() now works in str subclasses.
|
||||
|
||||
- Issue #25421: __sizeof__ methods of builtin types now use dynamic basic size.
|
||||
This allows sys.getsize() to work correctly with their subclasses with
|
||||
__slots__ defined.
|
||||
|
|
|
@ -3320,11 +3320,11 @@ bytes_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
|||
return PyBytes_FromStringAndSize(NULL, 0);
|
||||
}
|
||||
|
||||
if (PyUnicode_Check(x)) {
|
||||
if (encoding != NULL) {
|
||||
/* Encode via the codec registry */
|
||||
if (encoding == NULL) {
|
||||
if (!PyUnicode_Check(x)) {
|
||||
PyErr_SetString(PyExc_TypeError,
|
||||
"string argument without an encoding");
|
||||
"encoding without a string argument");
|
||||
return NULL;
|
||||
}
|
||||
new = PyUnicode_AsEncodedString(x, encoding, errors);
|
||||
|
@ -3334,10 +3334,11 @@ bytes_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
|||
return new;
|
||||
}
|
||||
|
||||
/* If it's not unicode, there can't be encoding or errors */
|
||||
if (encoding != NULL || errors != NULL) {
|
||||
if (errors != NULL) {
|
||||
PyErr_SetString(PyExc_TypeError,
|
||||
"encoding or errors without a string argument");
|
||||
PyUnicode_Check(x) ?
|
||||
"string argument without an encoding" :
|
||||
"errors without a string argument");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
@ -3362,6 +3363,11 @@ bytes_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
|||
else if (PyErr_Occurred())
|
||||
return NULL;
|
||||
|
||||
if (PyUnicode_Check(x)) {
|
||||
PyErr_SetString(PyExc_TypeError,
|
||||
"string argument without an encoding");
|
||||
return NULL;
|
||||
}
|
||||
/* Is it an integer? */
|
||||
size = PyNumber_AsSsize_t(x, PyExc_OverflowError);
|
||||
if (size == -1 && PyErr_Occurred()) {
|
||||
|
|
Loading…
Reference in New Issue