63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
import os
|
|
from datetime import datetime
|
|
|
|
|
|
class Logger:
|
|
def __init__(self, log_file_path: str):
|
|
self.log_file = log_file_path
|
|
self._init_log_file()
|
|
|
|
def _init_log_file(self):
|
|
"""Initialize log file if it doesn't exist"""
|
|
log_dir = os.path.dirname(self.log_file)
|
|
if not os.path.exists(log_dir):
|
|
os.makedirs(log_dir)
|
|
if not os.path.exists(self.log_file):
|
|
try:
|
|
with open(self.log_file, "w", encoding="utf-8") as f:
|
|
f.write("")
|
|
except Exception as e:
|
|
print(f"Error initializing log file {self.log_file}: {e}")
|
|
|
|
def append_log(self, message: str) -> None:
|
|
"""Append a message to the log file with timestamp."""
|
|
try:
|
|
timestamp = datetime.now().strftime("[%H:%M:%S] ")
|
|
lines = message.split("\n")
|
|
lines_with_timestamp = [
|
|
f"{timestamp}{line}\n" for line in lines if line.strip()
|
|
]
|
|
|
|
if lines_with_timestamp:
|
|
with open(self.log_file, "a", encoding="utf-8") as f:
|
|
f.writelines(lines_with_timestamp)
|
|
except Exception as e:
|
|
print(f"Error writing to log file {self.log_file}: {e}")
|
|
|
|
def read_log(self, lines_limit: int = None) -> str:
|
|
"""Read the log file, optionally limiting to the last N lines"""
|
|
try:
|
|
with open(self.log_file, "r", encoding="utf-8") as f:
|
|
if lines_limit is None:
|
|
return f.read()
|
|
else:
|
|
# Read all lines and get the last N lines
|
|
all_lines = f.readlines()
|
|
if len(all_lines) <= lines_limit:
|
|
return "".join(all_lines)
|
|
else:
|
|
return "".join(all_lines[-lines_limit:])
|
|
except Exception as e:
|
|
print(f"Error reading log file {self.log_file}: {e}")
|
|
return ""
|
|
|
|
def clear_log(self) -> bool:
|
|
"""Clear the log file"""
|
|
try:
|
|
with open(self.log_file, "w", encoding="utf-8") as f:
|
|
f.write("")
|
|
return True
|
|
except Exception as e:
|
|
print(f"Error clearing log file {self.log_file}: {e}")
|
|
return False
|