abs-utils 0.2.7__tar.gz → 0.3.1__tar.gz

{abs_utils-0.2.7 → abs_utils-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: abs-utils
-Version: 0.2.7
+Version: 0.3.1
 Summary: AutoBridge Systems Utility Library
 Author: AutoBridgeSystems
 Author-email: info@autobridgesystems.com

abs_utils-0.3.1/abs_utils/azure_service_bus/azure_service_bus.py

@@ -0,0 +1,79 @@
+from azure.servicebus.aio import ServiceBusClient
+from azure.servicebus import ServiceBusMessage, ServiceBusReceiveMode
+from azure.servicebus.exceptions import ServiceBusError
+import json
+from typing import Optional
+from ..logger import setup_logger
+
+logger = setup_logger(__name__)
+
+class AzureServiceBus:
+    def __init__(self, connection_string: str, queue_name: str):
+        self.connection_string = connection_string
+        self.queue_name = queue_name
+        self.client = ServiceBusClient.from_connection_string(conn_str=connection_string)
+        self.sender = None
+        self.receiver = None
+
+    async def connect(self):
+        """Establish sender and receiver clients"""
+        try:
+            self.sender = self.client.get_queue_sender(queue_name=self.queue_name)
+            self.receiver = self.client.get_queue_receiver(
+                queue_name=self.queue_name,
+                receive_mode=ServiceBusReceiveMode.PEEK_LOCK
+            )
+            logger.info(f"Connected to queue: {self.queue_name}")
+        except ServiceBusError as e:
+            logger.error(f"Connection error: {e}")
+            raise
+
+    async def disconnect(self):
+        """Close connections"""
+        try:
+            if self.sender:
+                await self.sender.close()
+            if self.receiver:
+                await self.receiver.close()
+            await self.client.close()
+            logger.info("Disconnected from Service Bus")
+        except ServiceBusError as e:
+            logger.error(f"Disconnection error: {e}")
+            raise
+
+    async def send(self, event_payload: dict):
+        """Send a message to the queue"""
+        if not self.sender:
+            await self.connect()
+        try:
+            message = ServiceBusMessage(
+                body=json.dumps(event_payload),
+                content_type="application/json"
+            )
+            await self.sender.send_messages(message)
+            logger.info("Message sent")
+        except ServiceBusError as e:
+            logger.error(f"Send error: {e}")
+            raise
+
+    async def receive_messages(self, max_message_count: int = 1, timeout: int = 30):
+        """Receive messages from the queue"""
+        if not self.receiver:
+            await self.connect()
+        try:
+            return await self.receiver.receive_messages(
+                max_message_count=max_message_count,
+                max_wait_time=timeout
+            )
+        except ServiceBusError as e:
+            logger.error(f"Receive error: {e}")
+            raise
+
+    async def complete_message(self, message):
+        """Mark a received message as completed"""
+        try:
+            await self.receiver.complete_message(message)
+            logger.info("Message completed")
+        except ServiceBusError as e:
+            logger.error(f"Completion error: {e}")
+            raise

{abs_utils-0.2.7 → abs_utils-0.3.1}/abs_utils/schemas/fields_schema/common_schema.py

@@ -1,5 +1,5 @@
 from pydantic import BaseModel, Field, model_validator, ConfigDict
-from typing import List, Optional, Any
+from typing import List, Optional, Any, Union
 from enum import Enum
 
 class ValidationRules(BaseModel):
@@ -57,6 +57,12 @@ class Reference(BaseModel):
             raise ValueError("entity_id and alias cannot be empty")
         
         return self
+    
+class UserFilter(BaseModel):
+    is_active: Optional[bool] = Field(default=None, description="Whether the user is active")
+    ids: Optional[List[Union[str, Any]]] = Field(default=None, description="User ids")
+    
+    model_config = ConfigDict(arbitrary_types_allowed=True)
 
 class FieldType(str, Enum):
     """Supported field types for form fields."""
@@ -76,6 +82,7 @@ class FieldType(str, Enum):
     RICHTEXT = "richtext"
     LIST = "list"
     ASSOCIATION = "association"
+    USER = "user"
 
 
 class BaseFieldSchema(BaseModel):
@@ -92,6 +99,7 @@ class BaseFieldSchema(BaseModel):
     endpoint_url: Optional[str] = None
     metadata: Optional[dict] = None
     reference: Optional[Reference] = None
+    user_filter: Optional[UserFilter] = None
 
     model_config = ConfigDict(extra="ignore")
     

{abs_utils-0.2.7 → abs_utils-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "abs-utils"
-version = "0.2.7"
+version = "0.3.1"
 description = "AutoBridge Systems Utility Library"
 authors = [
     {name = "AutoBridgeSystems", email = "info@autobridgesystems.com"}

abs_utils-0.2.7/abs_utils/azure_service_bus/azure_service_bus.py

@@ -1,21 +0,0 @@
-from azure.servicebus import ServiceBusClient, ServiceBusMessage
-import json
-from ..logger import setup_logger
-
-logger = setup_logger(__name__)
-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:
-            logger.error(f"Failed to send message to Azure Service Bus: {e}")