TheCloud/panda3d
1
0
Fork 0
mirror of https://github.com/panda3d/panda3d.git synced 2025-09-28 07:48:37 -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>::
writelines(PyObject *lines) {
PyObject *seq = PySequence_Fast(lines, "writelines() expects a sequence");
if (seq == nullptr) {
PyObject *iter = PyObject_GetIter(lines);
if (iter == nullptr) {
return;
}
PyObject **items = PySequence_Fast_ITEMS(seq);
Py_ssize_t len = PySequence_Fast_GET_SIZE(seq);
for (Py_ssize_t i = 0; i < len; ++i) {
write(items[i]);
PyObject *next = PyIter_Next(iter);
while (next != nullptr) {
write(next);
Py_DECREF(next);
next = PyIter_Next(iter);
}
Py_DECREF(seq);
Py_DECREF(iter);
}
#endif // HAVE_PYTHON