abs-utils 0.1.0__tar.gz

abs_utils-0.1.0/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.3
+Name: abs-utils
+Version: 0.1.0
+Summary: 
+Author: Your Name
+Author-email: you@example.com
+Requires-Python: >=3.13,<4.0
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.13
+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)
+Description-Content-Type: text/markdown
+
+

abs_utils-0.1.0/abs_utils/azure_service_bus/azure_service_bus.py

@@ -0,0 +1,19 @@
+from azure.servicebus import ServiceBusClient, ServiceBusMessage
+import json
+
+class AzureServiceBus:
+    def __init__(self, connection_string: str, queue_name: str):
+        self.client = ServiceBusClient.from_connection_string(conn_str=connection_string)
+        self.queue_name = queue_name
+
+    async def send(self, event_payload: dict):
+        """Send message to Azure Service Bus"""
+        if not self.client or not self.queue_name:
+            return
+
+        try:
+            sender = self.client.get_queue_sender(queue_name=self.queue_name)
+            with sender:
+                sender.send_messages(ServiceBusMessage(json.dumps(event_payload)))
+        except Exception as e:
+            print(f"Failed to send message to Azure Service Bus: {e}") 

abs_utils-0.1.0/abs_utils/azure_service_bus/event_decorator.py

@@ -0,0 +1,52 @@
+from functools import wraps
+from fastapi import Request
+from typing import Any, Callable
+from .azure_service_bus import AzureServiceBus
+from uuid import uuid4
+import json
+
+def azure_event_decorator(event_type: str):
+    """
+    A decorator that sends a message to Azure Service Bus after the function execution.
+    
+    Args:
+        event_name (str): The name of the event to be sent
+    
+    Returns:
+        Callable: The decorated function
+    """
+    def decorator(func: Callable) -> Callable:
+        @wraps(func)
+        async def wrapper(*args: Any, **kwargs: Any) -> Any:
+            # Extract required parameters from kwargs
+            entity_name = kwargs.get('entity_name')
+            request = kwargs.get('request')
+            data = kwargs.get('data')
+            azure_service_bus:AzureServiceBus = kwargs.get('azure_service_bus')
+            id = kwargs.get('id')
+            
+            # Execute the original function
+            result = await func(*args, **kwargs)
+            
+            event_payload = {
+                "event_id": str(uuid4()),
+                "event_type": event_type,
+                "entity_name": entity_name,
+                "entity_id": id or None,
+                "payload": data,
+                "user": {
+                    "id": request.state.user.id,
+                    "uuid": request.state.user.uuid,
+                    "email": request.state.user.email,
+                    "name": request.state.user.name,
+                },
+            }
+            # Send message to Azure Service Bus
+            event_payload = json.dumps(event_payload)
+            print("==============event_payload======================")
+            print(event_payload)
+            print("==============event_payload======================")
+            await azure_service_bus.send(event_payload)
+            return result
+        return wrapper
+    return decorator

abs_utils-0.1.0/abs_utils/logger/logger.py

@@ -0,0 +1,31 @@
+import logging
+import os
+from typing import Optional
+
+def setup_logger(
+    name: str,
+    level: int = logging.INFO
+) -> logging.Logger:
+    """
+    Simple logger setup function.
+    
+    Args:
+        name (str): Name of the logger
+        log_file (Optional[str]): Path to log file (default: None)
+        level (int): Logging level (default: logging.INFO)
+    
+    Returns:
+        logging.Logger: Configured logger instance
+    """
+    # Create logger
+    logger = logging.getLogger(name)
+    logger.setLevel(level)
+    
+    # Create formatter
+    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+    
+    # Console handler
+    console_handler = logging.StreamHandler()
+    console_handler.setFormatter(formatter)
+    logger.addHandler(console_handler) 
+    return logger

abs_utils-0.1.0/pyproject.toml

@@ -0,0 +1,24 @@
+[project]
+name = "abs-utils"
+version = "0.1.0"
+description = ""
+authors = [
+    {name = "Your Name",email = "you@example.com"}
+]
+readme = "README.md"
+requires-python = ">=3.13,<4.0"
+dependencies = [
+    "azure-servicebus (>=7.14.2,<8.0.0)",
+    "azure-identity (>=1.22.0,<2.0.0)",
+    "aiohttp (>=3.11.18,<4.0.0)",
+    "fastapi[standard] (>=0.115.12,<0.116.0)",
+]
+
+[build-system]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.poetry]
+packages = [
+    {include = "abs_utils"}
+]