fix: update recept

This commit is contained in:
2026-05-09 10:57:02 +03:00
parent 195c0a8e53
commit ea0ca899ac
7 changed files with 59 additions and 37 deletions

View File

@@ -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)