Files
NEP-auto/nep_auto/driver.py
2025-12-08 17:14:27 +08:00

37 lines
1.3 KiB
Python

import yaml
import time
import logging
from pathlib import Path
from nep_auto.status_manager import StatusManager
class NEPDriver:
def __init__(self):
self.logger = logging.getLogger("NEP_Auto")
self.root = Path(".")
# 1. 加载配置
self.config_sys = self._load_yaml("config/system.yaml")
self.config_param = self._load_yaml("config/param.yaml")
self.logger.info(f"项目名称: {self.config_sys.get('project_name')}")
# 2. 初始化状态管理器
self.status = StatusManager(self.root / "workspace")
def _load_yaml(self, path):
if not Path(path).exists():
raise FileNotFoundError(f"配置文件缺失: {path}")
with open(path, 'r') as f:
return yaml.safe_load(f)
def run(self):
"""主循环"""
self.logger.info("✅ 驱动器初始化完成,准备进入主循环...")
# 获取当前轮次
current_iter = self.status.get_current_iter()
self.logger.info(f"当前进度: iter_{current_iter:03d}")
# 暂时只打印一次就退出,用于测试环境
self.logger.info("测试阶段:环境检查通过。等待模块代码实现...")
# while True: ... (后续我们将在这里实现调度逻辑)