Fix remote stream termination

ml31415 · Jun 22, 2024 · 51ba36d · 51ba36d
1 parent 6dac1e1
commit 51ba36d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/betfair_parser/stream.py b/betfair_parser/stream.py
@@ -177,6 +177,8 @@ def iter_changes(self, stream: io.RawIOBase) -> Iterable[ChangeMessageType]:
 
         while True:
             msg = self.esm.receive(stream)
+            if not msg:
+                return
             if isinstance(msg, ChangeMessageType):  # type: ignore[arg-type,misc]
                 yield msg
 
@@ -191,6 +193,8 @@ def iter_changes_and_write(
         with open(path, "ab") as f:
             while True:
                 raw_msg = stream.readline()
+                if not raw_msg:
+                    return
                 msg = self.esm.receive_bytes(raw_msg)
                 if isinstance(msg, ChangeMessageType):  # type: ignore[arg-type,misc]
                     yield msg
@@ -268,6 +272,8 @@ async def iter_changes_async(self, stream: AsyncStream) -> AsyncGenerator[Change
 
         while True:
             msg = self.esm.receive_bytes(await stream.readline())
+            if not msg:
+                return
             if isinstance(msg, ChangeMessageType):  # type: ignore[arg-type,misc]
                 yield msg
 
@@ -283,6 +289,8 @@ async def iter_changes_and_write_async(
         with open(path, "ab") as f:
             while True:
                 raw_msg = await stream.readline()
+                if not raw_msg:
+                    return
                 msg = self.esm.receive_bytes(raw_msg)
                 if isinstance(msg, ChangeMessageType):  # type: ignore[arg-type,misc]
                     yield msg