Small optimization for corner case where maxlen==0.

This commit is contained in:
Raymond Hettinger 2009-03-10 09:36:07 +00:00
parent 2cf03a8204
commit 060c7f6bba
2 changed files with 41 additions and 1 deletions

View file

@ -276,6 +276,23 @@ deque_appendleft(dequeobject *deque, PyObject *item)
PyDoc_STRVAR(appendleft_doc, "Add an element to the left side of the deque.");
/* Run an iterator to exhaustion. Shortcut for
the extend/extendleft methods when maxlen == 0. */
static PyObject*
consume_iterator(PyObject *it)
{
PyObject *item;
while ((item = PyIter_Next(it)) != NULL) {
Py_DECREF(item);
}
Py_DECREF(it);
if (PyErr_Occurred())
return NULL;
Py_RETURN_NONE;
}
static PyObject *
deque_extend(dequeobject *deque, PyObject *iterable)
{
@ -285,6 +302,9 @@ deque_extend(dequeobject *deque, PyObject *iterable)
if (it == NULL)
return NULL;
if (deque->maxlen == 0)
return consume_iterator(it);
while ((item = PyIter_Next(it)) != NULL) {
deque->state++;
if (deque->rightindex == BLOCKLEN-1) {
@ -323,6 +343,9 @@ deque_extendleft(dequeobject *deque, PyObject *iterable)
if (it == NULL)
return NULL;
if (deque->maxlen == 0)
return consume_iterator(it);
while ((item = PyIter_Next(it)) != NULL) {
deque->state++;
if (deque->leftindex == 0) {