85 lines
2.4 KiB
Python
85 lines
2.4 KiB
Python
# src/utils.py
|
|
import yaml
|
|
import logging
|
|
import os
|
|
import sys
|
|
|
|
|
|
def load_yaml(path):
|
|
"""加载 YAML 配置文件"""
|
|
if not os.path.exists(path):
|
|
logging.error(f"Config file not found: {path}")
|
|
sys.exit(1)
|
|
with open(path, 'r') as f:
|
|
return yaml.safe_load(f)
|
|
|
|
|
|
def setup_logger(work_dir, log_file="autonep.log"):
|
|
"""配置日志：同时输出到文件和控制台"""
|
|
logger = logging.getLogger()
|
|
logger.setLevel(logging.INFO)
|
|
|
|
# 清楚之前的 handler 防止重复
|
|
if logger.hasHandlers():
|
|
logger.handlers.clear()
|
|
|
|
# 文件 Handler
|
|
file_handler = logging.FileHandler(os.path.join(work_dir, log_file))
|
|
file_handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
|
|
logger.addHandler(file_handler)
|
|
|
|
# 控制台 Handler
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setFormatter(logging.Formatter('%(message)s')) # 控制台只看消息，简洁点
|
|
logger.addHandler(console_handler)
|
|
|
|
return logger
|
|
|
|
|
|
class Notifier:
|
|
"""(预留) 通知模块"""
|
|
|
|
def __init__(self, url=None):
|
|
self.url = url
|
|
|
|
def send(self, title, msg, priority=5):
|
|
# 暂时只打印日志，不实际发送
|
|
logging.info(f"[[Notification]] {title}: {msg}")
|
|
|
|
|
|
# src/utils.py 添加在最后
|
|
|
|
def run_cmd_with_log(cmd, cwd, log_file="exec.log", input_str=None):
|
|
"""
|
|
执行命令并将 stdout/stderr 重定向到日志文件
|
|
"""
|
|
import subprocess
|
|
|
|
log_path = os.path.join(cwd, log_file)
|
|
mode = 'a' if os.path.exists(log_path) else 'w'
|
|
|
|
with open(log_path, mode) as f:
|
|
f.write(f"\n\n>>> Executing: {cmd}\n")
|
|
f.write(f">>> Input: {repr(input_str)}\n")
|
|
f.write("-" * 40 + "\n")
|
|
f.flush()
|
|
|
|
try:
|
|
process = subprocess.Popen(
|
|
cmd,
|
|
shell=True,
|
|
cwd=cwd,
|
|
stdin=subprocess.PIPE if input_str else None,
|
|
stdout=f, # 直接指向文件
|
|
stderr=subprocess.STDOUT, # 把错误也合并到同一个日志
|
|
text=True
|
|
)
|
|
|
|
# 发送输入并等待
|
|
process.communicate(input=input_str)
|
|
|
|
f.write(f"\n>>> Finished with Return Code: {process.returncode}\n")
|
|
return process.returncode == 0
|
|
except Exception as e:
|
|
f.write(f"\n>>> Exception: {str(e)}\n")
|
|
return False |