remove ugly generator function

cta-observatory · Feb 12, 2018 · a588a32 · a588a32
1 parent 65cd455
commit a588a32
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/protozfitsreader/__init__.py b/protozfitsreader/__init__.py
@@ -170,16 +170,17 @@ def __init__(self, fname):
         rawzfitsreader.open(self.fname + ":Events")
         self.numrows = rawzfitsreader.getNumRows()
         self.run_id = 0
-        self.generator = self.__generator()
-
-    def __generator(self):
-        for _ in range(self.numrows):
-            event = L0_pb2.CameraEvent()
-            event.ParseFromString(rawzfitsreader.readEvent())
-            yield Event(event, self.run_id)
 
     def __iter__(self):
-        return self.generator
+        return self
+
+    def __next__(self):
+        event = L0_pb2.CameraEvent()
+        try:
+            event.ParseFromString(rawzfitsreader.readEvent())
+            return Event(event, self.run_id)
+        except EOFError:
+            raise StopIteration
 
     def list_tables(self):
         return rawzfitsreader.listAllTables(self.fname)