abs-utils 0.2.3__tar.gz → 0.2.5__tar.gz

{abs_utils-0.2.3 → abs_utils-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: abs-utils
-Version: 0.2.3
+Version: 0.2.5
 Summary: AutoBridge Systems Utility Library
 Author: AutoBridgeSystems
 Author-email: info@autobridgesystems.com
@@ -12,6 +12,7 @@ Requires-Dist: aiohttp (>=3.11.18,<4.0.0)
 Requires-Dist: azure-identity (>=1.22.0,<2.0.0)
 Requires-Dist: azure-servicebus (>=7.14.2,<8.0.0)
 Requires-Dist: fastapi[standard] (>=0.115.12,<0.116.0)
+Requires-Dist: python-socketio (>=5.13.0,<6.0.0)
 Description-Content-Type: text/markdown
 
 # Utils Package

{abs_utils-0.2.3 → abs_utils-0.2.5}/abs_utils/azure_service_bus/event_decorator.py

@@ -4,7 +4,7 @@ from typing import Any, Callable
 from .azure_service_bus import AzureServiceBus
 from uuid import uuid4
 import json
-from ..constants import ON_RECORD_UPDATION
+
 def azure_event_decorator(event_type: str):
     """
     A decorator that sends a message to Azure Service Bus after the function execution.
@@ -44,9 +44,6 @@ def azure_event_decorator(event_type: str):
             
             # Execute the original function
             result = await func(*args, **kwargs)
-
-            if event_type != ON_RECORD_UPDATION:
-                data = result
             
             event_payload = {
                 "event_id": str(uuid4()),

abs_utils-0.2.5/abs_utils/schemas/libs/__init__.py

@@ -0,0 +1 @@
+from .make_optional import *

abs_utils-0.2.5/abs_utils/socket_io/__init__.py

@@ -0,0 +1,3 @@
+from .server import SocketIOService
+
+__all__ = ["SocketIOService"]

abs_utils-0.2.5/abs_utils/socket_io/server.py

@@ -0,0 +1,34 @@
+import socketio
+from fastapi import FastAPI
+from typing import Callable, Dict, Any
+from functools import wraps
+import asyncio
+from ..logger import setup_logger
+
+logger = setup_logger(__name__)
+class SocketIOService:
+    def __init__(self, cors_origins: str = "*"):
+        self.sio = socketio.AsyncServer(async_mode="asgi", cors_allowed_origins=cors_origins)
+
+    def bind_app(self, fastapi_app: FastAPI):
+        return socketio.ASGIApp(self.sio, other_asgi_app=fastapi_app)
+
+    async def emit_to_sid(self, sid: str, event: str, data: Dict[str, Any]):
+        await self.sio.emit(event, data, to=sid)
+
+    async def emit_broadcast(self, event: str, data: Dict[str, Any],namespace:str=None):
+        if namespace:
+            await self.sio.emit(event, data,namespace=namespace)
+        else:
+            await self.sio.emit(event, data)
+
+    async def join_room(self, sid: str, room: str):
+        await self.sio.enter_room(sid, room)
+        logger.info(f"{sid} joined room {room}")
+
+    async def leave_room(self, sid: str, room: str):
+        await self.sio.leave_room(sid, room)
+        logger.info(f"{sid} left room {room}")
+
+    async def emit_to_room(self, event: str, room: str, data: Dict[str, Any], namespace: str = None):
+        await self.sio.emit(event, data, room=room, namespace=namespace)

{abs_utils-0.2.3 → abs_utils-0.2.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "abs-utils"
-version = "0.2.3"
+version = "0.2.5"
 description = "AutoBridge Systems Utility Library"
 authors = [
     {name = "AutoBridgeSystems", email = "info@autobridgesystems.com"}
@@ -13,6 +13,7 @@ dependencies = [
     "aiohttp (>=3.11.18,<4.0.0)",
     "fastapi[standard] (>=0.115.12,<0.116.0)",
     "abs-exception-core (>=0.1.2,<0.2.0)",
+    "python-socketio (>=5.13.0,<6.0.0)",
 ]
 
 [build-system]