feat(chuni): Improved logging

core/logger/adapter.py

@@ -0,0 +1,46 @@
+from logging import Formatter, LogRecord, LoggerAdapter
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from typing import Any, MutableMapping
+
+# skip natural LogRecord attributes
+# http://docs.python.org/library/logging.html#logrecord-attributes
+RESERVED_ATTRS: set[str] = {
+    'args', 'asctime', 'created', 'exc_info', 'exc_text', 'filename',
+    'funcName', 'levelname', 'levelno', 'lineno', 'module',
+    'msecs', 'message', 'msg', 'name', 'pathname', 'process',
+    'processName', 'relativeCreated', 'stack_info', 'thread', 'threadName'}
+
+
+class MergedLoggerAdapter(LoggerAdapter):
+    def process(self, msg: "Any", kwargs: "MutableMapping[str, Any]") -> tuple["Any", "MutableMapping[str, Any]"]:
+        kwargs["extra"] = {**self.extra, **kwargs.get("extra", {})}
+
+        return msg, kwargs
+
+
+class ExtraFormatter(Formatter):
+    def format(self, record: LogRecord):
+        s = super().format(record)
+
+        extras = {}
+
+        for key, value in record.__dict__.items():
+            if key in {"title", "taskName"} or key in RESERVED_ATTRS or (hasattr(key, "startswith") and key.startswith("_")):
+                continue
+
+            extras[key] = value
+        
+        if len(extras) == 0:
+            return s
+        
+        str_extras = str(extras)
+
+        if len(str_extras) >= 300:
+            str_extras = str_extras[:297] + "..."
+        
+        s += " " + str_extras
+
+        return s
+