From e3c61e2afe8e34dc4bc88caf7254f42be4d17b74 Mon Sep 17 00:00:00 2001 From: Markus Bauer Date: Tue, 18 May 2021 12:31:11 +0200 Subject: [PATCH] fix iterator --- src/SimpleSseSever.py | 17 +++-------------- 1 file changed, 3 insertions(+), 14 deletions(-) diff --git a/src/SimpleSseSever.py b/src/SimpleSseSever.py index 2924bc4..6b51046 100644 --- a/src/SimpleSseSever.py +++ b/src/SimpleSseSever.py @@ -1,6 +1,4 @@ import queue -import sse -import json from exceptions import TooManyConnections class Sse(object): @@ -75,13 +73,8 @@ class Sse(object): self._buffer = [] def __iter__(self): - return self - - def __next__(self): - print("Next in SSE") - print(self._buffer) for item in self._buffer: - return item + yield item self.flush() @@ -133,18 +126,14 @@ class SseStream(): def __iter__(self): print("Iter called") - return self - - def __next__(self): - """ Method which is called regular to send out the event """ - print("Next called") print(f'Queue Size: {self._queue.qsize()}') while True: msg = self._queue.get(block=True, timeout=None) if isinstance( msg, Sse ): + print("Message: ", msg) for data in msg: print(data) - return data.encode('utf-8') + yield data.encode('utf-8') def addMessage(self, event_name, data, id = 0): """