71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
from datetime import datetime
|
|
|
|
|
|
class DebugLogger:
|
|
"""In-memory logger for code execution events.
|
|
|
|
Collects timestamped INFO and ERROR entries during a single run.
|
|
Call clear() before each new execution to start fresh.
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.logs: list[dict] = []
|
|
|
|
def log(self, message: str) -> None:
|
|
"""Append a general info message."""
|
|
self.logs.append({
|
|
"level": "INFO",
|
|
"message": message,
|
|
"timestamp": datetime.now().strftime("%H:%M:%S"),
|
|
})
|
|
|
|
def log_error(self, error_message: str) -> None:
|
|
"""Append an error message."""
|
|
self.logs.append({
|
|
"level": "ERROR",
|
|
"message": error_message,
|
|
"timestamp": datetime.now().strftime("%H:%M:%S"),
|
|
})
|
|
|
|
def get_logs(self) -> list[dict]:
|
|
"""Return a copy of all collected log entries."""
|
|
return list(self.logs)
|
|
|
|
def clear(self) -> None:
|
|
"""Reset the log — call before each new execution."""
|
|
self.logs = []
|
|
|
|
def format_debug_output(self, output: dict) -> str:
|
|
"""Format an ExecutionEngine result dict into a human-readable string.
|
|
|
|
Args:
|
|
output: dict with keys 'stdout', 'stderr', and 'rc'.
|
|
|
|
Returns:
|
|
A formatted string ready for display in the UI.
|
|
"""
|
|
lines = []
|
|
|
|
status = "SUCCESS" if output.get("rc") == 0 else "FAILED"
|
|
lines.append(f"[{status}] Exit code: {output.get('rc')}")
|
|
|
|
if output.get("stdout"):
|
|
lines.append("\n--- stdout ---")
|
|
lines.append(output["stdout"].rstrip())
|
|
|
|
if output.get("stderr"):
|
|
lines.append("\n--- stderr ---")
|
|
lines.append(output["stderr"].rstrip())
|
|
|
|
if not output.get("stdout") and not output.get("stderr"):
|
|
lines.append("No output produced.")
|
|
|
|
for entry in self.logs:
|
|
lines.append(f"[{entry['timestamp']}] [{entry['level']}] {entry['message']}")
|
|
|
|
return "\n".join(lines)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logger = DebugLogger()
|
|
print(logger.get_logs()) |