41 lines
1.5 KiB
Python
41 lines
1.5 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")
|
|
# 【新增】加载 machine 配置
|
|
self.config_machine = self._load_yaml("config/machine.yaml")
|
|
|
|
self.logger.info(f"项目名称: {self.config_sys.get('project_name')}")
|
|
self.logger.info(f"计算环境: {self.config_machine.get('current_system')}")
|
|
|
|
# 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: ... (后续我们将在这里实现调度逻辑) |