From 8d3284f261ba0b28716de067efbd27652471b24e Mon Sep 17 00:00:00 2001 From: Markus Bauer Date: Sun, 16 May 2021 23:17:07 +0200 Subject: [PATCH] Added "addMessage" method --- src/SimpleSseSever.py | 23 ++++++++++++++++++++++- 1 file changed, 22 insertions(+), 1 deletion(-) diff --git a/src/SimpleSseSever.py b/src/SimpleSseSever.py index 70a5452..b82d076 100644 --- a/src/SimpleSseSever.py +++ b/src/SimpleSseSever.py @@ -55,10 +55,31 @@ class SseStream(): """ Method which is called regular to send out the event """ while True: msg = self.queue.get(block=True, timeout=None) - if isinstance( msg, sse ): + if isinstance( msg, sse.Sse ): for data in msg: yield data + def addMessage(self, event_name, data, id = 0): + """ + Adds a event to the queue so it can be sent out + --- + Parameters: + ------ + event_name: str + Name of the event + data: string or list of strings + Datat to be send + id: int + id of the event, if set to 0 no ID will be sent """ + event = sse.Sse() + if id > 0: + event.set_event_id(id) + else: + event.reset_event_id() + event.add_message( event_name, data ) + + self._queue.put(event) +