改写战斗中替换group的逻辑

mower/utils/log.py

@@ -0,0 +1,114 @@
+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))