mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-07-07 19:35:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

gh-125196: Use PyUnicodeWriter for repr(contextvars.Token) (#125220)

Browse source 
Replace the private _PyUnicodeWriter with the public PyUnicodeWriter.
This commit is contained in:
Victor Stinner 2024-10-10 01:37:14 +02:00 • committed by GitHub
parent 1b2a5485f9
commit 942916378a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 19 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

39
Python/context.c
View file

@ -1154,48 +1154,31 @@ token_tp_dealloc(PyContextToken *self)
static PyObject *
token_tp_repr(PyContextToken *self)
{
_PyUnicodeWriter writer;
_PyUnicodeWriter_Init(&writer);
if (_PyUnicodeWriter_WriteASCIIString(&writer, "<Token", 6) < 0) {
PyUnicodeWriter *writer = PyUnicodeWriter_Create(0);
if (writer == NULL) {
return NULL;
}
if (PyUnicodeWriter_WriteUTF8(writer, "<Token", 6) < 0) {
goto error;
}
if (self->tok_used) {
if (_PyUnicodeWriter_WriteASCIIString(&writer, " used", 5) < 0) {
if (PyUnicodeWriter_WriteUTF8(writer, " used", 5) < 0) {
goto error;
}
}
if (_PyUnicodeWriter_WriteASCIIString(&writer, " var=", 5) < 0) {
if (PyUnicodeWriter_WriteUTF8(writer, " var=", 5) < 0) {
goto error;
}
PyObject *var = PyObject_Repr((PyObject *)self->tok_var);
if (var == NULL) {
if (PyUnicodeWriter_WriteRepr(writer, (PyObject *)self->tok_var) < 0) {
goto error;
}
if (_PyUnicodeWriter_WriteStr(&writer, var) < 0) {
Py_DECREF(var);
if (PyUnicodeWriter_Format(writer, " at %p>", self) < 0) {
goto error;
}
Py_DECREF(var);
PyObject *addr = PyUnicode_FromFormat(" at %p>", self);
if (addr == NULL) {
goto error;
}
if (_PyUnicodeWriter_WriteStr(&writer, addr) < 0) {
Py_DECREF(addr);
goto error;
}
Py_DECREF(addr);
return _PyUnicodeWriter_Finish(&writer);
return PyUnicodeWriter_Finish(writer);
error:
_PyUnicodeWriter_Dealloc(&writer);
PyUnicodeWriter_Discard(writer);
return NULL;
}