fix: update recept
This commit is contained in:
@@ -59,7 +59,7 @@ class Logger(ILogger):
|
||||
def clear_trace_id(self) -> None:
|
||||
trace_id_var.set("N/A")
|
||||
|
||||
def _prepare_log_data(self, level: LogLevel, message: str) -> dict[str, Any]:
|
||||
def _prepare_log_data(self, level: LogLevel, message: Any) -> dict[str, Any]:
|
||||
current_frame = inspect.currentframe()
|
||||
if (
|
||||
current_frame
|
||||
@@ -75,21 +75,24 @@ class Logger(ILogger):
|
||||
line_number = 0
|
||||
|
||||
log_data = {
|
||||
"timestamp": datetime.now().isoformat(),
|
||||
"level": level.name,
|
||||
"instance_id": self.instance_id,
|
||||
"file": filename,
|
||||
"line": line_number,
|
||||
"trace_id": trace_id_var.get(),
|
||||
"message": message,
|
||||
'timestamp': datetime.now().isoformat(),
|
||||
'level': level.name,
|
||||
'instance_id': self.instance_id,
|
||||
'file': filename,
|
||||
'line': line_number,
|
||||
'trace_id': trace_id_var.get(),
|
||||
}
|
||||
if isinstance(message, dict):
|
||||
log_data.update(message)
|
||||
else:
|
||||
log_data['message'] = message
|
||||
|
||||
if level == LogLevel.EXCEPTION:
|
||||
log_data["exception"] = traceback.format_exc()
|
||||
log_data['exception'] = traceback.format_exc()
|
||||
|
||||
return log_data
|
||||
|
||||
def _log(self, level: LogLevel, message: str) -> None:
|
||||
def _log(self, level: LogLevel, message: Any) -> None:
|
||||
if level >= self.min_level:
|
||||
log_data = self._prepare_log_data(level, message)
|
||||
|
||||
@@ -100,9 +103,9 @@ class Logger(ILogger):
|
||||
f"{log_data['timestamp']} - {log_data['level']} - "
|
||||
f"{log_data['instance_id']} - {log_data['trace_id']} - "
|
||||
f"{log_data['file']}:{log_data['line']} - "
|
||||
f"{log_data['message']}"
|
||||
f"{log_data.get('message', log_data.get('event', ''))}"
|
||||
)
|
||||
if "exception" in log_data:
|
||||
if 'exception' in log_data:
|
||||
log_message += f"\nTraceback:\n{log_data['exception']}"
|
||||
|
||||
self._write(log_message)
|
||||
@@ -110,20 +113,20 @@ class Logger(ILogger):
|
||||
def _write(self, message: str) -> None:
|
||||
sys.stdout.write(message + "\n")
|
||||
|
||||
def debug(self, message: str) -> None:
|
||||
def debug(self, message: Any) -> None:
|
||||
self._log(LogLevel.DEBUG, message)
|
||||
|
||||
def info(self, message: str) -> None:
|
||||
def info(self, message: Any) -> None:
|
||||
self._log(LogLevel.INFO, message)
|
||||
|
||||
def warning(self, message: str) -> None:
|
||||
def warning(self, message: Any) -> None:
|
||||
self._log(LogLevel.WARNING, message)
|
||||
|
||||
def error(self, message: str) -> None:
|
||||
def error(self, message: Any) -> None:
|
||||
self._log(LogLevel.ERROR, message)
|
||||
|
||||
def critical(self, message: str) -> None:
|
||||
def critical(self, message: Any) -> None:
|
||||
self._log(LogLevel.CRITICAL, message)
|
||||
|
||||
def exception(self, message: str) -> None:
|
||||
def exception(self, message: Any) -> None:
|
||||
self._log(LogLevel.EXCEPTION, message)
|
||||
Reference in New Issue
Block a user