chore(typing): add setLevel and addHandler to Logger for mypy/pyright (aws-powertools#2388)

heitorlessa · web-flow · commit 09f6d5e662fc · 2023-06-06T10:15:32.000+01:00
diff --git a/aws_lambda_powertools/logging/logger.py b/aws_lambda_powertools/logging/logger.py
@@ -287,9 +287,9 @@ def _init_logger(self, formatter_options: Optional[Dict] = None, log_level: Unio
         if self.child or is_logger_preconfigured:
             return
 
-        self._logger.setLevel(self._determine_log_level(log_level))
+        self.setLevel(self._determine_log_level(log_level))
         self._configure_sampling()
-        self._logger.addHandler(self.logger_handler)
+        self.addHandler(self.logger_handler)
         self.structure_logs(formatter_options=formatter_options, **kwargs)
 
         # Maintenance: We can drop this upon Py3.7 EOL. It's a backport for "location" key to work
@@ -633,6 +633,12 @@ def get_correlation_id(self) -> Optional[str]:
             return self.registered_formatter.log_format.get("correlation_id")
         return None
 
+    def setLevel(self, level: Union[str, int]) -> None:
+        return self._logger.setLevel(level)
+
+    def addHandler(self, handler: logging.Handler) -> None:
+        return self._logger.addHandler(handler)
+
     @property
     def registered_handler(self) -> logging.Handler:
         """Convenience property to access the first logger handler"""
