mirror of
https://github.com/dobin/SuperMega
synced 2026-06-02 17:27:10 +00:00
Dobin
83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
import logging
|
|
|
|
from model.defs import *
|
|
|
|
|
|
# ANSI escape sequences for colors
|
|
class LogColors:
|
|
HEADER = '\033[95m'
|
|
BLUE = '\033[94m'
|
|
GREEN = '\033[92m'
|
|
WARNING = '\033[93m'
|
|
FAIL = '\033[91m'
|
|
ENDC = '\033[0m'
|
|
BOLD = '\033[1m'
|
|
UNDERLINE = '\033[4m'
|
|
|
|
|
|
class CustomFormatter(logging.Formatter):
|
|
#format = "%(asctime)s - %(name)-12s - [%(levelname)-8s] - %(message)s (%(filename)s:%(lineno)d)"
|
|
format = "(%(filename)-12s) %(message)s"
|
|
|
|
FORMATS = {
|
|
logging.DEBUG: format,
|
|
logging.INFO: format,
|
|
logging.WARNING: LogColors.WARNING + format + LogColors.ENDC,
|
|
logging.ERROR: LogColors.FAIL + format + LogColors.ENDC,
|
|
logging.CRITICAL: LogColors.FAIL + LogColors.BOLD + format + LogColors.ENDC
|
|
}
|
|
|
|
def format(self, record):
|
|
log_fmt = self.FORMATS.get(record.levelno)
|
|
formatter = logging.Formatter(log_fmt, datefmt="%Y-%m-%d %H:%M:%S")
|
|
return formatter.format(record)
|
|
|
|
|
|
class ListHandler(logging.Handler):
|
|
def __init__(self, log_list):
|
|
super().__init__()
|
|
self.log_list = log_list
|
|
|
|
def emit(self, record):
|
|
# Format the log record and store it in the list
|
|
log_entry = self.format(record)
|
|
self.log_list.append(log_entry)
|
|
|
|
|
|
class _MyLog():
|
|
def __init__(self):
|
|
self.log_messages = []
|
|
|
|
def log(self, message):
|
|
self.log_messages.append(message)
|
|
|
|
def getlog(self):
|
|
return self.log_messages
|
|
|
|
def clearlog(self):
|
|
self.log_messages.clear()
|
|
|
|
def setup_logging(self, level = logging.INFO):
|
|
root_logger = logging.getLogger()
|
|
root_logger.setLevel(level)
|
|
|
|
ch = logging.StreamHandler()
|
|
ch.setLevel(level)
|
|
ch.setFormatter(CustomFormatter())
|
|
|
|
list_handler = ListHandler(self.log_messages)
|
|
list_handler.setLevel(level)
|
|
list_handler.setFormatter(CustomFormatter())
|
|
|
|
root_logger.addHandler(ch)
|
|
root_logger.addHandler(list_handler)
|
|
|
|
def writelog(self):
|
|
# write log to file
|
|
with open(f"{logs_dir}/supermega.log", "w") as f:
|
|
for line in self.log_messages:
|
|
f.write(line + "\n")
|
|
|
|
|
|
MyLog: _MyLog = _MyLog()
|