diff --git a/src/SimpleSseSever.py b/src/SimpleSseSever.py index 6b51046..5877026 100644 --- a/src/SimpleSseSever.py +++ b/src/SimpleSseSever.py @@ -125,14 +125,10 @@ class SseStream(): print( "Stream object is deleted") def __iter__(self): - print("Iter 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) yield data.encode('utf-8') def addMessage(self, event_name, data, id = 0): @@ -148,10 +144,10 @@ class SseStream(): id: int id of the event, if set to 0 no ID will be sent """ event = Sse() - # if id > 0: - # event.set_event_id(id) - # else: - # event.reset_event_id() + if id > 0: + event.set_event_id(id) + else: + event.reset_event_id() event.add_message( event_name, data ) self._queue.put( event )