qena-shared-lib 0.1.0__tar.gz → 0.1.2__tar.gz

{qena_shared_lib-0.1.0 → qena_shared_lib-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qena-shared-lib
-Version: 0.1.0
+Version: 0.1.2
 Summary: A shared tools for other services
 Requires-Python: >=3.10
 Requires-Dist: cronsim~=2.0
@@ -224,13 +224,11 @@ def main() -> FastAPI:
     ...
 
     rabbitmq = RabbitMqManager(
-        listeners=[
-            UserConsumer
-        ],
         logstash=logstash,
         container=builder.container,
     )
 
+    rabbitmq.include_listener(UserConsumer)
     builder.add_singleton(
         service=RabbitMqManager,
         instance=rabbitmq,
@@ -314,13 +312,11 @@ def main() -> FastAPI:
     ...
 
     schedule_manager = ScheduleManager(
-        schedulers=[
-            TaskScheduler
-        ],
         logstash=logstash,
         container=builder.container
     )
 
+    schedule_manager.include_scheduler(TaskScheduler)
     builder.with_singleton(
         service=ScheduleManager,
         instance=schedule_manager,

{qena_shared_lib-0.1.0 → qena_shared_lib-0.1.2}/README.md

@@ -207,13 +207,11 @@ def main() -> FastAPI:
     ...
 
     rabbitmq = RabbitMqManager(
-        listeners=[
-            UserConsumer
-        ],
         logstash=logstash,
         container=builder.container,
     )
 
+    rabbitmq.include_listener(UserConsumer)
     builder.add_singleton(
         service=RabbitMqManager,
         instance=rabbitmq,
@@ -297,13 +295,11 @@ def main() -> FastAPI:
     ...
 
     schedule_manager = ScheduleManager(
-        schedulers=[
-            TaskScheduler
-        ],
         logstash=logstash,
         container=builder.container
     )
 
+    schedule_manager.include_scheduler(TaskScheduler)
     builder.with_singleton(
         service=ScheduleManager,
         instance=schedule_manager,

{qena_shared_lib-0.1.0 → qena_shared_lib-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "qena-shared-lib"
-version = "0.1.0"
+version = "0.1.2"
 description = "A shared tools for other services"
 readme = "README.md"
 requires-python = ">=3.10"

{qena_shared_lib-0.1.0 → qena_shared_lib-0.1.2}/src/qena_shared_lib/rabbitmq/_base.py

@@ -94,7 +94,6 @@ class RabbitMqManager(AsyncEventLoopMixin):
 
     def __init__(
         self,
-        listeners: list[Listener | type[ListenerBase]],
         logstash: BaseLogstashSender,
         parameters: Parameters | str | None = None,
         reconnect_delay: float = 5.0,
@@ -102,24 +101,7 @@ class RabbitMqManager(AsyncEventLoopMixin):
         listener_global_retry_policy: RetryPolicy | None = None,
         container: Container | None = None,
     ):
-        for index, listener in enumerate(listeners):
-            if not isinstance(listener, Listener) and (
-                not isinstance(listener, type)
-                or not issubclass(listener, ListenerBase)
-            ):
-                raise TypeError(
-                    f"listener {index} is {type(listener)}, expected instance of type or subclass of `Listener` or `type[ListenerBase]`"
-                )
-
-        self._listener_classes = [
-            listener_class
-            for listener_class in listeners
-            if isinstance(listener_class, type)
-            and issubclass(listener_class, ListenerBase)
-        ]
-        self._listeners = [
-            listener for listener in listeners if isinstance(listener, Listener)
-        ]
+        self._listeners: list[Listener] = []
 
         if isinstance(parameters, str):
             self._parameters = URLParameters(parameters)
@@ -139,8 +121,6 @@ class RabbitMqManager(AsyncEventLoopMixin):
             type[Exception], ExceptionHandlerContainer
         ] = {}
 
-        self._register_listener_classes()
-
         self.exception_handler(RabbitMQException)(handle_rabbitmq_exception)
         self.exception_handler(ValidationError)(handle_validation_error)
         self.exception_handler(ServiceException)(handle_microservice_exception)
@@ -150,26 +130,6 @@ class RabbitMqManager(AsyncEventLoopMixin):
         self._logstash = logstash
         self._logger = LoggerProvider.default().get_logger("rabbitmq")
 
-    def _register_listener_classes(self):
-        for listener_class in self._listener_classes:
-            inner_listener = getattr(listener_class, LISTENER_ATTRIBUTE, None)
-
-            if inner_listener is None:
-                raise AttributeError(
-                    "listener is possibly not with `Consumer` or `RpcWorker`"
-                )
-
-            if not isinstance(inner_listener, Listener):
-                raise TypeError(
-                    f"listener class {type(listener_class)} is not a type `Listener`, posibilly not decorated with `Consumer` or `RpcWorker`"
-                )
-
-            self._container.register(
-                service=ListenerBase,
-                factory=listener_class,
-                scope=Scope.singleton,
-            )
-
     @property
     def container(self) -> Container:
         return self._container
@@ -239,6 +199,40 @@ class RabbitMqManager(AsyncEventLoopMixin):
 
         return True, None
 
+    def include_listener(self, listener: Listener | type[ListenerBase]):
+        if isinstance(listener, Listener):
+            self._listeners.append(listener)
+
+            return
+
+        if isinstance(listener, type) and issubclass(listener, ListenerBase):
+            self._register_listener_classes(listener)
+
+            return
+
+        raise TypeError(
+            f"listener is {type(listener)}, expected instance of type or subclass of `Listener` or `type[ListenerBase]`"
+        )
+
+    def _register_listener_classes(self, listener_class: type):
+        inner_listener = getattr(listener_class, LISTENER_ATTRIBUTE, None)
+
+        if inner_listener is None:
+            raise AttributeError(
+                "listener is possibly not with `Consumer` or `RpcWorker`"
+            )
+
+        if not isinstance(inner_listener, Listener):
+            raise TypeError(
+                f"listener class {type(listener_class)} is not a type `Listener`, posibilly not decorated with `Consumer` or `RpcWorker`"
+            )
+
+        self._container.register(
+            service=ListenerBase,
+            factory=listener_class,
+            scope=Scope.singleton,
+        )
+
     def include_service(
         self,
         rabbit_mq_service: AbstractRabbitMQService
@@ -407,9 +401,7 @@ class RabbitMqManager(AsyncEventLoopMixin):
         exception = task.exception()
 
         if exception is not None:
-            if not self._disconnected and not isinstance(
-                exception, (ConnectionClosedByClient, ChannelClosedByClient)
-            ):
+            if self._can_reconnect(exception):
                 self._logstash.error(
                     message="couldn't drain the channel pool",
                     exception=exception,
@@ -462,13 +454,7 @@ class RabbitMqManager(AsyncEventLoopMixin):
         if exception is not None:
             if not self._connected and not self._connected_future.done():
                 self._connected_future.set_exception(exception)
-            elif (
-                self._connected
-                and not self._disconnected
-                and not isinstance(
-                    exception, (ConnectionClosedByClient, ChannelClosedByClient)
-                )
-            ):
+            elif self._can_reconnect(exception):
                 self._logstash.error(
                     message="couldn't fill the channel pool",
                     exception=exception,
@@ -538,13 +524,7 @@ class RabbitMqManager(AsyncEventLoopMixin):
         if exception is not None:
             if not self._connected and not self._connected_future.done():
                 self._connected_future.set_exception(exception)
-            elif (
-                self._connected
-                and not self._disconnected
-                and not isinstance(
-                    exception, (ConnectionClosedByClient, ChannelClosedByClient)
-                )
-            ):
+            elif self._can_reconnect(exception):
                 self._logstash.error(
                     message="couldn't configure and initialize all listeners and services",
                     exception=exception,
@@ -671,13 +651,7 @@ class RabbitMqManager(AsyncEventLoopMixin):
 
             return
 
-        if (
-            self._connected
-            and not self._disconnected
-            and not isinstance(
-                exception, (ConnectionClosedByClient, ChannelClosedByClient)
-            )
-        ):
+        if self._can_reconnect(exception):
             self._logstash.error(
                 message="connection to rabbitmq closed unexpectedly, attempting to reconnect",
                 exception=exception,
@@ -696,13 +670,7 @@ class RabbitMqManager(AsyncEventLoopMixin):
         try:
             connected_future = self.connect()
         except Exception as e:
-            if (
-                self._connected
-                and not self._disconnected
-                and not isinstance(
-                    e, (ConnectionClosedByClient, ChannelClosedByClient)
-                )
-            ):
+            if self._can_reconnect(e):
                 if not self._connected_future.done():
                     self._connected_future.set_result(None)
 
@@ -727,15 +695,18 @@ class RabbitMqManager(AsyncEventLoopMixin):
         if exception is None:
             return
 
-        if (
-            self._connected
-            and not self._disconnected
-            and not isinstance(
-                exception, (ConnectionClosedByClient, ChannelClosedByClient)
-            )
-        ):
+        if self._can_reconnect(exception):
             self._logstash.error(
                 message="couldn't reconnect to rabbitmq, attempting to reconnect",
                 exception=exception,
             )
             self._reconnect()
+
+    def _can_reconnect(self, exception: BaseException) -> bool:
+        return (
+            self._connected
+            and not self._disconnected
+            and not isinstance(
+                exception, (ConnectionClosedByClient, ChannelClosedByClient)
+            )
+        )

{qena_shared_lib-0.1.0 → qena_shared_lib-0.1.2}/src/qena_shared_lib/scheduler.py

@@ -181,33 +181,34 @@ class ScheduleManager(AsyncEventLoopMixin):
 
     def __init__(
         self,
-        schedulers: list[Scheduler | type[SchedulerBase]],
         logstash: BaseLogstashSender,
         container: Container | None = None,
     ):
         self._container = container or Container()
         self._logstash = logstash
         self._scheduled_tasks: list[ScheduledTask] = []
-
-        for index, scheduler in enumerate(schedulers):
-            if isinstance(scheduler, Scheduler):
-                self._scheduled_tasks.extend(scheduler.scheduled_tasks)
-            elif isinstance(scheduler, type) and issubclass(
-                scheduler, SchedulerBase
-            ):
-                self._container.register(
-                    service=SchedulerBase,
-                    factory=scheduler,
-                    scope=Scope.singleton,
-                )
-            else:
-                raise TypeError(
-                    f"scheduler {index} is {type(scheduler)}, expected instance of type or subclass of `Scheduler` or `type[SchedulerBase]`"
-                )
-
         self._next_run_in = None
         self._logger = LoggerProvider.default().get_logger("schedule_manager")
 
+    def include_scheduler(self, scheduler: Scheduler | type[SchedulerBase]):
+        if isinstance(scheduler, Scheduler):
+            self._scheduled_tasks.extend(scheduler.scheduled_tasks)
+
+            return
+
+        if isinstance(scheduler, type) and issubclass(scheduler, SchedulerBase):
+            self._container.register(
+                service=SchedulerBase,
+                factory=scheduler,
+                scope=Scope.singleton,
+            )
+
+            return
+
+        raise TypeError(
+            f"scheduler is {type(scheduler)}, expected instance of type or subclass of `Scheduler` or `type[SchedulerBase]`"
+        )
+
     @property
     def container(self) -> Container:
         return self._container
@@ -230,7 +231,7 @@ class ScheduleManager(AsyncEventLoopMixin):
             getattr(self, "_scheduler_task", None) is not None
             and not self._scheduler_task.done()
         ):
-            return RuntimeError("scheduler already running")
+            raise RuntimeError("scheduler already running")
 
         self.use_schedulers()
         self._logger.info(

{qena_shared_lib-0.1.0 → qena_shared_lib-0.1.2}/src/qena_shared_lib/security.py

@@ -26,14 +26,14 @@ __all__ = [
 ]
 
 
-MESSAGE = "you are not authorized to access request resouce"
+MESSAGE = "you are not authorized to access requested resouce"
 RESPONSE_CODE = int(environ.get("UNAUTHORIZED_RESPONSE_CODE") or 0)
 
 
 class PasswordHasher(AsyncEventLoopMixin):
-    def __init__(self):
+    def __init__(self, schemes=None):
         self._crypt_context = CryptContext(
-            schemes=["bcrypt"], deprecated="auto"
+            schemes=schemes or ["bcrypt"], deprecated="auto"
         )
 
     def hash(self, password: str) -> Future[str]: