This commit is contained in:
HuangHai
2026-01-14 07:28:06 +08:00
parent 4cc6bc06d1
commit af60249fce
19 changed files with 166 additions and 28 deletions

View File

@@ -246,6 +246,37 @@ def detect_rabbit_ad_close(image_path, debug_dir=None):
return [best[3], best[4]]
def setup_logger(name, log_file=None):
"""
配置日志,支持同时输出到控制台和文件
"""
if log_file is None:
log_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "Logs")
if not os.path.exists(log_dir):
os.makedirs(log_dir)
log_file = os.path.join(log_dir, f"{name}_{time.strftime('%Y%m%d')}.log")
logger = logging.getLogger(name)
logger.setLevel(logging.INFO)
# 如果已经有 handler,不再添加
if logger.handlers:
return logger
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
# 控制台 Handler
ch = logging.StreamHandler()
ch.setFormatter(formatter)
logger.addHandler(ch)
# 文件 Handler
fh = logging.FileHandler(log_file, encoding='utf-8')
fh.setFormatter(formatter)
logger.addHandler(fh)
return logger
def detect_black_agree_button(image_path, debug_dir=None):
"""