rb-commons 0.1.24__tar.gz → 0.2.3__tar.gz

{rb-commons-0.1.24 → rb_commons-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: rb-commons
-Version: 0.1.24
+Version: 0.2.3
 Summary: Commons of project and simplified orm based on sqlalchemy.
 Home-page: https://github.com/RoboSell-organization/rb-commons
 Author: Abdulvoris
@@ -17,6 +17,16 @@ 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
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 # RB-Commons
 

rb_commons-0.2.3/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-0.1.24 → rb_commons-0.2.3}/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-0.2.3/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.2.3/rb_commons/http/base_api.py

@@ -0,0 +1,91 @@
+import json
+import requests
+from rb_commons.http.exceptions import BadRequestException, InternalException
+from requests import RequestException
+
+class BaseAPI:
+    def __init__(self, base_url: str):
+        self.BASE_URL = base_url
+
+    def _make_request(self, method: str, path: str, data: dict = None,
+                      params: dict = None, headers: dict = None, reset_base_url: bool = False, form_encoded: bool = False) -> requests.Response:
+        """
+        A general method to make HTTP requests (GET, POST, etc.)
+
+        This method abstracts the process of making HTTP requests, allowing for
+        different types of requests (GET, POST, PUT, DELETE) to be easily performed
+        by specifying the method and other parameters. This adheres to the DRY
+        (Don't Repeat Yourself) principle by centralizing request handling.
+
+        :param method: HTTP method to use for the request (e.g., 'GET', 'POST', 'PUT', 'DELETE').
+        :param path: The API endpoint path to which the request is made. This should be
+                     relative to the base URL defined in the class.
+        :param data: Optional dictionary containing the data to be sent in the body of
+                     the request (used for POST and PUT requests). Default is None.
+        :param params: Optional dictionary containing query parameters to be included
+                       in the request URL. Default is None.
+        :param headers: Optional dictionary containing any custom headers to be sent
+                        with the request. Default is None.
+        :return: Response object containing the server's response to the HTTP request.
+        :raises HttpException: Raises a custom HttpException if the request fails for
+                             any reason, including invalid HTTP methods, network issues,
+                             or JSON parsing errors.
+
+        The method constructs the full URL from the base URL and the provided path.
+        It uses a dictionary to map HTTP methods to their respective request functions
+        from the requests library. After making the request, it checks the response
+        status and attempts to parse the response as JSON.
+        """
+        url = self.BASE_URL + path
+
+        if reset_base_url:
+            url = path
+
+        try:
+            request_methods = {
+                'POST': requests.post,
+                'GET': requests.get,
+                'PUT': requests.put,
+                'DELETE': requests.delete
+            }
+
+            if method not in request_methods:
+                raise BadRequestException(f"Unsupported HTTP method: {method}")
+
+            kwargs = {
+                'params': params,
+                'headers': headers,
+                'data' if form_encoded else 'json': data
+            }
+
+            response = request_methods[method](url, **kwargs)
+
+            data = response.json()
+
+            if not (200 <= response.status_code < 300):
+                error_message = data.get("message")
+                if error_message is not None:
+                    raise BadRequestException(error_message)
+
+                raise BadRequestException(f"Unexpected error occured: {response.text}")
+
+            return response
+
+        except RequestException as e:
+            raise BadRequestException(f"Request failed: {e}")
+        except (json.JSONDecodeError, ValueError) as e:
+            raise InternalException(f"Failed to parse JSON: {e}")
+        except Exception as e:
+            raise InternalException("Something went wrong")
+
+    def _post(self, path: str, data: dict, headers: dict = None, params: dict = None, reset_base_url: bool = False, form_encoded: bool = False) -> requests.Response:
+        return self._make_request('POST', path, data=data, headers=headers, params=params, reset_base_url=reset_base_url, form_encoded=form_encoded)
+
+    def _get(self, path: str, params: dict = None, headers: dict = None, reset_base_url: bool = False, form_encoded: bool = False) -> requests.Response:
+        return self._make_request('GET', path, params=params, headers=headers, reset_base_url=reset_base_url, form_encoded=form_encoded)
+
+    def _put(self, path: str, params: dict = None, data: dict = None, headers: dict = None, reset_base_url: bool = False, form_encoded: bool = False) -> requests.Response:
+        return self._make_request('PUT', path, params=params, data=data, headers=headers, reset_base_url=reset_base_url, form_encoded=form_encoded)
+
+    def _delete(self, path: str, headers: dict = None, reset_base_url: bool = False, form_encoded: bool = False) -> requests.Response:
+        return self._make_request('DELETE', path, headers=headers, reset_base_url=reset_base_url, form_encoded=form_encoded)

{rb-commons-0.1.24 → rb_commons-0.2.3}/rb_commons.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: rb-commons
-Version: 0.1.24
+Version: 0.2.3
 Summary: Commons of project and simplified orm based on sqlalchemy.
 Home-page: https://github.com/RoboSell-organization/rb-commons
 Author: Abdulvoris
@@ -17,6 +17,16 @@ 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
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 # RB-Commons
 

{rb-commons-0.1.24 → rb_commons-0.2.3}/rb_commons.egg-info/SOURCES.txt

@@ -6,10 +6,14 @@ rb_commons.egg-info/SOURCES.txt
 rb_commons.egg-info/dependency_links.txt
 rb_commons.egg-info/requires.txt
 rb_commons.egg-info/top_level.txt
+rb_commons/broker/__init__.py
+rb_commons/broker/consumer.py
 rb_commons/configs/__init__.py
 rb_commons/configs/config.py
 rb_commons/configs/injections.py
+rb_commons/configs/rabbitmq.py
 rb_commons/http/__init__.py
+rb_commons/http/base_api.py
 rb_commons/http/exceptions.py
 rb_commons/orm/__init__.py
 rb_commons/orm/exceptions.py

{rb-commons-0.1.24 → rb_commons-0.2.3}/rb_commons.egg-info/requires.txt

@@ -5,3 +5,4 @@ PyJWT==2.10.1
 python-dotenv==1.0.1
 SQLAlchemy==2.0.36
 fastapi<0.120.0,>=0.115.6
+aio-pika==9.5.5

{rb-commons-0.1.24 → rb_commons-0.2.3}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="rb-commons",
-    version="0.1.24",
+    version="0.2.3",
     author="Abdulvoris",
     author_email="erkinovabdulvoris101@gmail.com",
     description="Commons of project and simplified orm based on sqlalchemy.",
@@ -26,6 +26,7 @@ setup(
         "PyJWT==2.10.1",
         "python-dotenv==1.0.1",
         "SQLAlchemy==2.0.36",
-        "fastapi>=0.115.6,<0.120.0"
+        "fastapi>=0.115.6,<0.120.0",
+        "aio-pika==9.5.5"
     ],
 )