jararaca 0.3.11a7__py3-none-any.whl → 0.3.11a9__py3-none-any.whl

jararaca/utils/rabbitmq_utils.py

@@ -1,4 +1,9 @@
+import logging
+
 from aio_pika.abc import AbstractChannel, AbstractExchange, AbstractQueue
+from aio_pika.exceptions import AMQPError, ChannelClosed, ChannelNotFoundEntity
+
+logger = logging.getLogger(__name__)
 
 
 class RabbitmqUtils:
@@ -6,6 +11,39 @@ class RabbitmqUtils:
     DEAD_LETTER_EXCHANGE = "dlx"
     DEAD_LETTER_QUEUE = "dlq"
 
+    # Note: get_worker_v1_queue method is already defined above
+
+    DEAD_LETTER_EXCHANGE = "dlx"
+    DEAD_LETTER_QUEUE = "dlq"
+
+    @classmethod
+    async def get_dl_exchange(cls, channel: AbstractChannel) -> AbstractExchange:
+        """
+        Get the Dead Letter Exchange (DLX) for the given channel.
+        """
+        try:
+            return await channel.get_exchange(
+                cls.DEAD_LETTER_EXCHANGE,
+            )
+        except ChannelNotFoundEntity as e:
+            logger.error(
+                f"Dead Letter Exchange '{cls.DEAD_LETTER_EXCHANGE}' does not exist. "
+                f"Please use the declare command to create it first. Error: {e}"
+            )
+            raise
+        except ChannelClosed as e:
+            logger.error(
+                f"Channel closed while getting Dead Letter Exchange '{cls.DEAD_LETTER_EXCHANGE}'. "
+                f"Error: {e}"
+            )
+            raise
+        except AMQPError as e:
+            logger.error(
+                f"AMQP error while getting Dead Letter Exchange '{cls.DEAD_LETTER_EXCHANGE}'. "
+                f"Error: {e}"
+            )
+            raise
+
     @classmethod
     async def declare_dl_exchange(
         cls, channel: AbstractChannel, passive: bool
@@ -22,6 +60,34 @@ class RabbitmqUtils:
             auto_delete=False,
         )
 
+    @classmethod
+    async def get_dl_queue(cls, channel: AbstractChannel) -> AbstractQueue:
+        """
+        Get the Dead Letter Queue (DLQ) for the given channel.
+        """
+        try:
+            return await channel.get_queue(
+                cls.DEAD_LETTER_QUEUE,
+            )
+        except ChannelNotFoundEntity as e:
+            logger.error(
+                f"Dead Letter Queue '{cls.DEAD_LETTER_QUEUE}' does not exist. "
+                f"Please use the declare command to create it first. Error: {e}"
+            )
+            raise
+        except ChannelClosed as e:
+            logger.error(
+                f"Channel closed while getting Dead Letter Queue '{cls.DEAD_LETTER_QUEUE}'. "
+                f"Error: {e}"
+            )
+            raise
+        except AMQPError as e:
+            logger.error(
+                f"AMQP error while getting Dead Letter Queue '{cls.DEAD_LETTER_QUEUE}'. "
+                f"Error: {e}"
+            )
+            raise
+
     @classmethod
     async def declare_dl_queue(
         cls, channel: AbstractChannel, passive: bool
@@ -40,6 +106,36 @@ class RabbitmqUtils:
             },
         )
 
+    @classmethod
+    async def get_dl_kit(
+        cls,
+        channel: AbstractChannel,
+    ) -> tuple[AbstractExchange, AbstractQueue]:
+        """
+        Get the Dead Letter Exchange and Queue (DLX and DLQ) for the given channel.
+        """
+        try:
+            dlx = await cls.get_dl_exchange(channel)
+            dlq = await cls.get_dl_queue(channel)
+            return dlx, dlq
+        except ChannelNotFoundEntity as e:
+            logger.error(
+                f"Dead Letter infrastructure does not exist completely. "
+                f"Please use the declare command to create it first. Error: {e}"
+            )
+            raise
+        except ChannelClosed as e:
+            logger.error(
+                f"Channel closed while getting Dead Letter infrastructure. "
+                f"Error: {e}"
+            )
+            raise
+        except AMQPError as e:
+            logger.error(
+                f"AMQP error while getting Dead Letter infrastructure. " f"Error: {e}"
+            )
+            raise
+
     @classmethod
     async def declare_dl_kit(
         cls,
@@ -54,6 +150,33 @@ class RabbitmqUtils:
         await dlq.bind(dlx, routing_key=cls.DEAD_LETTER_EXCHANGE)
         return dlx, dlq
 
+    @classmethod
+    async def get_main_exchange(
+        cls, channel: AbstractChannel, exchange_name: str
+    ) -> AbstractExchange:
+        """
+        Get the main exchange for the given channel.
+        """
+        try:
+            return await channel.get_exchange(exchange_name)
+        except ChannelNotFoundEntity as e:
+            logger.error(
+                f"Exchange '{exchange_name}' does not exist. "
+                f"Please use the declare command to create it first. Error: {e}"
+            )
+            raise
+        except ChannelClosed as e:
+            logger.error(
+                f"Channel closed while getting exchange '{exchange_name}'. "
+                f"Error: {e}"
+            )
+            raise
+        except AMQPError as e:
+            logger.error(
+                f"AMQP error while getting exchange '{exchange_name}'. " f"Error: {e}"
+            )
+            raise
+
     @classmethod
     async def declare_main_exchange(
         cls, channel: AbstractChannel, exchange_name: str, passive: bool
@@ -70,6 +193,34 @@ class RabbitmqUtils:
             auto_delete=False,
         )
 
+    @classmethod
+    async def get_queue(
+        cls,
+        channel: AbstractChannel,
+        queue_name: str,
+    ) -> AbstractQueue:
+        """
+        Get a queue with the given name.
+        """
+        try:
+            return await channel.get_queue(queue_name)
+        except ChannelNotFoundEntity as e:
+            logger.error(
+                f"Queue '{queue_name}' does not exist. "
+                f"Please use the declare command to create it first. Error: {e}"
+            )
+            raise
+        except ChannelClosed as e:
+            logger.error(
+                f"Channel closed while getting queue '{queue_name}'. " f"Error: {e}"
+            )
+            raise
+        except AMQPError as e:
+            logger.error(
+                f"AMQP error while getting queue '{queue_name}'. " f"Error: {e}"
+            )
+            raise
+
     @classmethod
     async def declare_queue(
         cls,
@@ -90,3 +241,152 @@ class RabbitmqUtils:
                 "x-dead-letter-routing-key": cls.DEAD_LETTER_EXCHANGE,
             },
         )
+
+    @classmethod
+    async def get_worker_v1_queue(
+        cls,
+        channel: AbstractChannel,
+        queue_name: str,
+    ) -> AbstractQueue:
+        """
+        Get a worker v1 queue.
+        """
+        try:
+            return await channel.get_queue(queue_name)
+        except ChannelNotFoundEntity as e:
+            logger.error(
+                f"Worker queue '{queue_name}' does not exist. "
+                f"Please use the declare command to create it first. Error: {e}"
+            )
+            raise
+        except ChannelClosed as e:
+            logger.error(
+                f"Channel closed while getting worker queue '{queue_name}'. "
+                f"Error: {e}"
+            )
+            raise
+        except AMQPError as e:
+            logger.error(
+                f"AMQP error while getting worker queue '{queue_name}'. " f"Error: {e}"
+            )
+            raise
+
+    @classmethod
+    async def declare_worker_v1_queue(
+        cls,
+        channel: AbstractChannel,
+        queue_name: str,
+        dlx_name: str,
+        dlq_name: str,
+        passive: bool = False,
+    ) -> AbstractQueue:
+        """
+        Declare a worker v1 queue with custom dead letter exchange and routing key.
+        """
+        return await channel.declare_queue(
+            passive=passive,
+            name=queue_name,
+            arguments={
+                "x-dead-letter-exchange": dlx_name,
+                "x-dead-letter-routing-key": dlq_name,
+            },
+            durable=True,
+        )
+
+    @classmethod
+    async def get_scheduler_queue(
+        cls,
+        channel: AbstractChannel,
+        queue_name: str,
+    ) -> AbstractQueue:
+        """
+        Get a scheduler queue.
+        """
+        try:
+            return await channel.get_queue(queue_name)
+        except ChannelNotFoundEntity as e:
+            logger.error(
+                f"Scheduler queue '{queue_name}' does not exist. "
+                f"Please use the declare command to create it first. Error: {e}"
+            )
+            raise
+        except ChannelClosed as e:
+            logger.error(
+                f"Channel closed while getting scheduler queue '{queue_name}'. "
+                f"Error: {e}"
+            )
+            raise
+        except AMQPError as e:
+            logger.error(
+                f"AMQP error while getting scheduler queue '{queue_name}'. "
+                f"Error: {e}"
+            )
+            raise
+
+    @classmethod
+    async def declare_scheduler_queue(
+        cls,
+        channel: AbstractChannel,
+        queue_name: str,
+        passive: bool = False,
+    ) -> AbstractQueue:
+        """
+        Declare a scheduler queue with simple durable configuration.
+        """
+        return await channel.declare_queue(
+            name=queue_name,
+            durable=True,
+            passive=passive,
+        )
+
+    @classmethod
+    async def delete_exchange(
+        cls,
+        channel: AbstractChannel,
+        exchange_name: str,
+        if_unused: bool = False,
+    ) -> None:
+        """
+        Delete an exchange.
+        """
+        try:
+            await channel.exchange_delete(
+                exchange_name=exchange_name,
+                if_unused=if_unused,
+            )
+        except ChannelNotFoundEntity:
+            # Exchange might not exist, which is fine
+            logger.info(
+                f"Exchange '{exchange_name}' does not exist, nothing to delete."
+            )
+        except ChannelClosed as e:
+            logger.warning(
+                f"Channel closed while deleting exchange '{exchange_name}': {e}"
+            )
+        except AMQPError as e:
+            logger.warning(f"AMQP error while deleting exchange '{exchange_name}': {e}")
+
+    @classmethod
+    async def delete_queue(
+        cls,
+        channel: AbstractChannel,
+        queue_name: str,
+        if_unused: bool = False,
+        if_empty: bool = False,
+    ) -> None:
+        """
+        Delete a queue.
+        """
+        try:
+            await channel.queue_delete(
+                queue_name=queue_name,
+                if_unused=if_unused,
+                if_empty=if_empty,
+            )
+        except ChannelNotFoundEntity:
+            # Queue might not exist, which is fine
+            logger.info(f"Queue '{queue_name}' does not exist, nothing to delete.")
+        except ChannelClosed as e:
+            logger.warning(f"Channel closed while deleting queue '{queue_name}': {e}")
+        except AMQPError as e:
+            logger.warning(f"AMQP error while deleting queue '{queue_name}': {e}")

{jararaca-0.3.11a7.dist-info → jararaca-0.3.11a9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: jararaca
-Version: 0.3.11a7
+Version: 0.3.11a9
 Summary: A simple and fast API framework for Python
 Author: Lucas S
 Author-email: me@luscasleo.dev

{jararaca-0.3.11a7.dist-info → jararaca-0.3.11a9.dist-info}/RECORD

@@ -3,7 +3,7 @@ jararaca/__main__.py,sha256=-O3vsB5lHdqNFjUtoELDF81IYFtR-DSiiFMzRaiSsv4,67
 jararaca/broker_backend/__init__.py,sha256=GzEIuHR1xzgCJD4FE3harNjoaYzxHMHoEL0_clUaC-k,3528
 jararaca/broker_backend/mapper.py,sha256=vTsi7sWpNvlga1PWPFg0rCJ5joJ0cdzykkIc2Tuvenc,696
 jararaca/broker_backend/redis_broker_backend.py,sha256=a7DHchy3NAiD71Ix8SwmQOUnniu7uup-Woa4ON_4J7I,5786
-jararaca/cli.py,sha256=YdIot8zcnRoW5_Qm4rn0Ow1kfjlgPCJwndG63vDzpaA,24572
+jararaca/cli.py,sha256=8B3WvRqZAGn2ErfgvldiumaeFHO74jU01GCF1VdnWr4,21950
 jararaca/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jararaca/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jararaca/core/providers.py,sha256=wktH84FK7c1s2wNq-fudf1uMfi3CQBR0neU2czJ_L0U,434
@@ -14,15 +14,15 @@ jararaca/lifecycle.py,sha256=qKlzLQQioS8QkxNJ_FC_5WbmT77cNbc_S7OcQeOoHkI,1895
 jararaca/messagebus/__init__.py,sha256=5jAqPqdcEMYBfQyfZDWPnplYdrfMyJLMcacf3qLyUhk,56
 jararaca/messagebus/bus_message_controller.py,sha256=Xd_qwnX5jUvgBTCarHR36fvtol9lPTsYp2IIGKyQQaE,1487
 jararaca/messagebus/consumers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jararaca/messagebus/decorators.py,sha256=pNefkrfTFuhRQQzmfL7MSKcVa55b0TUHqEtKPunxoAA,5847
+jararaca/messagebus/decorators.py,sha256=jkVkN-NNIefKjMGq0JcAjEprdoo04jZtHm7YMsja1to,5896
 jararaca/messagebus/interceptors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jararaca/messagebus/interceptors/aiopika_publisher_interceptor.py,sha256=_DEHwIH9LYsA26Hu1mo9oHzLZuATgjilU9E3o-ecDjs,6520
 jararaca/messagebus/interceptors/publisher_interceptor.py,sha256=ojy1bRhqMgrkQljcGGS8cd8-8pUjL8ZHjIUkdmaAnNM,1325
 jararaca/messagebus/message.py,sha256=U6cyd2XknX8mtm0333slz5fanky2PFLWCmokAO56vvU,819
 jararaca/messagebus/publisher.py,sha256=JTkxdKbvxvDWT8nK8PVEyyX061vYYbKQMxRHXrZtcEY,2173
-jararaca/messagebus/worker.py,sha256=IMMI5NCVYnXmETFgFymkFJj-dGUHM0WQUZXNItfW5cY,14018
-jararaca/messagebus/worker_v2.py,sha256=M8bN_9L1mo_uzJDut90WgKv3okwrzfKg4jD6jUFcBvA,20904
-jararaca/microservice.py,sha256=jTGzkoJcco1nXwZZDeHRwEHAu_Hpr5gxA_d51P49C2c,7915
+jararaca/messagebus/worker.py,sha256=Rc6-KG0Uxo89Q5JpOC-FMuGEwg9sIElUMVh2WLINGOc,14862
+jararaca/messagebus/worker_v2.py,sha256=18oD-6Ip_rOa90p53EcEPlvkXho3SWrC40l4OVSIsE4,22356
+jararaca/microservice.py,sha256=rRIimfeP2-wf289PKoUbk9wrSdA0ga_qWz5JNgQ5IE0,9667
 jararaca/observability/decorators.py,sha256=MOIr2PttPYYvRwEdfQZEwD5RxKHOTv8UEy9n1YQVoKw,2281
 jararaca/observability/interceptor.py,sha256=U4ZLM0f8j6Q7gMUKKnA85bnvD-Qa0ii79Qa_X8KsXAQ,1498
 jararaca/observability/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -58,18 +58,18 @@ jararaca/rpc/http/backends/otel.py,sha256=Uc6CjHSCZ5hvnK1fNFv3ota5xzUFnvIl1JOpG3
 jararaca/rpc/http/decorators.py,sha256=oUSzgMGI8w6SoKiz3GltDbd3BWAuyY60F23cdRRNeiw,11897
 jararaca/rpc/http/httpx.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jararaca/scheduler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jararaca/scheduler/decorators.py,sha256=ZvOMzQT1ypU7xRLuhu6YARek1YFQ69kje2NdGjlppY8,4325
-jararaca/scheduler/scheduler.py,sha256=PjX40P09tAeD77Z0f4U1XkW782aDk_1GlUXoLaEWXe8,5446
-jararaca/scheduler/scheduler_v2.py,sha256=x1snb4eaCmBqjAkAvAcqSNvgpQolXqf_zTfKRBK3bIM,11260
+jararaca/scheduler/decorators.py,sha256=iyWFvPLCRh9c0YReQRemI2mLuaUv7r929So-xuKIWUs,4605
+jararaca/scheduler/scheduler.py,sha256=1T7qKIOiU9WD2p0aJsiUW-MOArJZyUSMwpv_14ziTQM,6188
+jararaca/scheduler/scheduler_v2.py,sha256=zrpbiYEzEn7dugbH-MhodOBVjPRzpJ9Z3iEPg8qZVB4,11839
 jararaca/scheduler/types.py,sha256=4HEQOmVIDp-BYLSzqmqSFIio1bd51WFmgFPIzPpVu04,135
 jararaca/tools/app_config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jararaca/tools/app_config/decorators.py,sha256=-ckkMZ1dswOmECdo1rFrZ15UAku--txaNXMp8fd1Ndk,941
 jararaca/tools/app_config/interceptor.py,sha256=HV8h4AxqUc_ACs5do4BSVlyxlRXzx7HqJtoVO9tfRnQ,2611
 jararaca/tools/typescript/interface_parser.py,sha256=35xbOrZDQDyTXdMrVZQ8nnFw79f28lJuLYNHAspIqi8,30492
 jararaca/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jararaca/utils/rabbitmq_utils.py,sha256=MFbXpScRxUyeZMfhB-o22PTFFF_SrU8Y09APaGYUtbo,2623
-jararaca-0.3.11a7.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-jararaca-0.3.11a7.dist-info/METADATA,sha256=tNPHTsrnA-qsO31QGN1GzH5z8522sLFU2vlCrMDAe9o,4997
-jararaca-0.3.11a7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-jararaca-0.3.11a7.dist-info/entry_points.txt,sha256=WIh3aIvz8LwUJZIDfs4EeH3VoFyCGEk7cWJurW38q0I,45
-jararaca-0.3.11a7.dist-info/RECORD,,
+jararaca/utils/rabbitmq_utils.py,sha256=8_HLPODHhXjiU6TqiXQtrf50EiN1IvAEUkbzrCSbjP8,12221
+jararaca-0.3.11a9.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+jararaca-0.3.11a9.dist-info/METADATA,sha256=wr4fnnWJHt1xJnxQk57UjWtaV28FjNW8Xk2wYITk65s,4997
+jararaca-0.3.11a9.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+jararaca-0.3.11a9.dist-info/entry_points.txt,sha256=WIh3aIvz8LwUJZIDfs4EeH3VoFyCGEk7cWJurW38q0I,45
+jararaca-0.3.11a9.dist-info/RECORD,,