bpo-39087: Add PyUnicode_GetUTF8Buffer().

python · methane · methane · methane · Dec 19, 2019 · methane
diff --git a/Lib/test/test_unicode.py b/Lib/test/test_unicode.py
+    def test_getutf8buffer(self):
+        from _testcapi import unicode_getutf8buffer
+
+        ascii = "foo"
diff --git a/Lib/test/test_unicode.py b/Lib/test/test_unicode.py
+        self.assertEqual(c1 + 1, c2)
+
+        mv.release()
+        del mv
diff --git a/Lib/test/test_unicode.py b/Lib/test/test_unicode.py
+        self.assertEqual(unicode_getutf8buffer(bmp).tobytes(), b'\xc4\x80')
+        self.assertEqual(unicode_getutf8buffer(bmp2).tobytes(), b'\xef\xbf\xbf')
+        self.assertEqual(unicode_getutf8buffer(nonbmp).tobytes(), b'\xf4\x8f\xbf\xbf')
+        self.assertRaises(UnicodeEncodeError, unicode_getutf8buffer, 'a\ud800b\udfffc')
diff --git a/Modules/_testcapimodule.c b/Modules/_testcapimodule.c
+    }
+    // Instead of calling PyBuffer_Release(), move the ownership of the obj
+    // to the memoryview object.
+    ((PyMemoryViewObject*)result)->mbuf->master.obj = buffer.obj;
diff --git a/Doc/c-api/unicode.rst b/Doc/c-api/unicode.rst
+
+   Successful calls to :c:func:`PyUnicode_GetUTF8Buffer` must be paired with
+   calls to :c:func:`PyBuffer_Release`, similar to :c:func:`PyObject_GetBuffer`.
+
diff --git a/Objects/unicodeobject.c b/Objects/unicodeobject.c
+                1, PyBUF_SIMPLE);
+    }
+
+    PyObject *bytes = _PyUnicode_AsUTF8String(unicode, errors);
diff --git a/Objects/unicodeobject.c b/Objects/unicodeobject.c
+        return -1;
+    }
+
+    if (PyUnicode_UTF8(unicode) != NULL) {