orionis 0.404.0__py3-none-any.whl → 0.406.0__py3-none-any.whl

orionis/foundation/config/queue/entities/queue.py

@@ -1,69 +1,57 @@
-from dataclasses import asdict, dataclass, field, fields
+from dataclasses import dataclass, field, fields
 from orionis.foundation.exceptions import OrionisIntegrityException
 from orionis.foundation.config.queue.entities.brokers import Brokers
+from orionis.support.entities.base import BaseEntity
 
 @dataclass(unsafe_hash=True, kw_only=True)
-class Queue:
+class Queue(BaseEntity):
+    """
+    Represents the configuration for a queue system.
+
+    Attributes:
+        default (str): The default queue connection to use. Must be a string.
+        brokers (Brokers | dict): The configuration for the queue brokers. Can be an instance of Brokers or a dictionary.
+
+    Methods:
+        __post_init__():
+            Validates and normalizes the properties after initialization.
+            Ensures 'default' is a string and 'brokers' is an instance of Brokers or a dictionary.
+    """
 
     default: str = field(
-        default="sync",
-        metadata={
+        default = "sync",
+        metadata = {
             "description": "The default queue connection to use.",
             "default": "sync"
         }
     )
 
-    brokers: Brokers = field(
-        default_factory=Brokers,
+    brokers: Brokers | dict = field(
+        default_factory = lambda: Brokers(),
         metadata={
             "description": "The default queue broker to use.",
-            "default": "Brokers()"
+            "default": Brokers().toDict()
         }
     )
 
     def __post_init__(self):
         """
         Post-initialization validation for the Queue entity.
+
         Validates and normalizes the following properties:
         - default: Must be a string.
         - brokers: Must be a string or an instance of the Brokers class.
         """
+
+        # Validate 'default' property
         options = [f.name for f in fields(Brokers)]
         if not isinstance(self.default, str) or self.default not in options:
             raise OrionisIntegrityException(
                 f"The 'default' property must be a string and match one of the available options ({options})."
             )
 
-        if not isinstance(self.brokers, Brokers):
-            raise OrionisIntegrityException("brokers must be an instance of the Brokers class.")
-
-    def toDict(self) -> dict:
-        """
-        Converts the current instance into a dictionary representation.
-
-        Returns:
-            dict: A dictionary containing all the fields of the instance.
-        """
-        return asdict(self)
-
-    def getFields(self):
-        """
-        Retrieves a list of field information for the current dataclass instance.
-
-        Returns:
-            list: A list of dictionaries, each containing details about a field:
-                - name (str): The name of the field.
-                - type (type): The type of the field.
-                - default: The default value of the field, if specified; otherwise, the value from metadata or None.
-                - metadata (mapping): The metadata associated with the field.
-        """
-        __fields = []
-        for field in fields(self):
-            __metadata = dict(field.metadata) or {}
-            __fields.append({
-                "name": field.name,
-                "type": field.type.__name__ if hasattr(field.type, '__name__') else str(field.type),
-                "default": field.default if (field.default is not None and '_MISSING_TYPE' not in str(field.default)) else __metadata.get('default', None),
-                "metadata": __metadata
-            })
-        return __fields
+        # Validate 'brokers' property
+        if not isinstance(self.brokers, (Brokers, dict)):
+            raise OrionisIntegrityException("brokers must be an instance of the Brokers class or a dictionary.")
+        if isinstance(self.brokers, dict):
+            self.brokers = Brokers(**self.brokers)