diff options
| -rw-r--r-- | Modules/itertoolsmodule.c | 153 |
1 files changed, 24 insertions, 129 deletions
diff --git a/Modules/itertoolsmodule.c b/Modules/itertoolsmodule.c index 3515bc6..21659fb 100644 --- a/Modules/itertoolsmodule.c +++ b/Modules/itertoolsmodule.c @@ -64,33 +64,11 @@ groupby_dealloc(groupbyobject *gbo) static int groupby_traverse(groupbyobject *gbo, visitproc visit, void *arg) { - int err; - - if (gbo->it) { - err = visit(gbo->it, arg); - if (err) - return err; - } - if (gbo->keyfunc) { - err = visit(gbo->keyfunc, arg); - if (err) - return err; - } - if (gbo->tgtkey) { - err = visit(gbo->tgtkey, arg); - if (err) - return err; - } - if (gbo->currkey) { - err = visit(gbo->currkey, arg); - if (err) - return err; - } - if (gbo->currvalue) { - err = visit(gbo->currvalue, arg); - if (err) - return err; - } + Py_VISIT(gbo->it); + Py_VISIT(gbo->keyfunc); + Py_VISIT(gbo->tgtkey); + Py_VISIT(gbo->currkey); + Py_VISIT(gbo->currvalue); return 0; } @@ -671,18 +649,8 @@ cycle_dealloc(cycleobject *lz) static int cycle_traverse(cycleobject *lz, visitproc visit, void *arg) { - int err; - - if (lz->it) { - err = visit(lz->it, arg); - if (err) - return err; - } - if (lz->saved) { - err = visit(lz->saved, arg); - if (err) - return err; - } + Py_VISIT(lz->it); + Py_VISIT(lz->saved); return 0; } @@ -820,18 +788,8 @@ dropwhile_dealloc(dropwhileobject *lz) static int dropwhile_traverse(dropwhileobject *lz, visitproc visit, void *arg) { - int err; - - if (lz->it) { - err = visit(lz->it, arg); - if (err) - return err; - } - if (lz->func) { - err = visit(lz->func, arg); - if (err) - return err; - } + Py_VISIT(lz->it); + Py_VISIT(lz->func); return 0; } @@ -971,18 +929,8 @@ takewhile_dealloc(takewhileobject *lz) static int takewhile_traverse(takewhileobject *lz, visitproc visit, void *arg) { - int err; - - if (lz->it) { - err = visit(lz->it, arg); - if (err) - return err; - } - if (lz->func) { - err = visit(lz->func, arg); - if (err) - return err; - } + Py_VISIT(lz->it); + Py_VISIT(lz->func); return 0; } @@ -1167,8 +1115,7 @@ islice_dealloc(isliceobject *lz) static int islice_traverse(isliceobject *lz, visitproc visit, void *arg) { - if (lz->it) - return visit(lz->it, arg); + Py_VISIT(lz->it); return 0; } @@ -1309,18 +1256,8 @@ starmap_dealloc(starmapobject *lz) static int starmap_traverse(starmapobject *lz, visitproc visit, void *arg) { - int err; - - if (lz->it) { - err = visit(lz->it, arg); - if (err) - return err; - } - if (lz->func) { - err = visit(lz->func, arg); - if (err) - return err; - } + Py_VISIT(lz->it); + Py_VISIT(lz->func); return 0; } @@ -1462,18 +1399,8 @@ imap_dealloc(imapobject *lz) static int imap_traverse(imapobject *lz, visitproc visit, void *arg) { - int err; - - if (lz->iters) { - err = visit(lz->iters, arg); - if (err) - return err; - } - if (lz->func) { - err = visit(lz->func, arg); - if (err) - return err; - } + Py_VISIT(lz->iters); + Py_VISIT(lz->func); return 0; } @@ -1648,8 +1575,7 @@ chain_dealloc(chainobject *lz) static int chain_traverse(chainobject *lz, visitproc visit, void *arg) { - if (lz->ittuple) - return visit(lz->ittuple, arg); + Py_VISIT(lz->ittuple); return 0; } @@ -1778,18 +1704,8 @@ ifilter_dealloc(ifilterobject *lz) static int ifilter_traverse(ifilterobject *lz, visitproc visit, void *arg) { - int err; - - if (lz->it) { - err = visit(lz->it, arg); - if (err) - return err; - } - if (lz->func) { - err = visit(lz->func, arg); - if (err) - return err; - } + Py_VISIT(lz->it); + Py_VISIT(lz->func); return 0; } @@ -1929,18 +1845,8 @@ ifilterfalse_dealloc(ifilterfalseobject *lz) static int ifilterfalse_traverse(ifilterfalseobject *lz, visitproc visit, void *arg) { - int err; - - if (lz->it) { - err = visit(lz->it, arg); - if (err) - return err; - } - if (lz->func) { - err = visit(lz->func, arg); - if (err) - return err; - } + Py_VISIT(lz->it); + Py_VISIT(lz->func); return 0; } @@ -2199,18 +2105,8 @@ izip_dealloc(izipobject *lz) static int izip_traverse(izipobject *lz, visitproc visit, void *arg) { - int err; - - if (lz->ittuple) { - err = visit(lz->ittuple, arg); - if (err) - return err; - } - if (lz->result) { - err = visit(lz->result, arg); - if (err) - return err; - } + Py_VISIT(lz->ittuple); + Py_VISIT(lz->result); return 0; } @@ -2357,8 +2253,7 @@ repeat_dealloc(repeatobject *ro) static int repeat_traverse(repeatobject *ro, visitproc visit, void *arg) { - if (ro->element) - return visit(ro->element, arg); + Py_VISIT(ro->element); return 0; } |