zenbroker 1.1.0__py3-none-any.whl → 1.3.0__py3-none-any.whl

zenbroker/__init__.py

@@ -1,6 +1,7 @@
-from .client import ZenbrokerClient, PostPublishEventResponse
+from .client import ZenbrokerClient, PostPublishEventResponse, ZenBrokerIncommingMessage
 
 __all__ = [
     "ZenbrokerClient",
-    "PostPublishEventResponse"
+    "PostPublishEventResponse",
+    "ZenBrokerIncommingMessage"
 ]

zenbroker/client.py

@@ -1,6 +1,10 @@
+import json
+import threading
 import httpx
-from typing import Dict, Any
+from typing import Dict, Any, Set
 from pydantic import BaseModel, Field, HttpUrl
+import socketio
+import time
 
 class PostPublishEventPayload(BaseModel):
     applicationId: str = Field(..., description="Application ID")
@@ -11,13 +15,71 @@ class PostPublishEventResponse(BaseModel):
     message: str = Field(..., description="Message from the server")
     id: str = Field(..., description="ID of the message")
 
+class ZenBrokerIncommingMessage(BaseModel):
+    id: str = Field(..., description="ID of the message")
+    channel: str = Field(..., description="Channel of the message")
+    createdAt: str = Field(..., description="Created at timestamp")
+    data: str = Field(..., description="Data of the message")
+
 class ZenbrokerClient:
     def __init__(self, base_url: HttpUrl, application_id: str) -> None:
         self._url: str = str(base_url)
         self._application_id: str = str(application_id)
 
         self._api = httpx.Client(base_url=self._url)
+
+        # Socket.IO client
+        self._sio = socketio.Client(logger=True)
+        self._sio.connect(f"{self._url}?applicationId={self._application_id}", socketio_path="/ws")
+        self._sio.on("message", self._on_socket_message)
+        
+        # Channels
+        self._channels: Set[str] = set()
+
+        # Callbacks
+        self._callbacks: list[callable] = []
+
+    def _on_socket_message(self, data):
+        _message = ZenBrokerIncommingMessage.model_validate(json.loads(data))
+        for callback in self._callbacks:
+            callback(_message)
+    
+
+    def on_message(self, callback: callable) -> callable:
+        self._callbacks.append(callback)
+        
+        def unsubscribe():
+            if callback in self._callbacks:
+                self._callbacks.remove(callback)
+        
+        return unsubscribe
+
+    def subscribe(self, channel: str) -> bool:
+        if channel in self._channels:
+            return False
+
+        self._sio.emit("subscribe", channel)
+        self._channels.add(channel)
+
+        return True
     
+    def unsubscribe(self, channel: str) -> bool:
+        if channel not in self._channels:
+            return False
+
+        self._sio.emit("unsubscribe", channel)
+        self._channels.remove(channel)
+        return True
+    
+    def publish_ws(self, channel: str, data: Dict[str, Any]) -> bool:
+        post_data: PostPublishEventPayload = PostPublishEventPayload(
+            applicationId=self._application_id,
+            channel=channel,
+            data=data
+        )
+        self._sio.emit("message", data=post_data.model_dump())
+        return True
+
     def publish(self, channel: str, data: Dict[str, Any]) -> PostPublishEventResponse:
         post_data: PostPublishEventPayload = PostPublishEventPayload(
             applicationId=self._application_id,
@@ -38,3 +100,18 @@ class ZenbrokerClient:
             message=result['message']
         )
     
+    def listen(self):
+        """
+        Keeps the Socket.IO client running and listening for messages.
+        This method will not block the main thread.
+        """
+        def _listen():
+            try:
+                while True:
+                    time.sleep(1)
+            except KeyboardInterrupt:
+                self._sio.disconnect()
+
+        thread = threading.Thread(target=_listen)
+        thread.start()
+    

{zenbroker-1.1.0.dist-info → zenbroker-1.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zenbroker
-Version: 1.1.0
+Version: 1.3.0
 Summary: Zenbroker client
 Home-page: UNKNOWN
 License: UNKNOWN
@@ -8,6 +8,7 @@ Platform: UNKNOWN
 Description-Content-Type: text/markdown
 Requires-Dist: pydantic<3.0.0,>=2.0.0
 Requires-Dist: httpx<1.0.0,>=0.28.0
+Requires-Dist: python-socketio<6.0.0,>=5.0.0
 
 # Zenbroker Client Example
 

zenbroker-1.3.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+zenbroker/__init__.py,sha256=1FVuCI5MNs_uk5hukwIlkHiPhXVhLtc4DUYXNYh5WIY,190
+zenbroker/client.py,sha256=hwWF1St1vAzDEF8BalsHgZV3JYHqCcGZr9g-JSHz6Y8,3674
+zenbroker-1.3.0.dist-info/METADATA,sha256=mGtepoX1emIlq2kp9h-MDErAbFLUwXwhz79dumowDJI,1057
+zenbroker-1.3.0.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+zenbroker-1.3.0.dist-info/top_level.txt,sha256=u-V5a1mTLsnD3DQb01LQo8WtbYc75BIZ6jSWrZ7vDkY,10
+zenbroker-1.3.0.dist-info/RECORD,,

zenbroker-1.1.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-zenbroker/__init__.py,sha256=J-UtdbKCUJ2xJVcng2slPY81nl015BHBTNNSE6ZAwvA,130
-zenbroker/client.py,sha256=FDjIbV9yRhdbAsWp_kc-0GHqUYuAENlOI3b5l2ZB85o,1279
-zenbroker-1.1.0.dist-info/METADATA,sha256=1tE4Lq1SA_zvz-jdPrHBUn8vMqw9piwNNxTtW_lUnYU,1012
-zenbroker-1.1.0.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-zenbroker-1.1.0.dist-info/top_level.txt,sha256=u-V5a1mTLsnD3DQb01LQo8WtbYc75BIZ6jSWrZ7vDkY,10
-zenbroker-1.1.0.dist-info/RECORD,,