changeset:   99643:1e54adef4064
branch:      3.5
parent:      99640:5895cb9fc8d0
user:        Serhiy Storchaka <storchaka@gmail.com>
date:        Sun Dec 20 16:36:34 2015 +0200
files:       Lib/test/test_bytes.py Misc/NEWS Objects/bytesobject.c
description:
Issue #25766: Special method __bytes__() now works in str subclasses.


diff -r 5895cb9fc8d0 -r 1e54adef4064 Lib/test/test_bytes.py
--- a/Lib/test/test_bytes.py	Sun Dec 20 11:39:14 2015 +0200
+++ b/Lib/test/test_bytes.py	Sun Dec 20 16:36:34 2015 +0200
@@ -779,6 +779,12 @@
             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):
diff -r 5895cb9fc8d0 -r 1e54adef4064 Misc/NEWS
--- a/Misc/NEWS	Sun Dec 20 11:39:14 2015 +0200
+++ b/Misc/NEWS	Sun Dec 20 16:36:34 2015 +0200
@@ -10,6 +10,8 @@
 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.
diff -r 5895cb9fc8d0 -r 1e54adef4064 Objects/bytesobject.c
--- a/Objects/bytesobject.c	Sun Dec 20 11:39:14 2015 +0200
+++ b/Objects/bytesobject.c	Sun Dec 20 16:36:34 2015 +0200
@@ -3175,11 +3175,11 @@
         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);
@@ -3189,10 +3189,11 @@
         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;
     }
 
@@ -3217,6 +3218,11 @@
     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()) {