remove interface and add setup

Author: ohmayr

google/api_core/client_logging.py

@@ -1,16 +1,52 @@
 import logging
+import json
 
 
-class BaseLogger:
+# class BaseLogger:
 
-    def __init__(self, log_level=logging.CRITICAL):
-        self._logger = logging.getLogger("google")
-        self._logger.setLevel(log_level)
+#     def __init__(self, log_level=logging.DEBUG):
+#         self._logger = logging.getLogger("google")
+#         self._logger.setLevel(log_level)
+#         console_handler = logging.StreamHandler()
+#         console_handler.setLevel(log_level)
+#         formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+#         # formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(user_id)s - %(request_id)s - %(message)s')  # Include extra fields
+#         # formatter = DefaultFormatter()
+#         console_handler.setFormatter(formatter)
+#         self._logger.addHandler(console_handler)
+
+#     def get_logger(self):
+#         return self._logger
+
+
+class DefaultFormatter(logging.Formatter):
+    def format(self, record):
+        log_obj = {
+            'timestamp': self.formatTime(record),
+            'severity': record.levelname,
+            'name': record.name,
+            'message': record.getMessage(),
+        }
+        if hasattr(record, 'request'):
+            log_obj['request'] = record.request
+        if hasattr(record, 'response'):
+            log_obj['response'] = record.response
+        if hasattr(record, 'client_id'):
+            log_obj['client_id'] = record.client_id
+        
+        return json.dumps(log_obj)
+
+
+def setup_logging(log_level="DEBUG", namespace="google"):
+    if not logging.getLogger().hasHandlers():
+
+        # define a module for our repositories
+        logger = logging.getLogger(namespace)
+        logger.setLevel(log_level)
+
+        # Default settings
         console_handler = logging.StreamHandler()
         console_handler.setLevel(log_level)
         formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
         console_handler.setFormatter(formatter)
-        self._logger.addHandler(console_handler)
-
-    def get_logger(self):
-        return self._logger
+        logger.addHandler(console_handler)