auto-coder 0.1.264__py3-none-any.whl → 0.1.266__py3-none-any.whl

autocoder/common/conf_import_export.py

@@ -0,0 +1,105 @@
+import os
+import json
+import shutil
+from loguru import logger
+from autocoder.common.printer import Printer
+from autocoder.common.result_manager import ResultManager
+
+result_manager = ResultManager()
+
+def export_conf(project_root: str, export_path: str) -> bool:
+    printer = Printer()
+    """
+    Export conf from memory.json to a specified directory
+
+    Args:
+        project_root: Project root directory
+        export_path: Path to export the conf file
+
+    Returns:
+        bool: True if successful, False otherwise
+    """
+    project_root = os.path.abspath(project_root) or os.getcwd()
+    try:
+        memory_path = os.path.join(project_root, ".auto-coder", "plugins", "chat-auto-coder", "memory.json")
+        if not os.path.exists(memory_path):
+            printer.print_in_terminal("conf_not_found", path=memory_path)
+            return False
+
+        # Read and extract conf
+        with open(memory_path, "r",encoding="utf-8") as f:
+            memory_data = json.load(f)
+
+        conf_data = memory_data.get("conf", {})
+
+        # Write to export location
+        export_file = os.path.join(export_path, "conf.json")
+        os.makedirs(export_path, exist_ok=True)
+        with open(export_file, "w",encoding="utf-8") as f:
+            json.dump(conf_data, f, indent=2)
+        printer.print_in_terminal("conf_export_success", path=export_file)
+        result_manager.add_result(content=printer.get_message_from_key_with_format("conf_export_success", path=export_file), meta={"action": "conf_export", "input": {
+            "path": export_file
+        }})
+        return True
+
+    except Exception as e:
+        result_manager.add_result(content=printer.get_message_from_key_with_format("conf_export_error", error=str(e)), meta={"action": "conf_export", "input": {
+            "path": export_file
+        }})
+        printer.print_in_terminal("conf_export_error", error=str(e))
+        return False
+
+
+def import_conf(project_root: str, import_path: str) -> bool:
+    project_root = os.path.abspath(project_root) or os.getcwd()
+    printer = Printer()
+    """
+    Import conf from a specified directory into memory.json
+
+    Args:
+        project_root: Project root directory
+        import_path: Path containing the conf file to import
+
+    Returns:
+        bool: True if successful, False otherwise
+    """
+    try:
+        import_file = os.path.join(import_path, "conf.json")
+        if not os.path.exists(import_file):
+            printer.print_in_terminal("conf_not_found", path=import_file)
+            return False
+
+        # Read conf file
+        with open(import_file, "r",encoding="utf-8") as f:
+            conf_data = json.load(f)
+
+        # Backup existing memory
+        memory_path = os.path.join(project_root, ".auto-coder", "plugins", "chat-auto-coder", "memory.json")
+        if os.path.exists(memory_path):
+            backup_path = memory_path + ".bak"
+            shutil.copy2(memory_path, backup_path)
+            printer.print_in_terminal("conf_backup_success", path=backup_path)
+
+        # Update conf in memory
+        with open(memory_path, "r",encoding="utf-8") as f:
+            memory_data = json.load(f)
+
+        memory_data["conf"] = conf_data
+
+        # Write updated memory
+        with open(memory_path, "w",encoding="utf-8") as f:
+            json.dump(memory_data, f, indent=2)
+        
+        printer.print_in_terminal("conf_import_success", path=memory_path)
+        result_manager.add_result(content=printer.get_message_from_key_with_format("conf_import_success", path=memory_path), meta={"action": "conf_import", "input": {
+            "path": memory_path
+        }})
+        return True
+
+    except Exception as e:
+        result_manager.add_result(content=printer.get_message_from_key_with_format("conf_import_error", error=str(e)), meta={"action": "conf_import", "input": {
+            "path": memory_path
+        }})
+        printer.print_in_terminal("conf_import_error", error=str(e))
+        return False

autocoder/common/conf_validator.py

@@ -142,7 +142,7 @@ class ConfigValidator:
     }
 
     @classmethod
-    def validate(cls, key: str, value: Any, product_mode: str) -> Any:
+    def validate(cls, key: str, value: Any, product_mode: str) -> Any:        
         # 获取配置规范
         spec = cls.CONFIG_SPEC.get(key)
         if not spec:

autocoder/common/files.py

@@ -1,5 +1,5 @@
 from autocoder.common.auto_coder_lang import get_message_with_format
-from typing import List, Dict, Union
+from typing import List, Dict, Union, Generator, Tuple
 
 def read_file(file_path):
     """Read a file with automatic encoding detection.
@@ -30,6 +30,45 @@ def read_file(file_path):
         file_path=file_path, 
         encodings=", ".join(encodings)))
 
+def read_lines(file_path:str):
+    encodings = ['utf-8', 'gbk', 'utf-16', 'latin-1']
+    for encoding in encodings:
+        try:
+            with open(file_path, 'r', encoding=encoding) as f:
+                return f.readlines()
+        except UnicodeDecodeError:
+            continue
+    raise ValueError(get_message_with_format("file_decode_error", 
+        file_path=file_path, 
+        encodings=", ".join(encodings)))
+
+
+
+def read_file_with_line_numbers(file_path: str,line_number_start:int=0) -> Generator[Tuple[int, str], None, None]:
+    """Read a file and return its content with line numbers.
+
+    Args:
+        file_path (str): Path to the file to read
+
+    Returns:
+        List[str]: A list of strings where each string is in the format "line_number:line_content"
+
+    Raises:
+        ValueError: If the file cannot be decoded with any of the tried encodings
+    """
+    encodings = ['utf-8', 'gbk', 'utf-16', 'latin-1']
+    
+    for encoding in encodings:
+        try:
+            with open(file_path, 'r', encoding=encoding) as file:
+                for line_number, line in enumerate(file, start=line_number_start):
+                    yield (line_number, line)                
+        except UnicodeDecodeError:
+            continue
+
+    raise ValueError(get_message_with_format("file_decode_error", 
+        file_path=file_path, 
+        encodings=", ".join(encodings)))
 
 
 def save_file(file_path: str, content: Union[str, List[str]]) -> None:
@@ -55,4 +94,4 @@ def save_file(file_path: str, content: Union[str, List[str]]) -> None:
     except IOError as e:
         raise IOError(get_message_with_format("file_write_error",
             file_path=file_path,
-            error=str(e)))
+            error=str(e)))

autocoder/common/image_to_page.py

@@ -122,7 +122,7 @@ class ImageToPageDirectly:
 
         counter = 1 
         target_html_path = os.path.join(html_dir,f"{html_file_name}-{counter}.html") 
-        with open(target_html_path, "w") as f:
+        with open(target_html_path, "w",encoding="utf-8") as f:
             f.write(html)               
         
         while counter < max_iter:
@@ -137,11 +137,11 @@ class ImageToPageDirectly:
             
             target_html_path = os.path.join(html_dir,f"{html_file_name}-{counter}.html")
             logger.info(f"generate html: {target_html_path}")                
-            with open(target_html_path, "w") as f:
+            with open(target_html_path, "w",encoding="utf-8") as f:
                 f.write(html)            
         
         logger.info(f"finally generate html: {html_path}")
-        with open(html_path, "w") as f:
+        with open(html_path, "w",encoding="utf-8") as f:
             f.write(html)    
              
 
@@ -248,7 +248,7 @@ class ImageToPage:
                 file_path = block.path
                 os.makedirs(os.path.dirname(file_path), exist_ok=True)
 
-                with open(file_path, "w") as f:
+                with open(file_path, "w",encoding="utf-8") as f:
                     logger.info(f"Upsert path: {file_path}")                                       
                     f.write(block.content)
                     file_modified_num += 1
@@ -268,7 +268,7 @@ class ImageToPage:
         ## generate html by image description        
         content_contains_html_prompt = self.generate_html.prompt(desc,html_path) 
 
-        with open(self.args.target_file, "w") as f:
+        with open(self.args.target_file, "w",encoding="utf-8") as f:
             f.write(content_contains_html_prompt) 
 
         t = self.llm.chat_oai(conversations=[{
@@ -278,7 +278,7 @@ class ImageToPage:
 
         content_contains_html = t[0].output 
 
-        with open(self.args.target_file, "w") as f:
+        with open(self.args.target_file, "w",encoding="utf-8") as f:
             f.write(content_contains_html)
         
         
@@ -296,7 +296,7 @@ class ImageToPage:
 
         for i in range(max_iter):
             logger.info(f"iterate  {i}")
-            with open(html_path,"r") as f:
+            with open(html_path,"r",encoding="utf-8") as f:
                 prev_html = f.read()
 
             gen_screenshots(url=html_path,image_dir=new_image_dir)        
@@ -309,7 +309,7 @@ class ImageToPage:
             ## get new description prompt by comparing old and new image
             new_desc_prompt = self.get_optimize(self.score(origin_image,new_image))
 
-            with open(self.args.target_file, "w") as f:
+            with open(self.args.target_file, "w",encoding="utf-8") as f:
                 f.write(new_desc_prompt) 
 
             t = self.llm.chat_oai(conversations=[{
@@ -319,7 +319,7 @@ class ImageToPage:
 
             new_desc = t[0].output 
 
-            with open(self.args.target_file, "w") as f:
+            with open(self.args.target_file, "w",encoding="utf-8") as f:
                 f.write(new_desc)            
 
             logger.info(f"score old/new image: {new_desc}")
@@ -327,7 +327,7 @@ class ImageToPage:
             ## generate new html by new description
             optimze_html_prompt =  self.optimize_html.prompt(desc=new_desc,html=prev_html,html_path=html_path)                                    
             
-            with open(self.args.target_file, "w") as f:
+            with open(self.args.target_file, "w",encoding="utf-8") as f:
                 f.write(optimze_html_prompt) 
 
             t = self.llm.chat_oai(conversations=[{
@@ -336,7 +336,7 @@ class ImageToPage:
             }],llm_config={**extra_llm_config}) 
             new_code = t[0].output
             
-            with open(self.args.target_file, "w") as f:
+            with open(self.args.target_file, "w",encoding="utf-8") as f:
                 f.write(new_code) 
 
             self.write_code(new_code,html_path)

autocoder/common/index_import_export.py

@@ -3,6 +3,9 @@ import json
 import shutil
 from loguru import logger
 from autocoder.common.printer import Printer
+from autocoder.common.result_manager import ResultManager
+
+result_manager = ResultManager()
 
 
 def export_index(project_root: str, export_path: str) -> bool:
@@ -22,11 +25,11 @@ def export_index(project_root: str, export_path: str) -> bool:
         if not os.path.exists(index_path):
             printer.print_in_terminal("index_not_found", path=index_path)
             return False
-            
+
         # Read and convert paths
-        with open(index_path, "r") as f:
+        with open(index_path, "r",encoding="utf-8") as f:
             index_data = json.load(f)
-            
+
         # Convert absolute paths to relative
         converted_data = {}
         for abs_path, data in index_data.items():
@@ -35,21 +38,29 @@ def export_index(project_root: str, export_path: str) -> bool:
                 data["module_name"] = rel_path
                 converted_data[rel_path] = data
             except ValueError:
-                printer.print_in_terminal("index_convert_path_fail", path=abs_path)
+                printer.print_in_terminal(
+                    "index_convert_path_fail", path=abs_path)
                 converted_data[abs_path] = data
-        
+
         # Write to export location
         export_file = os.path.join(export_path, "index.json")
         os.makedirs(export_path, exist_ok=True)
-        with open(export_file, "w") as f:
+        with open(export_file, "w",encoding="utf-8") as f:
             json.dump(converted_data, f, indent=2)
-            
+        printer.print_in_terminal("index_export_success", path=export_file)
+        result_manager.add_result(content=printer.get_message_from_key_with_format("index_export_success", path=export_file), meta={"action": "index_export", "input": {
+            "path": export_file
+        }})
         return True
-        
+
     except Exception as e:
         printer.print_in_terminal("index_error", error=str(e))
+        result_manager.add_result(content=printer.get_message_from_key_with_format("index_error", error=str(e)), meta={"action": "index_export", "input": {
+            "path": export_file
+        }})
         return False
 
+
 def import_index(project_root: str, import_path: str) -> bool:
     printer = Printer()
     """
@@ -67,11 +78,11 @@ def import_index(project_root: str, import_path: str) -> bool:
         if not os.path.exists(import_file):
             printer.print_in_terminal("index_not_found", path=import_file)
             return False
-            
+
         # Read and convert paths
-        with open(import_file, "r") as f:
+        with open(import_file, "r",encoding="utf-8") as f:
             index_data = json.load(f)
-            
+
         # Convert relative paths to absolute
         converted_data = {}
         for rel_path, data in index_data.items():
@@ -80,22 +91,31 @@ def import_index(project_root: str, import_path: str) -> bool:
                 data["module_name"] = abs_path
                 converted_data[abs_path] = data
             except Exception:
-                printer.print_in_terminal("index_convert_path_fail", path=rel_path)
+                printer.print_in_terminal(
+                    "index_convert_path_fail", path=rel_path)
                 converted_data[rel_path] = data
-        
+
         # Backup existing index
         index_path = os.path.join(project_root, ".auto-coder", "index.json")
         if os.path.exists(index_path):
             backup_path = index_path + ".bak"
             shutil.copy2(index_path, backup_path)
             printer.print_in_terminal("index_backup_success", path=backup_path)
-            
+
         # Write new index
-        with open(index_path, "w") as f:
+        with open(index_path, "w",encoding="utf-8") as f:
             json.dump(converted_data, f, indent=2)
-            
-        return True
         
+        printer.print_in_terminal("index_import_success", path=index_path)
+        result_manager.add_result(content=printer.get_message_from_key_with_format("index_import_success", path=index_path), meta={"action": "index_import", "input": {
+            "path": index_path
+        }})
+
+        return True
+
     except Exception as e:
         printer.print_in_terminal("index_error", error=str(e))
-        return False
+        result_manager.add_result(content=printer.get_message_from_key_with_format("index_error", error=str(e)), meta={"action": "index_import", "input": {
+            "path": index_path
+        }})
+        return False

autocoder/common/mcp_hub.py

@@ -116,7 +116,7 @@ class McpHub:
     def _write_default_settings(self):
         """Write default MCP settings file"""
         default_settings = {"mcpServers": {}}
-        with open(self.settings_path, "w") as f:
+        with open(self.settings_path, "w",encoding="utf-8") as f:
             json.dump(default_settings, f, indent=2)
 
     async def add_server_config(self, name: str, config:Dict[str,Any]) -> None:
@@ -129,7 +129,7 @@ class McpHub:
         try:
             settings = self._read_settings()            
             settings["mcpServers"][name] = config
-            with open(self.settings_path, "w") as f:
+            with open(self.settings_path, "w",encoding="utf-8") as f:
                 json.dump(settings, f, indent=2, ensure_ascii=False)
             await self.initialize()
             logger.info(f"Added/updated MCP server config: {name}")
@@ -148,7 +148,7 @@ class McpHub:
             settings = self._read_settings()
             if name in settings["mcpServers"]:
                 del settings["mcpServers"][name]
-                with open(self.settings_path, "w") as f:
+                with open(self.settings_path, "w",encoding="utf-8") as f:
                     json.dump(settings, f, indent=2, ensure_ascii=False)
                 logger.info(f"Removed MCP server config: {name}")
                 await self.initialize()

autocoder/common/mcp_server.py

@@ -80,7 +80,7 @@ def get_mcp_external_servers() -> List[McpExternalServer]:
     if os.path.exists(cache_file):
         cache_time = os.path.getmtime(cache_file)
         if time.time() - cache_time < 3600:  # 1 hour cache
-            with open(cache_file, "r") as f:
+            with open(cache_file, "r",encoding="utf-8") as f:
                 raw_data = json.load(f)
                 return [McpExternalServer(**item) for item in raw_data]
 
@@ -91,7 +91,7 @@ def get_mcp_external_servers() -> List[McpExternalServer]:
         response = requests.get(url)
         if response.status_code == 200:
             raw_data = response.json()
-            with open(cache_file, "w") as f:
+            with open(cache_file, "w",encoding="utf-8") as f:
                 json.dump(raw_data, f)
             return [McpExternalServer(**item) for item in raw_data]
         return []