[dy] Fix DBT logging (mage-ai#4577)

Author: dy46

mage_ai/data_preparation/logging/logger.py

@@ -38,6 +38,12 @@ def log(self, log_level, message, **kwargs):
     def warning(self, message, **kwargs):
         self.__send_message('warning', message, **kwargs)
 
+    def warn(self, message, **kwargs):
+        """
+        Same as warning, but in some cases we may need to use warn instead of warning
+        """
+        self.__send_message('warning', message, **kwargs)
+
     def __send_message(
         self,
         method_name,
@@ -80,6 +86,7 @@ def __getattr__(self, method_name):
             'info',
             'log',
             'warning',
+            'warn',
         ]:
             return getattr(self.logger, method_name)
         return getattr(self, method_name)

mage_ai/data_preparation/models/block/dbt/constants.py

@@ -21,3 +21,5 @@ class Flag(str, Enum):
 class LogLevel(str, Enum):
     DEBUG = 'debug'
     INFO = 'info'
+    WARN = 'warn'
+    ERROR = 'error'

mage_ai/data_preparation/models/block/dbt/dbt_cli.py

@@ -15,15 +15,10 @@
 
 def build_logging_callback(logging_func: Callable, log_level: LogLevel = None):
     def __callback(event, log_level=log_level, logging_func=logging_func):
-        # print('event.info', event.info)
-        # print('event.data', event.data)
-        # print(dir(event))
-        log_levels = {
-            LogLevel.INFO: True,
-        }
+        log_levels = set([LogLevel.INFO, LogLevel.WARN, LogLevel.ERROR])
 
         if LogLevel.DEBUG == log_level or event.info.level in log_levels:
-            logging_func(event.info.msg)
+            logging_func(event.info.level, event.info.msg)
 
     return __callback
 
@@ -58,7 +53,7 @@ def invoke(
         dbt = dbtRunner(
             callbacks=[
                 build_logging_callback(
-                    self.__debug if LogLevel.DEBUG == log_level else self.__info,
+                    self.__log,
                     log_level=log_level,
                 ),
             ],

mage_ai/frontend/components/Logs/Table/index.tsx

@@ -150,7 +150,7 @@ function LogsTable({
       uuid,
     } = logData || {};
 
-    let displayText = message || content;
+    let displayText = message == null ? content : message;
     if (Array.isArray(displayText)) {
       displayText = displayText.join(' ');
     } else if (typeof displayText === 'object') {