rb-commons 0.1.24__py3-none-any.whl → 0.2.2__py3-none-any.whl

rb_commons/broker/consumer.py

@@ -0,0 +1,29 @@
+from aio_pika import ExchangeType, IncomingMessage
+from rb_commons.configs.rabbitmq import RabbitMQConnection
+
+
+class BaseRabbitMQConsumer:
+    def __init__(self, exchange_name: str, queue_name: str, routing_key: str):
+        self.exchange_name = exchange_name
+        self.queue_name = queue_name
+        self.routing_key = routing_key
+
+    async def setup(self, channel):
+        exchange = await channel.declare_exchange(self.exchange_name, ExchangeType.DIRECT, durable=True)
+        queue = await channel.declare_queue(self.queue_name, durable=True)
+        await queue.bind(exchange, routing_key=self.routing_key)
+        return queue
+
+    async def consume(self):
+        async with RabbitMQConnection.get_channel() as channel:
+            queue = await self.setup(channel)
+            async with queue.iterator() as queue_iter:
+                async for message in queue_iter:
+                    await self.process_message(message)
+
+    async def process_message(self, message: IncomingMessage):
+        async with message.process():
+            await self.handle_message(message.body.decode())
+
+    async def handle_message(self, body: str):
+        raise NotImplementedError("This method should be overridden by subclasses.")

rb_commons/configs/config.py

@@ -20,6 +20,10 @@ class CommonConfigs(BaseSettings):
     POSTGRES_DB: Optional[str] = None
     DB_ALEMBIC_URL: Optional[str] = None
 
+    # Broker
+
+    RABBITMQ_URL: Optional[str] = None
+
     DIGITALOCEAN_STORAGE_BUCKET_NAME: Optional[str] = None
     DIGITALOCEAN_S3_ENDPOINT_URL: Optional[str] = None
 

rb_commons/configs/rabbitmq.py

@@ -0,0 +1,26 @@
+from contextlib import asynccontextmanager
+from aio_pika import connect_robust, Message, ExchangeType
+from rb_commons.configs.config import configs
+
+class RabbitMQConnection:
+    _connection = None
+    _channel = None
+
+    @classmethod
+    @asynccontextmanager
+    async def get_channel(cls):
+        if not cls._connection:
+            cls._connection = await connect_robust(configs.RABBITMQ_URL)
+            cls._channel = await cls._connection.channel()
+
+        try:
+            yield cls._channel
+        finally:
+            pass
+
+    @classmethod
+    async def close(cls):
+        if cls._connection:
+            await cls._connection.close()
+            cls._connection = None
+            cls._channel = None

{rb_commons-0.1.24.dist-info → rb_commons-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rb-commons
-Version: 0.1.24
+Version: 0.2.2
 Summary: Commons of project and simplified orm based on sqlalchemy.
 Home-page: https://github.com/RoboSell-organization/rb-commons
 Author: Abdulvoris
@@ -17,6 +17,7 @@ Requires-Dist: PyJWT==2.10.1
 Requires-Dist: python-dotenv==1.0.1
 Requires-Dist: SQLAlchemy==2.0.36
 Requires-Dist: fastapi<0.120.0,>=0.115.6
+Requires-Dist: aio-pika==9.5.5
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{rb_commons-0.1.24.dist-info → rb_commons-0.2.2.dist-info}/RECORD

@@ -1,7 +1,10 @@
 rb_commons/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rb_commons/broker/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rb_commons/broker/consumer.py,sha256=84yLq8kJIGNYAH9_ySSTDpOMNL54NSs3cA8lNG8CTGY,1264
 rb_commons/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rb_commons/configs/config.py,sha256=dlsf1GDT81B_gOJkMHnnU5uHiCHiTIlvXNIdTu52E5k,1559
+rb_commons/configs/config.py,sha256=zla62zy3UfjE2YpvSDwsfhP2Qsm0BbGsycngfyiVvE4,1617
 rb_commons/configs/injections.py,sha256=6B1EOgIGnkWv3UrFaV9PRgG0-CJAbLu1UZ3kq-SjPVU,273
+rb_commons/configs/rabbitmq.py,sha256=vUqa_PcZkPp1lX0B6HLSAXVMUcPR2_8-8cN-C7xFxRI,741
 rb_commons/http/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rb_commons/http/exceptions.py,sha256=EGRMr1cRgiJ9Q2tkfANbf0c6-zzXf1CD6J3cmCaT_FA,1885
 rb_commons/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -13,7 +16,7 @@ rb_commons/schemes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 rb_commons/schemes/jwt.py,sha256=F66JJDhholuOPPzlKeoC6f1TL4gXg4oRUrV5yheNpyo,1675
 rb_commons/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rb_commons/utils/media.py,sha256=KNY_9SdRa3Rp7d3B1tZaXkhmzVa65RcS62BYwZP1bVM,332
-rb_commons-0.1.24.dist-info/METADATA,sha256=4LV30IZhpybCS8S5KXALPnQ7YqJVyIcjVKlPDkGh-Q8,6539
-rb_commons-0.1.24.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-rb_commons-0.1.24.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
-rb_commons-0.1.24.dist-info/RECORD,,
+rb_commons-0.2.2.dist-info/METADATA,sha256=_WLAr4eS8xznDVugZa1-aRQz4eKYo4_2Y12t8P7OH8c,6570
+rb_commons-0.2.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+rb_commons-0.2.2.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
+rb_commons-0.2.2.dist-info/RECORD,,