TheCloud/panda3d
1
0
Fork 0
mirror of https://github.com/panda3d/panda3d.git synced 2025-10-01 09:23:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

dtoolutil: writelines() should accept any iterable

Browse Source
This commit is contained in:
rdb 2024-01-28 13:07:50 +01:00
parent 84ed141b2a
commit 128000b03a
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
dtool/src/dtoolutil/iostream_ext.cxx
View File

@ -283,19 +283,18 @@ write(PyObject *b) {
*/ */
void Extension<ostream>:: void Extension<ostream>::
writelines(PyObject *lines) { writelines(PyObject *lines) {
PyObject *seq = PySequence_Fast(lines, "writelines() expects a sequence"); PyObject *iter = PyObject_GetIter(lines);
if (seq == nullptr) { if (iter == nullptr) {
return; return;
} }
PyObject **items = PySequence_Fast_ITEMS(seq); PyObject *next = PyIter_Next(iter);
Py_ssize_t len = PySequence_Fast_GET_SIZE(seq); while (next != nullptr) {
write(next);
for (Py_ssize_t i = 0; i < len; ++i) { Py_DECREF(next);
write(items[i]); next = PyIter_Next(iter);
} }
Py_DECREF(iter);
Py_DECREF(seq);
} }
#endif // HAVE_PYTHON #endif // HAVE_PYTHON