修复路径

app/tools/code_generator.py

@@ -7,6 +7,11 @@ import os
 class CodeGenerator:
     """通用代码生成器"""
     
+    # 添加类级别的缓存
+    _assets_dir_cache = None
+    _assets_dir_initialized = False
+    _template_dir_logged = False
+    
     @staticmethod
     def load_template(template_path: str) -> str:
         """加载代码模板"""
@@ -66,8 +71,12 @@ class CodeGenerator:
         # 使用统一的get_assets_dir方法来获取路径
         template_dir = CodeGenerator.get_assets_dir("User_code/bsp")
         
-        print(f"模板目录路径: {template_dir}")
-        if not os.path.exists(template_dir):
+        # 只在第一次或出现问题时打印日志
+        if not hasattr(CodeGenerator, '_template_dir_logged'):
+            print(f"模板目录路径: {template_dir}")
+            CodeGenerator._template_dir_logged = True
+            
+        if template_dir and not os.path.exists(template_dir):
             print(f"警告：模板目录不存在: {template_dir}")
         
         return template_dir
@@ -80,31 +89,78 @@ class CodeGenerator:
         Returns:
             str: 完整的assets路径
         """
-        if getattr(sys, 'frozen', False):
-            # 打包后的环境 - 使用可执行文件所在目录而不是临时目录
-            exe_dir = os.path.dirname(sys.executable)
-            assets_dir = os.path.join(exe_dir, "assets")
-            print(f"打包环境：尝试使用路径: {assets_dir}")
+        # 使用缓存机制，避免重复计算和日志输出
+        if not CodeGenerator._assets_dir_initialized:
+            assets_dir = ""
             
-            # 如果exe_dir/assets不存在，尝试使用sys._MEIPASS作为后备
-            if not os.path.exists(assets_dir) and hasattr(sys, '_MEIPASS'):
-                base_path = getattr(sys, '_MEIPASS')
-                assets_dir = os.path.join(base_path, "assets")
-                print(f"后备路径: {assets_dir}")
+            if getattr(sys, 'frozen', False):
+                # 打包后的环境
+                print("检测到打包环境")
+                
+                # 优先使用sys._MEIPASS（PyInstaller的临时解包目录）
+                if hasattr(sys, '_MEIPASS'):
+                    base_path = getattr(sys, '_MEIPASS')
+                    assets_dir = os.path.join(base_path, "assets")
+                    print(f"使用PyInstaller临时目录: {assets_dir}")
+                else:
+                    # 后备方案：使用可执行文件所在目录
+                    exe_dir = os.path.dirname(sys.executable)
+                    assets_dir = os.path.join(exe_dir, "assets")
+                    print(f"使用可执行文件目录: {assets_dir}")
+                
+                # 如果都不存在，尝试其他可能的位置
+                if not os.path.exists(assets_dir):
+                    # 尝试从当前工作目录查找
+                    cwd_assets = os.path.join(os.getcwd(), "assets")
+                    if os.path.exists(cwd_assets):
+                        assets_dir = cwd_assets
+                        print(f"从工作目录找到assets: {assets_dir}")
+                    else:
+                        print(f"警告：无法找到assets目录，使用默认路径: {assets_dir}")
+            else:
+                # 开发环境
+                current_dir = os.path.dirname(os.path.abspath(__file__))
+                
+                # 向上查找直到找到MRobot目录或到达根目录
+                while current_dir != os.path.dirname(current_dir):  # 防止无限循环
+                    if os.path.basename(current_dir) == 'MRobot':
+                        break
+                    parent = os.path.dirname(current_dir)
+                    if parent == current_dir:  # 已到达根目录
+                        break
+                    current_dir = parent
+                
+                assets_dir = os.path.join(current_dir, "assets")
+                print(f"开发环境：使用路径: {assets_dir}")
+                
+                # 如果找不到，尝试从当前工作目录
+                if not os.path.exists(assets_dir):
+                    cwd_assets = os.path.join(os.getcwd(), "assets")
+                    if os.path.exists(cwd_assets):
+                        assets_dir = cwd_assets
+                        print(f"开发环境后备：使用工作目录: {assets_dir}")
+            
+            # 缓存基础assets目录
+            CodeGenerator._assets_dir_cache = assets_dir
+            CodeGenerator._assets_dir_initialized = True
         else:
-            # 开发环境
-            current_dir = os.path.dirname(os.path.abspath(__file__))
-            while os.path.basename(current_dir) != 'MRobot' and current_dir != '/':
-                current_dir = os.path.dirname(current_dir)
-            assets_dir = os.path.join(current_dir, "assets")
-            print(f"开发环境：使用路径: {assets_dir}")
+            # 使用缓存的路径
+            assets_dir = CodeGenerator._assets_dir_cache or ""
         
+        # 构建完整路径
         if sub_path:
             full_path = os.path.join(assets_dir, sub_path)
         else:
             full_path = assets_dir
             
-        if not os.path.exists(full_path):
+        # 规范化路径（处理路径分隔符）
+        full_path = os.path.normpath(full_path)
+        
+        # 只在第一次访问某个路径时检查并警告
+        safe_sub_path = sub_path.replace('/', '_').replace('\\', '_')
+        warning_key = f"_warned_{safe_sub_path}"
+        if full_path and not os.path.exists(full_path) and not hasattr(CodeGenerator, warning_key):
             print(f"警告：资源目录不存在: {full_path}")
+            setattr(CodeGenerator, warning_key, True)
         
         return full_path