From 9d051be1387a18756e9e01e93e8d52ecd53cadd1 Mon Sep 17 00:00:00 2001 From: Fredrik Tolf Date: Mon, 15 May 2017 05:05:28 +0200 Subject: [PATCH] Fixed up cursor iteration more properly(?) --- manga/lib.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/manga/lib.py b/manga/lib.py index 8cc96cb..da64e0c 100644 --- a/manga/lib.py +++ b/manga/lib.py @@ -206,9 +206,6 @@ class cursor(object): return self.cur raise StopIteration() - def __next__(self): - return self.next() - def prev(self): for n, i in reversed(self.cur.stack): if i > 0: @@ -217,7 +214,14 @@ class cursor(object): raise StopIteration() def __iter__(self): - return self + def iterator(): + yield self.cur + while True: + try: + yield self.next() + except StopIteration: + break + return iterator() loaded = {} def findlib(name): -- 2.11.0