gh-133968: Add fast path to PyUnicodeWriter_WriteStr() (#133969)

Don't call PyObject_Str() if the input type is str.
This commit is contained in:
Victor Stinner 2025-05-13 15:31:41 +02:00 committed by GitHub
parent fc3cddd90a
commit fe9f6e829a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -13929,7 +13929,12 @@ _PyUnicodeWriter_WriteStr(_PyUnicodeWriter *writer, PyObject *str)
int
PyUnicodeWriter_WriteStr(PyUnicodeWriter *writer, PyObject *obj)
{
if (Py_TYPE(obj) == &PyLong_Type) {
PyTypeObject *type = Py_TYPE(obj);
if (type == &PyUnicode_Type) {
return _PyUnicodeWriter_WriteStr((_PyUnicodeWriter*)writer, obj);
}
if (type == &PyLong_Type) {
return _PyLong_FormatWriter((_PyUnicodeWriter*)writer, obj, 10, 0);
}