get func name

nationalarchives · Dec 10, 2024 · cfc10e1 · cfc10e1
1 parent 8a9d4ec
commit cfc10e1
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/app/logger_config.py b/app/logger_config.py
@@ -1,3 +1,4 @@
+import inspect
 import json
 import logging
 
@@ -11,7 +12,7 @@ def format(self, record):
             "timestamp": self.formatTime(record),
             "level": record.levelname,
             "module": record.module,
-            "function": record.funcName,
+            "function": self.get_original_function_name(record),
             "line_number": record.lineno,
         }
 
@@ -34,6 +35,12 @@ def format(self, record):
 
         return json.dumps(log_record)
 
+    def get_original_function_name(self, record):
+        for frame in inspect.stack():
+            if frame.function == record.funcName:
+                return frame.function
+        return record.funcName
+
 
 def setup_logger(name, level, formatter):
     """Helper function to set up a logger."""