feat: Add custom_exception attribute to log records in CustomLogger c…

…lass
MolarVerse · May 5, 2024 · a0c9ce7 · a0c9ce7
1 parent cec6f78
commit a0c9ce7
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/PQAnalysis/utils/custom_logging.py b/PQAnalysis/utils/custom_logging.py
@@ -107,6 +107,8 @@ def _log(self,   # pylint: disable=arguments-differ
             if the level is logging.ERROR or logging.CRITICAL and the logger
             is not enabled for logging.DEBUG.
         """
+        self.custom_exception = exception
+
         self._original_log(
             level,
             msg,
@@ -164,6 +166,19 @@ def _original_log(self,
 
         super()._log(level, msg, args, **kwargs)
 
+    def makeRecord(self, name, level, fn, lno, msg, args, exc_info,
+                   func=None, extra=None, sinfo=None):
+        """
+        This method creates a log record.
+        """
+
+        record = super().makeRecord(name, level, fn, lno, msg, args, exc_info,
+                                    func=func, extra=extra, sinfo=sinfo)
+        if hasattr(self, 'custom_exception'):
+            setattr(record, 'custom_exception', self.custom_exception)
+
+        return record
+
     def error(self,
               msg: Any,
               *args,
@@ -321,7 +336,11 @@ def format(self, record: logging.LogRecord) -> str:
         longest_level_key = max(level_keys, key=len)
 
         name = record.name
-        header = f'{formatted_level} {name}\n\n'
+
+        if hasattr(record, 'custom_exception'):
+            header = f'{formatted_level} {name} - {record.custom_exception}\n\n'
+        else:
+            header = f'{formatted_level} {name}\n\n'
 
         messages = message.split('\n')
         wrapper = textwrap.TextWrapper(