chore: Made the base.Logger class inherit from logging.Logger

This removes the need to dynamically dispatch all `Logger` methods and attributes via `__getattr__`.
This change improves the static type checking and the IDE experience.

PiperOrigin-RevId: 566442238

Author: Ark-kun

google/cloud/aiplatform/base.py

@@ -57,7 +57,7 @@
 _DEFAULT_RETRY = retry.Retry()
 
 
-class Logger:
+class VertexLogger(logging.getLoggerClass()):
     """Logging wrapper class with high level helper methods."""
 
     def __init__(self, name: str):
@@ -66,12 +66,12 @@ def __init__(self, name: str):
         Args:
             name (str): Name to associate with logger.
         """
-        self._logger = logging.getLogger(name)
-        self._logger.setLevel(logging.INFO)
+        super().__init__(name)
+        self.setLevel(logging.INFO)
 
         # To avoid writing duplicate logs, skip adding the new handler if
         # StreamHandler already exists in logger hierarchy.
-        logger = self._logger
+        logger = self
         while logger:
             for handler in logger.handlers:
                 if isinstance(handler, logging.StreamHandler):
@@ -81,7 +81,7 @@ def __init__(self, name: str):
         handler = logging.StreamHandler(sys.stdout)
         handler.setLevel(logging.INFO)
 
-        self._logger.addHandler(handler)
+        self.addHandler(handler)
 
     def log_create_with_lro(
         self,
@@ -96,12 +96,10 @@ def log_create_with_lro(
             lro (operation.Operation):
                 Optional. Backing LRO for creation.
         """
-        self._logger.info(f"Creating {cls.__name__}")
+        self.info(f"Creating {cls.__name__}")
 
         if lro:
-            self._logger.info(
-                f"Create {cls.__name__} backing LRO: {lro.operation.name}"
-            )
+            self.info(f"Create {cls.__name__} backing LRO: {lro.operation.name}")
 
     def log_create_complete(
         self,
@@ -120,11 +118,9 @@ def log_create_complete(
                 Vertex AI Resource proto.Message
             variable_name (str): Name of variable to use for code snippet
         """
-        self._logger.info(f"{cls.__name__} created. Resource name: {resource.name}")
-        self._logger.info(f"To use this {cls.__name__} in another session:")
-        self._logger.info(
-            f"{variable_name} = aiplatform.{cls.__name__}('{resource.name}')"
-        )
+        self.info(f"{cls.__name__} created. Resource name: {resource.name}")
+        self.info(f"To use this {cls.__name__} in another session:")
+        self.info(f"{variable_name} = aiplatform.{cls.__name__}('{resource.name}')")
 
     def log_create_complete_with_getter(
         self,
@@ -143,11 +139,9 @@ def log_create_complete_with_getter(
                 Vertex AI Resource proto.Message
             variable_name (str): Name of variable to use for code snippet
         """
-        self._logger.info(f"{cls.__name__} created. Resource name: {resource.name}")
-        self._logger.info(f"To use this {cls.__name__} in another session:")
-        self._logger.info(
-            f"{variable_name} = aiplatform.{cls.__name__}.get('{resource.name}')"
-        )
+        self.info(f"{cls.__name__} created. Resource name: {resource.name}")
+        self.info(f"To use this {cls.__name__} in another session:")
+        self.info(f"{variable_name} = aiplatform.{cls.__name__}.get('{resource.name}')")
 
     def log_action_start_against_resource(
         self, action: str, noun: str, resource_noun_obj: "VertexAiResourceNoun"
@@ -160,7 +154,7 @@ def log_action_start_against_resource(
             resource_noun_obj (VertexAiResourceNoun):
                 Resource noun object the action is acting against.
         """
-        self._logger.info(
+        self.info(
             f"{action} {resource_noun_obj.__class__.__name__} {noun}: {resource_noun_obj.resource_name}"
         )
 
@@ -180,9 +174,7 @@ def log_action_started_against_resource_with_lro(
                 Resource noun object the action is acting against.
             lro (operation.Operation): Backing LRO for action.
         """
-        self._logger.info(
-            f"{action} {cls.__name__} {noun} backing LRO: {lro.operation.name}"
-        )
+        self.info(f"{action} {cls.__name__} {noun} backing LRO: {lro.operation.name}")
 
     def log_action_completed_against_resource(
         self, noun: str, action: str, resource_noun_obj: "VertexAiResourceNoun"
@@ -195,13 +187,18 @@ def log_action_completed_against_resource(
             resource_noun_obj (VertexAiResourceNoun):
                 Resource noun object the action is acting against
         """
-        self._logger.info(
+        self.info(
             f"{resource_noun_obj.__class__.__name__} {noun} {action}. Resource name: {resource_noun_obj.resource_name}"
         )
 
-    def __getattr__(self, attr: str):
-        """Forward remainder of logging to underlying logger."""
-        return getattr(self._logger, attr)
+
+def Logger(name: str) -> VertexLogger:  # pylint: disable=invalid-name
+    old_class = logging.getLoggerClass()
+    try:
+        logging.setLoggerClass(VertexLogger)
+        return logging.getLogger(name)
+    finally:
+        logging.setLoggerClass(old_class)
 
 
 _LOGGER = Logger(__name__)

google/cloud/aiplatform/pipeline_jobs.py

@@ -409,7 +409,7 @@ def submit(
                 credentials=self.credentials,
             )
         except:  # noqa: E722
-            _LOGGER._logger.exception(
+            _LOGGER.exception(
                 "Error when trying to get or create a GCS bucket for the pipeline output artifacts"
             )