Fixed up cursor iteration more properly(?)

[automanga.git] / manga / lib.py

diff --git a/manga/lib.py b/manga/lib.py
index 662ecc3..da64e0c 100644 (file)
--- a/manga/lib.py
+++ b/manga/lib.py
@@ -161,7 +161,6 @@ class stdimgstream(imgstream):
     def __init__(self, url):
         import urllib.request
         req = urllib.request.Request(url, headers={"User-Agent": "automanga/1"})
-        print(req)
         self.bk = urllib.request.urlopen(req)
         ok = False
         try:
@@ -215,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):