Issue #7415: PyUnicode_FromEncodedObject() now uses the new buffer API

properly.  Patch by Stefan Behnel.
This commit is contained in:
Antoine Pitrou 2010-09-01 15:10:12 +00:00
parent f68c2a701b
commit b0fa831d1e
3 changed files with 30 additions and 27 deletions

View File

@ -62,6 +62,7 @@ David Beazley
Robin Becker
Neal Becker
Bill Bedford
Stefan Behnel
Reimer Behrends
Ben Bell
Thomas Bellman

View File

@ -12,6 +12,9 @@ What's New in Python 3.2 Alpha 2?
Core and Builtins
-----------------
- Issue #7415: PyUnicode_FromEncodedObject() now uses the new buffer API
properly. Patch by Stefan Behnel.
- Issue #5553: The Py_LOCAL_INLINE macro now results in inlining on
most platforms. Previously, it online inlined when using Microsoft
Visual C.

View File

@ -1234,8 +1234,7 @@ PyObject *PyUnicode_FromEncodedObject(register PyObject *obj,
const char *encoding,
const char *errors)
{
const char *s = NULL;
Py_ssize_t len;
Py_buffer buffer;
PyObject *v;
if (obj == NULL) {
@ -1243,44 +1242,44 @@ PyObject *PyUnicode_FromEncodedObject(register PyObject *obj,
return NULL;
}
/* Decoding bytes objects is the most common case and should be fast */
if (PyBytes_Check(obj)) {
if (PyBytes_GET_SIZE(obj) == 0) {
Py_INCREF(unicode_empty);
v = (PyObject *) unicode_empty;
}
else {
v = PyUnicode_Decode(
PyBytes_AS_STRING(obj), PyBytes_GET_SIZE(obj),
encoding, errors);
}
return v;
}
if (PyUnicode_Check(obj)) {
PyErr_SetString(PyExc_TypeError,
"decoding str is not supported");
return NULL;
}
/* Coerce object */
if (PyBytes_Check(obj)) {
s = PyBytes_AS_STRING(obj);
len = PyBytes_GET_SIZE(obj);
}
else if (PyByteArray_Check(obj)) {
s = PyByteArray_AS_STRING(obj);
len = PyByteArray_GET_SIZE(obj);
}
else if (PyObject_AsCharBuffer(obj, &s, &len)) {
/* Overwrite the error message with something more useful in
case of a TypeError. */
if (PyErr_ExceptionMatches(PyExc_TypeError))
/* Retrieve a bytes buffer view through the PEP 3118 buffer interface */
if (PyObject_GetBuffer(obj, &buffer, PyBUF_SIMPLE) < 0) {
PyErr_Format(PyExc_TypeError,
"coercing to str: need bytes, bytearray or char buffer, "
"%.80s found",
"coercing to str: need bytes, bytearray "
"or buffer-like object, %.80s found",
Py_TYPE(obj)->tp_name);
goto onError;
return NULL;
}
/* Convert to Unicode */
if (len == 0) {
if (buffer.len == 0) {
Py_INCREF(unicode_empty);
v = (PyObject *) unicode_empty;
}
else
v = PyUnicode_Decode(s, len, encoding, errors);
v = PyUnicode_Decode((char*) buffer.buf, buffer.len, encoding, errors);
PyBuffer_Release(&buffer);
return v;
onError:
return NULL;
}
/* Convert encoding to lower case and replace '_' with '-' in order to