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: ... (后续我们将在这里实现调度逻辑)