rewrite IpReader iterator to accumulate IPs before yielding

still needs work

respodns/dns.py

@@ -225,7 +225,7 @@ async def main(db, filepaths, checks, opts: Options):
             for i, ip in enumerate(read_ips(f)):
                 first = i == 0
                 if opts.progress:
-                    print(f"#{i}: {ip}", file=stderr)
+                    print(f"#{i + 1}: {ip}", file=stderr)
                     stderr.flush()
                 if not first:
                     await sleep(opts.ip_wait)
@@ -239,7 +239,7 @@ async def main(db, filepaths, checks, opts: Options):
             for i, ip in enumerate(reader):
                 first = i == 0
                 if opts.progress:
-                    print(f"#{i}/{reader.total}: {ip}", file=stderr)
+                    print(f"#{i + 1}/{reader.total}: {ip}", file=stderr)
                     stderr.flush()
                 if not first:
                     await sleep(opts.ip_wait)

respodns/ip_util.py

@@ -52,27 +52,29 @@ class IpReader:
         self.threads = []
         self.total = 0
 
-    def running(self):
+    def is_running(self):
         return any(thread.is_alive() for thread in self.threads)
 
     def __iter__(self):
-        return self
-
-    def __next__(self):
-        # TODO: rewrite such that self.total is useful. (get as many at once)
         from queue import Empty
+        from sys import stderr
 
-        while self.running() or not self.queue.empty():
+        def _next():
-            try:
+            while self.is_running() or not self.queue.empty():
-                res = self.queue.get(block=True, timeout=1.0)
+                results = []
-                if res is not None:
+                if self.queue.empty():
-                    self.total += 1
+                    try:
-                    return res
+                        results.append(self.queue.get(timeout=1.0))
-            except Empty:
+                    except Empty:
-                from sys import stderr
+                        print("blocking on IpReader", file=stderr)
-                print("blocking on IpReader", file=stderr)
+                else:
+                    while not self.queue.empty():
+                        results.append(self.queue.get())
+                self.total += len(results)
+                for res in results:
+                    yield res
 
-        raise StopIteration
+        return _next()
 
     def __enter__(self):
         from threading import Thread