feat: ✨ Be able to add your own loggers

Author: remimd

injection/__init__.pyi

@@ -2,6 +2,7 @@ from abc import abstractmethod
 from collections.abc import Callable
 from contextlib import ContextDecorator
 from enum import StrEnum
+from logging import Logger
 from types import UnionType
 from typing import (
     Any,
@@ -179,6 +180,7 @@ class Module:
         Function to unlock the module by deleting cached instances of singletons.
         """
 
+    def add_logger(self, logger: Logger) -> Self: ...
     @classmethod
     def from_name(cls, name: str) -> Self:
         """

injection/core/module.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import inspect
-import logging
 from abc import ABC, abstractmethod
 from collections import OrderedDict
 from collections.abc import (
@@ -17,6 +16,7 @@
 from enum import StrEnum
 from functools import partialmethod, singledispatchmethod, update_wrapper
 from inspect import Signature, isclass
+from logging import Logger, getLogger
 from queue import Empty, Queue
 from types import MethodType, UnionType
 from typing import (
@@ -59,8 +59,6 @@
     "PriorityStr",
 )
 
-_logger = logging.getLogger(__name__)
-
 """
 Events
 """
@@ -386,6 +384,11 @@ class Module(EventListener, Broker):
         init=False,
         repr=False,
     )
+    __loggers: list[Logger] = field(
+        default_factory=lambda: [getLogger(__name__)],
+        init=False,
+        repr=False,
+    )
 
     __instances: ClassVar[dict[str, Module]] = {}
 
@@ -585,6 +588,10 @@ def unlock(self) -> Self:
 
         return self
 
+    def add_logger(self, logger: Logger) -> Self:
+        self.__loggers.append(logger)
+        return self
+
     def add_listener(self, listener: EventListener) -> Self:
         self.__channel.add_listener(listener)
         return self
@@ -603,7 +610,7 @@ def notify(self, event: Event):
 
         with self.__channel.dispatch(event):
             yield
-            _logger.debug(event)
+            self.__send_debug(event)
 
     def __check_locking(self):
         if self.is_locked:
@@ -619,6 +626,10 @@ def __move_module(self, module: Module, priority: Priority):
                 f"`{module}` can't be found in the modules used by `{self}`."
             ) from exc
 
+    def __send_debug(self, message: object):
+        for logger in self.__loggers:
+            logger.debug(message)
+
     @classmethod
     def from_name(cls, name: str) -> Self:
         with suppress(KeyError):