Add trace data to record

AlexxIT · Apr 13, 2024 · 804b94f · 804b94f
1 parent b2f5c81
commit 804b94f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/custom_components/hass_diagnostics/core/smart_log.py b/custom_components/hass_diagnostics/core/smart_log.py
@@ -118,6 +118,7 @@ def convert_log_entry_to_record(entry: dict):
         "levelname": entry["level"],
         "created": entry["timestamp"],
         "pathname": entry["source"][0],
+        "lineno": entry["source"][1],
         "message": entry["message"][0],
         "exc_info": entry["exception"] != "",
         "exc_text": entry["exception"],

diff --git a/custom_components/hass_diagnostics/sensor.py b/custom_components/hass_diagnostics/sensor.py
@@ -63,9 +63,21 @@ def emit(self, record: logging.LogRecord, count: int = 1):
         if item := self.records.get(key):
             item["count"] += count
         else:
+            entry["count"] = count
+            if record.exc_text:
+                entry["exception"] = record.exc_text
+            if hasattr(record, "funcName"):
+                entry["source"] = [
+                    record.pathname,
+                    record.lineno,
+                    record.processName,
+                    record.threadName,
+                    record.funcName,
+                ]
+            else:
+                entry["source"] = [record.pathname, record.lineno]
             if github := github_get_link(record):
                 entry["github"] = github
-            entry["count"] = count
             self.records[key] = entry
 
         self._attr_native_value += count