BE: logging format

src/mainframe/clients/logs.py

@@ -1,4 +1,3 @@
-import json
 import logging
 
 import axiom_py
@@ -9,25 +8,16 @@
 logging.basicConfig(format=FORMAT, level=logging.INFO)
 
 
-def parse_value(value):
-    try:
-        json.dumps(value)
-        return value
-    except TypeError:
-        return ""
-
-
-def parse_record(record: logging.LogRecord) -> dict:
-    record = {
-        field: parse_value(getattr(record, field, None)) for field in record.__dict__
-    }
-    record["message"] = f"[Mainframe] {record['message']}"
-    return record
+class MainframeHandler(AxiomHandler):
+    def emit(self, record: logging.LogRecord) -> None:
+        record.message = record.msg % record.args
+        record.env = settings.ENV
+        super().emit(record)
 
 
 def get_default_logger(name):
     logger = logging.getLogger(name)
-    if settings.ENV in ("prod", "rpi") and not logger.handlers:
+    if settings.ENV in ("local",) and not logger.handlers:
         client = axiom_py.Client()
-        logger.addHandler(AxiomHandler(client, "mainframe"))
+        logger.addHandler(MainframeHandler(client, "mainframe"))
     return logger