nep框架重构

main.py

@@ -1,32 +1,33 @@
+# main.py
+import os
 import sys
-import time
-import traceback
-from nep_auto.driver import NEPDriver
-from nep_auto.utils.logger import setup_logger
+from src.utils import setup_logger
+from src.workflow import Workflow
 
 
 def main():
-    # 1. 初始化全局日志
-    logger = setup_logger("logs/sys_runtime.log")
-    logger.info("========================================")
-    logger.info("🚀 NEP Automation Framework Starting...")
-    logger.info("========================================")
+    root_dir = os.getcwd()
 
+    # 1. 初始化日志
+    # 既然 workspace 还没创建，先放到根目录，Workflow 初始化后再放到 workspace 也可以
+    # 这里简单起见放在根目录
+    setup_logger(root_dir)
+
+    # 2. 检查基本文件是否存在
+    required_dirs = ['config', 'data', 'template']
+    for d in required_dirs:
+        if not os.path.exists(os.path.join(root_dir, d)):
+            print(f"Error: Missing directory '{d}'. Please check file structure.")
+            sys.exit(1)
+
+    # 3. 启动工作流
     try:
-        # 2. 初始化驱动器 (加载配置，恢复状态)
-        driver = NEPDriver()
-
-        # 3. 启动主循环
-        driver.run()
-
-    except KeyboardInterrupt:
-        logger.warning("⚠️ 用户手动中断程序 (KeyboardInterrupt)")
-        sys.exit(0)
+        app = Workflow(root_dir)
+        app.run()
     except Exception as e:
-        logger.error(f"❌ 程序发生严重崩溃: {str(e)}")
-        logger.error(traceback.format_exc())
-        # 这里可以加入发送崩溃通知的逻辑
-        sys.exit(1)
+        import traceback
+        traceback.print_exc()
+        print(f"Critical Error: {e}")
 
 
 if __name__ == "__main__":