fix: Set the logger class in botcore.utils.logging.py to the CustomLogger class

Fixes an issue when using trace logging on loggers that were instantiated before the class was set.

1 files changed, 6 insertions, 0 deletions

diff --git a/botcore/utils/logging.py b/botcore/utils/logging.py
index 71ce4e66..1f1c8bac 100644
--- a/botcore/utils/logging.py
+++ b/botcore/utils/logging.py
@@ -43,3 +43,9 @@ def get_logger(name: typing.Optional[str] = None) -> CustomLogger:
         An instance of the :obj:`CustomLogger` class.
     """
     return typing.cast(CustomLogger, logging.getLogger(name))
+
+
+# Setup trace level logging so that we can use it within botcore.
+logging.TRACE = TRACE_LEVEL
+logging.setLoggerClass(CustomLogger)
+logging.addLevelName(TRACE_LEVEL, "TRACE")