114 lines
3.2 KiB
Python
114 lines
3.2 KiB
Python
import logging
|
|
import shutil
|
|
import sys
|
|
import time
|
|
import traceback
|
|
from datetime import datetime, timedelta
|
|
from logging.handlers import QueueHandler, QueueListener, TimedRotatingFileHandler
|
|
from pathlib import Path
|
|
from queue import Queue
|
|
from threading import Thread
|
|
|
|
import colorlog
|
|
|
|
from mower.utils import config
|
|
from mower.utils.path import get_path
|
|
|
|
BASIC_FORMAT = (
|
|
"%(asctime)s %(relativepath)s:%(lineno)d %(levelname)s %(funcName)s: %(message)s"
|
|
)
|
|
COLOR_FORMAT = f"%(log_color)s{BASIC_FORMAT}"
|
|
DATE_FORMAT = None
|
|
basic_formatter = logging.Formatter(BASIC_FORMAT, DATE_FORMAT)
|
|
color_formatter = colorlog.ColoredFormatter(COLOR_FORMAT, DATE_FORMAT)
|
|
|
|
|
|
class PackagePathFilter(logging.Filter):
|
|
def filter(self, record: logging.LogRecord) -> bool:
|
|
relativepath = Path(record.pathname)
|
|
try:
|
|
relativepath = relativepath.relative_to(get_path("@install"))
|
|
except ValueError:
|
|
pass
|
|
record.relativepath = relativepath
|
|
return True
|
|
|
|
|
|
filter = PackagePathFilter()
|
|
|
|
logger = logging.getLogger(__name__)
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
# d(ebug)hlr: 终端输出
|
|
dhlr = logging.StreamHandler(stream=sys.stdout)
|
|
dhlr.setFormatter(color_formatter)
|
|
dhlr.setLevel(logging.DEBUG)
|
|
dhlr.addFilter(filter)
|
|
|
|
# f(ile)hlr: 文件记录
|
|
folder = Path(get_path("@app/log"))
|
|
folder.mkdir(exist_ok=True, parents=True)
|
|
fhlr = TimedRotatingFileHandler(
|
|
folder.joinpath("runtime.log"), encoding="utf8", backupCount=168
|
|
)
|
|
fhlr.setFormatter(basic_formatter)
|
|
fhlr.setLevel("DEBUG")
|
|
fhlr.addFilter(filter)
|
|
|
|
|
|
class Handler(logging.StreamHandler):
|
|
def emit(self, record: logging.LogRecord):
|
|
msg = f"{record.asctime} {record.levelname} {record.message}"
|
|
if record.exc_info:
|
|
msg += "\n" + "".join(traceback.format_exception(*record.exc_info))
|
|
config.log_queue.put(msg)
|
|
|
|
|
|
# w(ebsocket)hlr: WebSocket
|
|
whlr = Handler()
|
|
whlr.setLevel(logging.INFO)
|
|
|
|
log_queue = Queue()
|
|
queue_handler = QueueHandler(log_queue)
|
|
logger.addHandler(queue_handler)
|
|
listener = QueueListener(log_queue, dhlr, fhlr, whlr, respect_handler_level=True)
|
|
listener.start()
|
|
|
|
screenshot_folder = get_path("@app/screenshot")
|
|
screenshot_folder.mkdir(exist_ok=True, parents=True)
|
|
screenshot_queue = Queue()
|
|
cleanup_time = datetime.now()
|
|
|
|
|
|
def screenshot_cleanup():
|
|
logger.info("清理过期截图")
|
|
start_time_ns = time.time_ns() - config.conf.screenshot * 3600 * 10**9
|
|
for i in screenshot_folder.iterdir():
|
|
if i.is_dir():
|
|
shutil.rmtree(i)
|
|
elif not i.stem.isnumeric():
|
|
i.unlink()
|
|
elif int(i.stem) < start_time_ns:
|
|
i.unlink()
|
|
global cleanup_time
|
|
cleanup_time = datetime.now()
|
|
|
|
|
|
def screenshot_worker():
|
|
screenshot_cleanup()
|
|
while True:
|
|
now = datetime.now()
|
|
if now - cleanup_time > timedelta(hours=1):
|
|
screenshot_cleanup()
|
|
img, filename = screenshot_queue.get()
|
|
with screenshot_folder.joinpath(filename).open("wb") as f:
|
|
f.write(img)
|
|
|
|
|
|
Thread(target=screenshot_worker, daemon=True).start()
|
|
|
|
|
|
def save_screenshot(img: bytes) -> None:
|
|
filename = f"{time.time_ns()}.jpg"
|
|
logger.debug(filename)
|
|
screenshot_queue.put((img, filename))
|