qena-shared-lib 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

qena_shared_lib/exception_handlers.py

@@ -1,9 +1,10 @@
+from collections.abc import Iterable
 from typing import Any
 
 from fastapi import Request, Response, status
-from fastapi.encoders import jsonable_encoder
 from fastapi.exceptions import RequestValidationError
 from fastapi.responses import JSONResponse
+from pydantic_core import to_jsonable_python
 
 from .dependencies.http import get_service
 from .exceptions import ServiceException, Severity
@@ -93,7 +94,19 @@ def handle_service_exception(
         content["correctiveAction"] = exception.corrective_action
 
     if exception.body is not None:
-        content.update(dict(exception.body))
+        extra_body = to_jsonable_python(exception.body)
+        is_updated = False
+
+        try:
+            if isinstance(extra_body, Iterable):
+                content.update(extra_body)
+
+                is_updated = True
+        except:
+            pass
+
+        if not is_updated:
+            content["data"] = extra_body
 
     return JSONResponse(
         content=content,
@@ -117,7 +130,7 @@ def handle_request_validation_error(
             "severity": Severity.MEDIUM.name,
             "message": message,
             "code": 100,
-            "detail": jsonable_encoder(error.errors()),
+            "detail": to_jsonable_python(error.errors()),
         },
         status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
     )

qena_shared_lib/logstash/_base.py

@@ -228,7 +228,6 @@ class BaseLogstashSender(AsyncEventLoopMixin):
         self._max_log_retry = max_log_retry
         self._started = False
         self._closed = False
-        self._close_future = self.loop.create_future()
         self._log_queue: Queue[LogstashLogRecord | EndOfLogMarker] = Queue(
             log_queue_size
         )
@@ -246,6 +245,7 @@ class BaseLogstashSender(AsyncEventLoopMixin):
 
         self._started = True
         self._closed = False
+        self._close_future = self.loop.create_future()
         _, _ = await gather(
             self.loop.run_in_executor(executor=None, func=self._hook_on_start),
             self._hook_on_start_async(),

qena_shared_lib/rabbitmq/_base.py

@@ -381,7 +381,7 @@ class RabbitMqManager(AsyncEventLoopMixin):
         )
 
         if self._connected:
-            self.loop.create_task(self._channel_pool.dain()).add_done_callback(
+            self.loop.create_task(self._channel_pool.drain()).add_done_callback(
                 self._channel_pool_drained
             )
 

qena_shared_lib/rabbitmq/_listener.py

@@ -21,11 +21,10 @@ from pydantic_core import from_json, to_json
 from ..dependencies.miscellaneous import validate_annotation
 from ..logging import LoggerProvider
 from ..logstash import BaseLogstashSender
-from ..utils import AsyncEventLoopMixin
+from ..utils import AsyncEventLoopMixin, TypeAdapterCache
 from ._channel import BaseChannel
 from ._exceptions import RabbitMQException
 from ._pool import ChannelPool
-from ._utils import TypeAdapterCache
 
 __all__ = [
     "BackoffRetryDelay",

qena_shared_lib/rabbitmq/_pool.py

@@ -28,7 +28,7 @@ class ChannelPool(AsyncEventLoopMixin):
             for _ in range(self._initial_pool_size):
                 await self._add_new_channel()
 
-    async def dain(self):
+    async def drain(self):
         async with self._pool_lock:
             for channel_base in self._pool.values():
                 with channel_base as channel:

qena_shared_lib/rabbitmq/_rpc_client.py

@@ -13,10 +13,9 @@ from prometheus_client import Counter, Summary
 from pydantic_core import from_json, to_json
 
 from ..logging import LoggerProvider
-from ..utils import AsyncEventLoopMixin
+from ..utils import AsyncEventLoopMixin, TypeAdapterCache
 from ._exceptions import RabbitMQException
 from ._pool import ChannelPool
-from ._utils import TypeAdapterCache
 
 __all__ = ["RpcClient"]
 

qena_shared_lib/utils.py

@@ -1,6 +1,8 @@
 from asyncio import AbstractEventLoop, get_running_loop
 
-__all__ = ["AsyncEventLoopMixin"]
+from pydantic import TypeAdapter
+
+__all__ = ["AsyncEventLoopMixin", "TypeAdapterCache"]
 
 ABSTRACT_EVENT_LOOP_ATTRIBUTE = "__abstract_event_loop__"
 
@@ -26,3 +28,18 @@ class AsyncEventLoopMixin:
             setattr(self, ABSTRACT_EVENT_LOOP_ATTRIBUTE, current_running_loop)
 
         return getattr(self, ABSTRACT_EVENT_LOOP_ATTRIBUTE)
+
+
+class TypeAdapterCache:
+    _cache = {}
+
+    @classmethod
+    def cache_annotation(cls, annotation: type):
+        if annotation not in cls._cache:
+            cls._cache[annotation] = TypeAdapter(annotation)
+
+    @classmethod
+    def get_type_adapter(cls, annotation: type) -> TypeAdapter:
+        cls.cache_annotation(annotation)
+
+        return cls._cache[annotation]

{qena_shared_lib-0.1.5.dist-info → qena_shared_lib-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qena-shared-lib
-Version: 0.1.5
+Version: 0.1.7
 Summary: A shared tools for other services
 Requires-Python: >=3.10
 Requires-Dist: cronsim~=2.0

{qena_shared_lib-0.1.5.dist-info → qena_shared_lib-0.1.7.dist-info}/RECORD

@@ -1,31 +1,30 @@
 qena_shared_lib/__init__.py,sha256=WokKEFaMNow6h2ZY_fyB-tiYnic-Ed6K6kyzKgg3Rlw,371
 qena_shared_lib/application.py,sha256=2bpoEHW9FXRjNgBUuij2djfTLbhk01zymqp6kwasRq0,5470
 qena_shared_lib/background.py,sha256=upWeJ747-4S4xsBOR-P2_oNar1YiaxCpnLhEDvnArmQ,3097
-qena_shared_lib/exception_handlers.py,sha256=od_4g2Ro0HTeiIip_Saq6OCW0FMNjwLuaGevVWRPwFo,4824
+qena_shared_lib/exception_handlers.py,sha256=GclZUv8_QuJtbVSkzux8hxNbZNG7DWI2u0oGXgaEMD4,5143
 qena_shared_lib/exceptions.py,sha256=cNeksC8ZavKgoqKLBik1NR-lCcdLZzITHMXww5deY5Y,6789
 qena_shared_lib/http.py,sha256=90bmDfs522KxKBj2o0vO-MgmDqa3EZOfaNUFp1DuUdQ,23864
 qena_shared_lib/logging.py,sha256=JL6bAmkK1BJA84rZNpvDEmrec3dogQRpSug4fj1Omkw,1618
 qena_shared_lib/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qena_shared_lib/scheduler.py,sha256=qxM3zdKJnAKUWw8i_I2M5o3uL_fEOTaBzYUBL-HvUzg,11874
 qena_shared_lib/security.py,sha256=n872UE3eQfezo1lkHwz5hcAngO0VekA_qDQ9LsZ8JZ8,6072
-qena_shared_lib/utils.py,sha256=bvPnjaFx7npVGSUO6IKdpiAHfUECsGRCeDW8UOjVckU,845
+qena_shared_lib/utils.py,sha256=y60AKFknK-vus4i-nesgL00BWuAXJyd8Qx79Gi7lRq4,1272
 qena_shared_lib/dependencies/__init__.py,sha256=W12RgJbhqZ9GiSV1nLlHmpwPzvQv8t7f4JEoazM_WYg,350
 qena_shared_lib/dependencies/http.py,sha256=3KopQjq05qbYROMPZjji2Zz7qQLU6_2u3qKisLrjLks,1469
 qena_shared_lib/dependencies/miscellaneous.py,sha256=iGwAjatXb_JVSF13n1vdTRAgSKv19VtHo9ZbjjbkIco,753
 qena_shared_lib/logstash/__init__.py,sha256=KlWFXqwPGwKM3yWnz0lTmeZymkobxFPPNeOgyfdGu4g,315
-qena_shared_lib/logstash/_base.py,sha256=b_wWbdIpl9aIqGdnNI-oYuZmiiUhx5rWXIbZzURyOnU,15722
+qena_shared_lib/logstash/_base.py,sha256=sX0FLr9R4mkYL3KJO6ADlZaqjn3KbGOLyVTfK_ZTbSo,15722
 qena_shared_lib/logstash/_http_sender.py,sha256=Sq3bJOXuOafqloROGblnmyUd0RuU3zF0QLpQDYjiy-g,1765
 qena_shared_lib/logstash/_tcp_sender.py,sha256=ncvc3wYxOJPkB7c_0W8vboaxMRJ_1UYGdA8LN2NXUpo,2306
 qena_shared_lib/rabbitmq/__init__.py,sha256=WKAcKpRXXX9kqfHVEkyX6p2yIWzB07TMsBBWbV8nrKA,1209
-qena_shared_lib/rabbitmq/_base.py,sha256=rOUpYK3FJu3U5rU4vF--31c_Ycpnt4FL9qGkZNGLnYY,23068
+qena_shared_lib/rabbitmq/_base.py,sha256=IKWwi7B82dhkIPwvY12l4ko1cjdmedklBxYkUQVQVL0,23069
 qena_shared_lib/rabbitmq/_channel.py,sha256=R0xzZvLkeE4YWsyLhx8bPDramsTFDZIcCgfsDyFmSB4,5429
 qena_shared_lib/rabbitmq/_exception_handlers.py,sha256=7-WguEUqUSIdbvCIWn10f2CRaWPCM3MDZ6sxSATCCYA,4762
 qena_shared_lib/rabbitmq/_exceptions.py,sha256=Mg56Uk4eBEwTnWC91_WPS3_8d-yaqNkbb_32IpX_R-c,1208
-qena_shared_lib/rabbitmq/_listener.py,sha256=1c6Qrz87hL10kaGobTZhI48v3ou88A5u371QdkyZIXA,44153
-qena_shared_lib/rabbitmq/_pool.py,sha256=1_ftW0D3sPHI6v5c72oA03s05up4-1QqZ7d8icnZeNs,1951
+qena_shared_lib/rabbitmq/_listener.py,sha256=eYrlaptPoc72y7UMRyo_Lnn_OE0HGVlBJIh5UEptnww,44134
+qena_shared_lib/rabbitmq/_pool.py,sha256=MGubA-nYxxjgsZRF1yK4UBkzUy65t-suh0W1jZDk5q8,1952
 qena_shared_lib/rabbitmq/_publisher.py,sha256=48JdpvNzFaw3Bw5LhunFxcjm3qNWz3AXKyXtcpzcohQ,2380
-qena_shared_lib/rabbitmq/_rpc_client.py,sha256=E-J5PIfA1T3j1y9KoxjNTzSLdK95MbaePulNxiqGb-M,8724
-qena_shared_lib/rabbitmq/_utils.py,sha256=5WZjYm4MdDrjL1dsrZuT8eKKiWfUef29_FJ262-hfSM,438
-qena_shared_lib-0.1.5.dist-info/METADATA,sha256=77Kw8La2gcdfVt11wk7iHHJwTKGSZs8XLIXCd_m8I8Q,8520
-qena_shared_lib-0.1.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-qena_shared_lib-0.1.5.dist-info/RECORD,,
+qena_shared_lib/rabbitmq/_rpc_client.py,sha256=5hA8IWvydRufYdv0zViIsi6_W2tWkHcmZotgCV2yjfc,8705
+qena_shared_lib-0.1.7.dist-info/METADATA,sha256=f6AHqMQrm79vJ1H0n_O5t14CnKvFec8zq1xJZi54H-Q,8520
+qena_shared_lib-0.1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+qena_shared_lib-0.1.7.dist-info/RECORD,,

qena_shared_lib/rabbitmq/_utils.py

@@ -1,18 +0,0 @@
-from pydantic import TypeAdapter
-
-__all__ = ["TypeAdapterCache"]
-
-
-class TypeAdapterCache:
-    _cache = {}
-
-    @classmethod
-    def cache_annotation(cls, annotation: type):
-        if annotation not in cls._cache:
-            cls._cache[annotation] = TypeAdapter(annotation)
-
-    @classmethod
-    def get_type_adapter(cls, annotation: type) -> TypeAdapter:
-        cls.cache_annotation(annotation)
-
-        return cls._cache[annotation]