Class Codes

applications/chatdev/chatdev/codes.py:8–185

Manages code files, including extraction, updates, rewriting, and version control.

class Codes:
    """
    Manages code files, including extraction, updates, rewriting, and version control.
    """

    def __init__(self, generated_content: list | dict | None = None):
        """
        Initializes the Codes class and stores parsed generated code content in codebooks.
        Args:
            generated_content: Generated code content (in string format).
        """
        self.directory: str = None  # Project directory path
        self.version: float = 0.0  # Code version
        self.generated_content: str = generated_content  # Generated code content
        self.codebooks = {}  # Stores filename and code content pairs
        
        def extract_filename_from_code(code):
            """
            Extracts filename from code content.
            Args:
                code: Code content
            Returns:
                file_name: Extracted filename
            """
            file_name = ""
            regex_extract = r"class (\S+?):\n"
            matches_extract = re.finditer(regex_extract, code, re.DOTALL)
            for match_extract in matches_extract:
                file_name = match_extract.group(1)
            file_name = file_name.lower().split("(")[0] + ".py"
            return file_name

        def process_code_item(item_dict):
            """
            Processes a single code dictionary item, extracting filename and code, and storing them in codebooks.
            Args:
                item_dict: Dictionary containing filename and code
            """
            filename = item_dict.get("filename", "").lower()
            code = item_dict.get("code", "")
            # If code contains __main__, set filename as main.py
            if "__main__" in code:
                filename = "main.py"
            # If filename is not provided, try extracting it from code
            if filename == "":
                filename = extract_filename_from_code(code)
            assert filename != "", "Filename extraction failed"
            if filename and code:
                self.codebooks[filename] = self._format_code(code)

        # Parse and store the generated code if content is provided
        if generated_content:
            if isinstance(generated_content, list):
                for item in generated_content:
                    process_code_item(item)
            elif isinstance(generated_content, dict):
                process_code_item(generated_content)