Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-10-31 13:41:24 +00:00
Code Issues Projects Releases Packages Wiki Activity

gh-119182: Add PyUnicodeWriter_WriteUCS4() function (#120849)

Browse source
This commit is contained in:
Victor Stinner 2024-06-24 17:40:39 +02:00 • committed by GitHub
parent a47abdb45d
commit 2e157851e3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 131 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

31
Modules/_testcapi/unicode.c
View file

@ -360,6 +360,36 @@ writer_write_widechar(PyObject *self_raw, PyObject *args)
}
static PyObject*
writer_write_ucs4(PyObject *self_raw, PyObject *args)
{
WriterObject *self = (WriterObject *)self_raw;
if (writer_check(self) < 0) {
return NULL;
}
PyObject *str;
Py_ssize_t size;
if (!PyArg_ParseTuple(args, "Un", &str, &size)) {
return NULL;
}
Py_ssize_t len = PyUnicode_GET_LENGTH(str);
size = Py_MIN(size, len);
Py_UCS4 *ucs4 = PyUnicode_AsUCS4Copy(str);
if (ucs4 == NULL) {
return NULL;
}
int res = PyUnicodeWriter_WriteUCS4(self->writer, ucs4, size);
PyMem_Free(ucs4);
if (res < 0) {
return NULL;
}
Py_RETURN_NONE;
}
static PyObject*
writer_write_str(PyObject *self_raw, PyObject *args)
{
@ -484,6 +514,7 @@ static PyMethodDef writer_methods[] = {
{"write_char", _PyCFunction_CAST(writer_write_char), METH_VARARGS},
{"write_utf8", _PyCFunction_CAST(writer_write_utf8), METH_VARARGS},
{"write_widechar", _PyCFunction_CAST(writer_write_widechar), METH_VARARGS},
{"write_ucs4", _PyCFunction_CAST(writer_write_ucs4), METH_VARARGS},
{"write_str", _PyCFunction_CAST(writer_write_str), METH_VARARGS},
{"write_repr", _PyCFunction_CAST(writer_write_repr), METH_VARARGS},
{"write_substring", _PyCFunction_CAST(writer_write_substring), METH_VARARGS},