group_manage/ruoyi_generator/util.py

import os
from typing import List
from jinja2 import Template
from ruoyi_common.utils import StringUtil
from ruoyi_generator.domain.entity import GenTable
from ruoyi_generator.config import GeneratorConfig
import zipfile
from io import BytesIO
from datetime import datetime
import re


def to_underscore(name: str) -> str:
    """
    将驼峰命名转换为下划线命名
    
    Args:
        name (str): 驼峰命名的字符串
        
    Returns:
        str: 下划线命名的字符串
    """
    # 在大写字母前添加下划线，然后转为小写
    s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', name)
    return re.sub('([a-z0-9])([A-Z])', r'\1_\2', s1).lower()


class GenUtils:
    @staticmethod
    def get_file_name(template_file: str, table: GenTable) -> str:
        """
        根据模板文件名和表信息生成文件名
        
        Args:
            template_file (str): 模板文件名
            table (GenTable): 表信息
            
        Returns:
            str: 生成的文件名
        """
        # 标准化路径分隔符
        template_file = template_file.replace('\\', '/')
        
        # 移除.vm后缀
        base_name = template_file[:-3] if template_file.endswith('.vm') else template_file
        
        # 确定模块路径：后端代码使用 pythonModelName，前端代码使用 modelName
        if table.package_name:
            # 使用 package_name 作为路径（如 com.yy.test -> com/yy/test）
            module_path = table.package_name.replace('.', '/')
        else:
            # 如果 package_name 为空，使用 pythonModelName 作为后端模块名
            # 注意：这里只用于后端 Python 代码路径，前端代码路径在下面单独处理
            module_path = GeneratorConfig.python_model_name if hasattr(GeneratorConfig, 'python_model_name') else (table.module_name if table.module_name else 'ruoyi_generator')
        
        # 根据模板类型生成文件名和路径
        if 'py/entity.py' in template_file:
            # Entity文件放在 domain/entity/ 目录下，使用下划线命名法
            entity_name = to_underscore(table.class_name)
            return f"{module_path}/domain/entity/{entity_name}.py"
        elif 'py/controller.py' in template_file:
            # 使用下划线命名法
            controller_name = f"{to_underscore(table.class_name)}_controller"
            return f"{module_path}/controller/{controller_name}.py"
        elif 'py/service.py' in template_file:
            # 使用下划线命名法
            service_name = f"{to_underscore(table.class_name)}_service"
            return f"{module_path}/service/{service_name}.py"
        elif 'py/mapper.py' in template_file:
            # 使用下划线命名法
            mapper_name = f"{to_underscore(table.class_name)}_mapper"
            return f"{module_path}/mapper/{mapper_name}.py"
        elif 'py/po.py' in template_file:
            # PO文件放在 domain/po/ 目录下，使用下划线命名法
            po_name = f"{to_underscore(table.class_name)}_po"
            return f"{module_path}/domain/po/{po_name}.py"
        elif 'vue/index.vue' in template_file or 'vue/index-tree.vue' in template_file:
            # 无论是树表还是普通表，Vue文件名都是index.vue
            # 使用数据库中的 module_name（前端模块名）
            frontend_module = table.module_name if table.module_name else GeneratorConfig.model_name
            return f"vue/views/{frontend_module}/{table.business_name}/index.vue"
        elif 'js/api.js' in template_file:
            # 使用数据库中的 module_name（前端模块名）
            frontend_module = table.module_name if table.module_name else GeneratorConfig.model_name
            return f"vue/api/{frontend_module}/{table.business_name}.js"
        elif 'py/__init__.py' in template_file:
            # 模块根目录的 __init__.py
            return f"{module_path}/__init__.py"
        elif 'sql/menu.sql' in template_file:
            return f"sql/{table.business_name}_menu.sql"
        elif 'README.md' in template_file:
            return f"{table.business_name}_README.md"
        else:
            # 处理其他模板文件，保持原有目录结构
            filename = os.path.basename(base_name)
            if '.' not in filename:
                filename += '.py'  # 默认添加.py扩展名
            return filename

    @staticmethod
    def get_table_prefix() -> str:
        """
        获取表前缀
        
        Returns:
            str: 表前缀
        """
        return GeneratorConfig.table_prefix or ""

    @staticmethod
    def remove_table_prefix(table_name: str) -> str:
        """
        移除表前缀
        
        Args:
            table_name (str): 表名
            
        Returns:
            str: 移除前缀后的表名
        """
        prefix = GenUtils.get_table_prefix()
        if prefix and table_name.startswith(prefix):
            return table_name[len(prefix):]
        return table_name

    @staticmethod
    def table_to_class_name(table_name: str) -> str:
        """
        将表名转换为类名
        
        Args:
            table_name (str): 表名
            
        Returns:
            str: 类名
        """
        # 移除表前缀
        clean_table_name = GenUtils.remove_table_prefix(table_name)
        # 转换为驼峰命名
        return GenUtils.to_camel_case(clean_table_name)

    @staticmethod
    def get_business_name(table_name: str) -> str:
        """
        获取业务名
        
        Args:
            table_name (str): 表名
            
        Returns:
            str: 业务名
        """
        # 移除表前缀
        clean_table_name = GenUtils.remove_table_prefix(table_name)
        # 获取下划线分隔的第一部分
        return GenUtils.substring_before(clean_table_name, "_") if "_" in clean_table_name else clean_table_name

    @staticmethod
    def get_import_path(package_name: str, module_name: str, module_type: str, class_name: str = None) -> str:
        """
        生成导入路径
        
        Args:
            package_name (str): 包名，如 "com.yy.project" 或空字符串
            module_name (str): 前端模块名（从数据库读取，用于前端代码），但这里应该传入 pythonModelName
            module_type (str): 模块类型，如 "domain", "service", "mapper", "controller"
            class_name (str): 类名（可选，用于PO导入）
            
        Returns:
            str: 导入路径，Python包名保持点分隔格式
        """
        # Python 后端代码使用 pythonModelName，而不是前端模块名
        # 如果 package_name 为空，使用 pythonModelName 作为 Python 模块名
        if not package_name:
            # 使用 pythonModelName 作为 Python 模块名
            python_package = GeneratorConfig.python_model_name if hasattr(GeneratorConfig, 'python_model_name') else (module_name if module_name else 'ruoyi_generator')
        else:
            # Python导入路径使用点分隔，保持原样
            # 例如: "com.yy.project" -> "com.yy.project"
            python_package = package_name
        
        # 生成导入路径
        if module_type == "domain" and class_name:
            # PO 文件在 domain/po/ 目录下
            return f"{python_package}.domain.po"
        elif module_type == "domain":
            return f"{python_package}.domain.entity"
        else:
            return f"{python_package}.{module_type}"

    @staticmethod
    def to_camel_case(name: str) -> str:
        """
        将下划线命名转换为驼峰命名
        
        Args:
            name (str): 下划线命名
            
        Returns:
            str: 驼峰命名
        """
        if hasattr(StringUtil, 'to_camel_case'):
            return StringUtil.to_camel_case(name)
        # 如果StringUtil没有to_camel_case方法，则手动实现
        parts = name.split('_')
        if len(parts) == 1:
            return parts[0]
        return parts[0] + ''.join(word.capitalize() for word in parts[1:])

    @staticmethod
    def substring_before(string: str, separator: str) -> str:
        """
        获取字符串中分隔符之前的部分
        
        Args:
            string (str): 输入字符串
            separator (str): 分隔符
            
        Returns:
            str: 分隔符之前的部分
        """
        if hasattr(StringUtil, 'substring_before'):
            return StringUtil.substring_before(string, separator)
        # 如果StringUtil没有substring_before方法，则手动实现
        if separator in string:
            return string.split(separator, 1)[0]
        return string

    @staticmethod
    def substring_after(string: str, separator: str) -> str:
        """
        获取字符串中分隔符之后的部分
        
        Args:
            string (str): 输入字符串
            separator (str): 分隔符
            
        Returns:
            str: 分隔符之后的部分
        """
        if hasattr(StringUtil, 'substring_after'):
            return StringUtil.substring_after(string, separator)
        # 如果StringUtil没有substring_after方法，则手动实现
        if separator in string:
            return string.split(separator, 1)[1]
        return ""
        
    @staticmethod
    def generator_code(table: GenTable) -> BytesIO:
        """
        生成代码
        
        Args:
            table (GenTable): 表信息
            
        Returns:
            BytesIO: 生成的代码文件
        """
        # 设置列的 list_index 属性
        GenUtils.set_column_list_index(table)
        
        # 设置主键列
        pk_columns = [column for column in table.columns if column.is_pk == '1']
        if pk_columns:
            table.pk_column = pk_columns[0]
        else:
            table.pk_column = None
        
        # 从 options 中解析 parentMenuId
        if table.options:
            import json
            try:
                if isinstance(table.options, str):
                    options_dict = json.loads(table.options)
                else:
                    options_dict = table.options
                # 从 options 中提取 parentMenuId 并设置到 table
                if 'parentMenuId' in options_dict:
                    table.parent_menu_id = options_dict.get('parentMenuId')
            except Exception as e:
                print(f"解析 options 字段出错: {e}")
        
        # 强制使用前端模块名（modelName），而不是 Python 模块名
        # module_name 必须使用 modelName（test），不能使用 pythonModelName（ruoyi_test）
        # 如果 module_name 是空的、等于 python_model_name 或包含 python_model_name，强制替换为 model_name
        original_module_name = table.module_name
        if not table.module_name or table.module_name == GeneratorConfig.python_model_name or (table.module_name and GeneratorConfig.python_model_name in table.module_name):
            table.module_name = GeneratorConfig.model_name
            if original_module_name != table.module_name:
                print(f"警告：table.module_name 从 '{original_module_name}' 强制替换为 '{table.module_name}'（前端模块名）")
        
        # 获取模板目录
        template_dir = os.path.join(os.path.dirname(__file__), 'vm')
        
        # 定义核心模板文件
        core_templates = [
            'py/entity.py.vm',
            'py/po.py.vm', 
            'py/controller.py.vm',
            'py/service.py.vm',
            'py/mapper.py.vm',
            'js/api.js.vm',
            'sql/menu.sql.vm'
        ]
        
        # 根据表类型添加相应的Vue模板
        if table.tpl_category == 'tree':
            core_templates.append('vue/index-tree.vue.vm')
        else:
            core_templates.append('vue/index.vue.vm')
        
        # 创建内存中的ZIP文件
        zip_buffer = BytesIO()
        
        # 确定模块路径，用于生成 __init__.py
        # 后端 Python 代码使用 pythonModelName，而不是前端模块名
        if table.package_name:
            module_path = table.package_name.replace('.', '/')
        else:
            # 使用 pythonModelName 作为后端模块路径
            module_path = GeneratorConfig.python_model_name if hasattr(GeneratorConfig, 'python_model_name') else (table.module_name if table.module_name else 'ruoyi_generator')
        
        # 收集需要生成 __init__.py 的目录和文件信息
        init_dirs = set()
        # 收集每个目录下的文件，用于生成导入语句
        dir_files = {}  # {dir_path: [file_info]}
        
        with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
            # 跟踪已添加的文件名以避免重复
            added_files = set()
            
            # 首先生成模块根目录的 __init__.py
            init_template_path = os.path.join(template_dir, 'py/__init__.py.vm')
            if os.path.exists(init_template_path):
                try:
                    with open(init_template_path, 'r', encoding='utf-8') as f:
                        template_content = f.read()
                    
                    # 准备模板上下文（单个表时，tables 为 None）
                    context = {
                        'table': table,
                        'tables': None,  # 单个表生成时，tables 为 None
                        'datetime': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
                        'underscore': to_underscore,
                        'get_import_path': GenUtils.get_import_path
                    }
                    
                    # 使用Jinja2渲染模板
                    template = Template(template_content)
                    rendered_content = template.render(**context)
                    
                    # 生成文件名
                    init_file_path = f"{module_path}/__init__.py"
                    
                    # 将渲染后的内容写入ZIP文件
                    if rendered_content.strip():
                        zip_file.writestr(init_file_path, rendered_content)
                        added_files.add(init_file_path)
                except Exception as e:
                    print(f"处理模块 __init__.py 模板时出错: {e}")
            
            # 处理每个核心模板文件
            for relative_path in core_templates:
                template_path = os.path.join(template_dir, relative_path)
                if os.path.exists(template_path):
                    # 读取模板内容
                    try:
                        with open(template_path, 'r', encoding='utf-8') as f:
                            template_content = f.read()
                            
                        # 准备模板上下文
                        # table.module_name 是从数据库读取的前端模块名（真正的模块名，用于权限、前端、SQL）
                        # GeneratorConfig.python_model_name 是 Python 模块名（只用于 Python 后端代码路径）
                        context = {
                            'table': table,
                            'datetime': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
                            'underscore': to_underscore,  # 添加自定义过滤器
                            'get_import_path': GenUtils.get_import_path  # 添加导入路径生成函数
                        }
                        
                        # 使用Jinja2渲染模板
                        template = Template(template_content)
                        rendered_content = template.render(**context)
                        
                        # 如果是 SQL 模板，恢复原始 module_name（虽然已经强制设置了，但为了安全）
                        if 'sql/menu.sql' in relative_path:
                            pass  # 已经强制设置为 model_name，不需要恢复
                        
                        # 生成文件名
                        output_file_name = GenUtils.get_file_name(relative_path, table)
                        
                        # 收集目录路径和文件信息，用于生成 __init__.py
                        # 跳过 sql 和 vue 目录，这些目录不需要 __init__.py
                        dir_path = os.path.dirname(output_file_name)
                        if dir_path and 'sql' not in dir_path and 'vue' not in dir_path:
                            init_dirs.add(dir_path)
                            # 同时收集父目录（但也要跳过 sql 和 vue）
                            parts = dir_path.split('/')
                            for i in range(1, len(parts)):
                                parent_dir = '/'.join(parts[:i])
                                if parent_dir and 'sql' not in parent_dir and 'vue' not in parent_dir:
                                    init_dirs.add(parent_dir)
                            
                            # 收集文件信息用于生成导入语句
                            file_name = os.path.basename(output_file_name)
                            file_base = os.path.splitext(file_name)[0]
                            
                            if dir_path not in dir_files:
                                dir_files[dir_path] = []
                            
                            # 根据文件类型确定导入的类名
                            if '/entity/' in output_file_name and '.py' in output_file_name:
                                # Entity 文件在 domain/entity/ 目录下
                                dir_files[dir_path].append(('entity', table.class_name, table))
                            elif '/po/' in output_file_name and '_po.py' in output_file_name:
                                # PO 文件在 domain/po/ 目录下
                                dir_files[dir_path].append(('po', f"{to_underscore(table.class_name)}_po", table))
                            elif '_service.py' in output_file_name:
                                dir_files[dir_path].append(('service', f"{table.class_name}Service", table))
                            elif '_mapper.py' in output_file_name:
                                dir_files[dir_path].append(('mapper', f"{table.class_name}Mapper", table))
                            elif '_controller.py' in output_file_name:
                                dir_files[dir_path].append(('controller', 'gen', table))
                        
                        # 检查渲染后的内容是否为空
                        if rendered_content.strip():
                            # 将渲染后的内容写入ZIP文件
                            zip_file.writestr(output_file_name, rendered_content)
                        else:
                            print(f"警告: 模板 {relative_path} 渲染后内容为空")
                    except Exception as e:
                        print(f"处理模板 {relative_path} 时出错: {e}")
            
            # 为每个目录生成 __init__.py 文件，使其成为完整的 Python 模块
            for dir_path in sorted(init_dirs):
                # 跳过模块根目录，因为已经在开始时生成
                if dir_path == module_path:
                    continue
                
                # 跳过 sql 和 vue 目录，这些目录不需要 __init__.py
                if 'sql' in dir_path or 'vue' in dir_path or dir_path.startswith('sql/') or dir_path.startswith('vue/'):
                    continue
                
                init_file_path = os.path.join(dir_path, '__init__.py').replace('\\', '/')
                
                # 生成 __init__.py 内容
                init_lines = ["# -*- coding: utf-8 -*-"]
                init_lines.append(f"# @Module: {dir_path}")
                init_lines.append("")
                
                # 特殊处理 controller 目录：参考 ruoyi_generator/controller/__init__.py 的格式
                if 'controller' in dir_path and dir_path.endswith('/controller'):
                    # 在 controller/__init__.py 中为每个 controller 创建蓝图
                    if dir_path in dir_files:
                        # 先导入 Blueprint
                        init_lines.append("from flask import Blueprint")
                        init_lines.append("")
                        
                        # 为每个 controller 创建蓝图
                        for file_type, class_name, table_info in dir_files[dir_path]:
                            if file_type == 'controller':
                                blueprint_name = to_underscore(table_info.class_name)
                                url_prefix = f"/{table_info.module_name}/{table_info.business_name}"
                                init_lines.append(f"{blueprint_name} = Blueprint('{blueprint_name}', __name__, url_prefix='{url_prefix}')")
                        
                        # 导入各个 controller 模块
                        init_lines.append("")
                        init_lines.append("")
                        for file_type, class_name, table_info in dir_files[dir_path]:
                            if file_type == 'controller':
                                controller_module_name = f"{to_underscore(table_info.class_name)}_controller"
                                init_lines.append(f"from . import {controller_module_name}")
                else:
                    # 其他目录正常生成导入语句
                    if dir_path in dir_files:
                        imports = []
                        for file_type, class_name, table_info in dir_files[dir_path]:
                            if file_type == 'entity':
                                # Entity 文件在 domain/entity/ 目录下，导入时使用文件名
                                entity_file_name = to_underscore(class_name)
                                imports.append(f"from .{entity_file_name} import {class_name}")
                            elif file_type == 'po':
                                # PO 文件在 domain/po/ 目录下，导入时使用文件名
                                po_file_name = class_name  # class_name 已经是 address_info_po
                                imports.append(f"from .{po_file_name} import {class_name}")
                            elif file_type == 'service':
                                imports.append(f"from .{to_underscore(class_name.replace('Service', ''))}_service import {class_name}")
                            elif file_type == 'mapper':
                                imports.append(f"from .{to_underscore(class_name.replace('Mapper', ''))}_mapper import {class_name}")
                        
                        if imports:
                            init_lines.extend(sorted(set(imports)))
                
                init_content = "\n".join(init_lines) + "\n"
                zip_file.writestr(init_file_path, init_content)
        
        zip_buffer.seek(0)
        return zip_buffer
        
    @staticmethod
    def batch_generator_code(tables: List[GenTable]) -> BytesIO:
        """
        批量生成代码
        
        Args:
            tables (List[GenTable]): 表列表
            
        Returns:
            BytesIO: 生成的代码文件
        """
        # 为每个表设置列的 list_index 属性和主键列
        for table in tables:
            GenUtils.set_column_list_index(table)
            # 设置主键列
            pk_columns = [column for column in table.columns if column.is_pk == '1']
            if pk_columns:
                table.pk_column = pk_columns[0]
            else:
                table.pk_column = None
            
            # 从 options 中解析 parentMenuId
            if table.options:
                import json
                try:
                    if isinstance(table.options, str):
                        options_dict = json.loads(table.options)
                    else:
                        options_dict = table.options
                    # 从 options 中提取 parentMenuId 并设置到 table
                    if 'parentMenuId' in options_dict:
                        table.parent_menu_id = options_dict.get('parentMenuId')
                except Exception as e:
                    print(f"解析 options 字段出错: {e}")
            
            # 强制使用前端模块名（modelName），而不是 Python 模块名
            # module_name 必须使用 modelName（test），不能使用 pythonModelName（ruoyi_test）
            if not table.module_name or table.module_name == GeneratorConfig.python_model_name or table.module_name.strip() == GeneratorConfig.python_model_name:
                table.module_name = GeneratorConfig.model_name
            # 额外检查：如果 module_name 等于 python_model_name，强制替换
            if table.module_name == GeneratorConfig.python_model_name:
                table.module_name = GeneratorConfig.model_name
        
        # 定义核心模板文件（每个表都会生成，但 __init__.py 只生成一次）
        core_templates = [
            'py/entity.py.vm',
            'py/po.py.vm', 
            'py/controller.py.vm',
            'py/service.py.vm',
            'py/mapper.py.vm',
            'js/api.js.vm',
            'sql/menu.sql.vm'
        ]
        
        # 创建内存中的ZIP文件
        zip_buffer = BytesIO()
        
        # 获取模板目录
        template_dir = os.path.join(os.path.dirname(__file__), 'vm')
        
        # 确定模块路径（使用第一个表的模块路径）
        if tables and len(tables) > 0:
            first_table = tables[0]
            if first_table.package_name:
                module_path = first_table.package_name.replace('.', '/')
            else:
                # 使用 pythonModelName 作为后端模块路径
                module_path = GeneratorConfig.python_model_name if hasattr(GeneratorConfig, 'python_model_name') else (first_table.module_name if first_table.module_name else 'ruoyi_generator')
        else:
            module_path = GeneratorConfig.python_model_name if hasattr(GeneratorConfig, 'python_model_name') else 'ruoyi_generator'
        
        # 收集需要生成 __init__.py 的目录和文件信息
        init_dirs = set()
        # 收集每个目录下的文件，用于生成导入语句 {dir_path: [(file_type, class_name, table)]}
        dir_files = {}
        
        with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
            # 跟踪已添加的文件名以避免重复
            added_files = set()
            
            # 首先为模块根目录生成 __init__.py（只生成一次，使用第一个表的信息）
            if tables and len(tables) > 0:
                init_template_path = os.path.join(template_dir, 'py/__init__.py.vm')
                if os.path.exists(init_template_path):
                    try:
                        with open(init_template_path, 'r', encoding='utf-8') as f:
                            template_content = f.read()
                        
                        # 准备模板上下文（批量生成时，传入所有表）
                        # table.module_name 是从数据库读取的前端模块名（真正的模块名，用于权限、前端、SQL）
                        # GeneratorConfig.python_model_name 是 Python 模块名（只用于 Python 后端代码路径）
                        context = {
                            'table': first_table,  # 用于兼容模板中的 table 变量
                            'tables': tables,  # 传入所有表，用于循环注册所有蓝图
                            'datetime': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
                            'underscore': to_underscore,
                            'get_import_path': GenUtils.get_import_path
                        }
                        
                        # 使用Jinja2渲染模板
                        template = Template(template_content)
                        rendered_content = template.render(**context)
                        
                        # 生成文件名
                        init_file_path = f"{module_path}/__init__.py"
                        
                        # 将渲染后的内容写入ZIP文件（只写入一次）
                        if rendered_content.strip() and init_file_path not in added_files:
                            zip_file.writestr(init_file_path, rendered_content)
                            added_files.add(init_file_path)
                    except Exception as e:
                        print(f"处理模块 __init__.py 模板时出错: {e}")
            
            # 处理每个表
            for table in tables:
                # 根据表类型添加相应的Vue模板
                if table.tpl_category == 'tree':
                    current_templates = core_templates + ['vue/index-tree.vue.vm']
                else:
                    current_templates = core_templates + ['vue/index.vue.vm']
                
                # 处理每个核心模板文件
                for relative_path in current_templates:
                    # 跳过模块根目录的 __init__.py，因为已经在开始时生成
                    if relative_path == 'py/__init__.py.vm':
                        continue
                    
                    template_path = os.path.join(template_dir, relative_path)
                    if os.path.exists(template_path):
                        # 读取模板内容
                        try:
                            with open(template_path, 'r', encoding='utf-8') as f:
                                template_content = f.read()
                                
                            # 准备模板上下文
                            # table.module_name 是从数据库读取的前端模块名（真正的模块名，用于权限、前端、SQL）
                            # GeneratorConfig.python_model_name 是 Python 模块名（只用于 Python 后端代码路径）
                            context = {
                                'table': table,
                                'datetime': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
                                'underscore': to_underscore,  # 添加自定义过滤器
                                'get_import_path': GenUtils.get_import_path  # 添加导入路径生成函数
                            }
                            
                            # 使用Jinja2渲染模板
                            template = Template(template_content)
                            rendered_content = template.render(**context)
                            
                            # 生成文件名
                            output_file_name = GenUtils.get_file_name(relative_path, table)
                            
                            # 收集目录路径和文件信息，用于生成 __init__.py
                            # 跳过 sql 和 vue 目录，这些目录不需要 __init__.py
                            dir_path = os.path.dirname(output_file_name)
                            if dir_path and 'sql' not in dir_path and 'vue' not in dir_path:
                                init_dirs.add(dir_path)
                                # 同时收集父目录（但也要跳过 sql 和 vue）
                                parts = dir_path.split('/')
                                for i in range(1, len(parts)):
                                    parent_dir = '/'.join(parts[:i])
                                    if parent_dir and 'sql' not in parent_dir and 'vue' not in parent_dir:
                                        init_dirs.add(parent_dir)
                                
                                # 收集文件信息用于生成导入语句
                                if dir_path not in dir_files:
                                    dir_files[dir_path] = []
                                
                                # 根据文件类型确定导入的类名
                                if '/entity/' in output_file_name and '.py' in output_file_name:
                                    # Entity 文件在 domain/entity/ 目录下
                                    dir_files[dir_path].append(('entity', table.class_name, table))
                                elif '/po/' in output_file_name and '_po.py' in output_file_name:
                                    # PO 文件在 domain/po/ 目录下
                                    dir_files[dir_path].append(('po', f"{to_underscore(table.class_name)}_po", table))
                                elif '_service.py' in output_file_name:
                                    dir_files[dir_path].append(('service', f"{table.class_name}Service", table))
                                elif '_mapper.py' in output_file_name:
                                    dir_files[dir_path].append(('mapper', f"{table.class_name}Mapper", table))
                                elif '_controller.py' in output_file_name:
                                    dir_files[dir_path].append(('controller', 'gen', table))
                            
                            # 检查是否已添加同名文件
                            if output_file_name in added_files:
                                # 为重复文件添加序号
                                name, ext = os.path.splitext(output_file_name)
                                counter = 1
                                new_name = f"{name}_{counter}{ext}"
                                while new_name in added_files:
                                    counter += 1
                                    new_name = f"{name}_{counter}{ext}"
                                output_file_name = new_name
                            
                            # 检查渲染后的内容是否为空
                            if rendered_content.strip():
                                # 将渲染后的内容写入ZIP文件
                                zip_file.writestr(output_file_name, rendered_content)
                                added_files.add(output_file_name)
                            else:
                                print(f"警告: 模板 {relative_path} 渲染后内容为空")
                        except Exception as e:
                            print(f"处理表 {table.table_name} 的模板 {relative_path} 时出错: {e}")
            
            # 为每个目录生成 __init__.py 文件，使其成为完整的 Python 模块
            for dir_path in sorted(init_dirs):
                # 跳过模块根目录，因为已经在开始时生成
                if dir_path == module_path:
                    continue
                
                # 跳过 sql 和 vue 目录，这些目录不需要 __init__.py
                if 'sql' in dir_path or 'vue' in dir_path or dir_path.startswith('sql/') or dir_path.startswith('vue/'):
                    continue
                
                init_file_path = os.path.join(dir_path, '__init__.py').replace('\\', '/')
                
                # 生成 __init__.py 内容
                init_lines = ["# -*- coding: utf-8 -*-"]
                init_lines.append(f"# @Module: {dir_path}")
                init_lines.append("")
                
                # 特殊处理 controller 目录：参考 ruoyi_generator/controller/__init__.py 的格式
                if 'controller' in dir_path and dir_path.endswith('/controller'):
                    # 在 controller/__init__.py 中为每个 controller 创建蓝图
                    if dir_path in dir_files:
                        # 先导入 Blueprint
                        init_lines.append("from flask import Blueprint")
                        init_lines.append("")
                        
                        # 为每个 controller 创建蓝图
                        for file_type, class_name, table_info in dir_files[dir_path]:
                            if file_type == 'controller':
                                blueprint_name = to_underscore(table_info.class_name)
                                url_prefix = f"/{table_info.module_name}/{table_info.business_name}"
                                init_lines.append(f"{blueprint_name} = Blueprint('{blueprint_name}', __name__, url_prefix='{url_prefix}')")
                        
                        # 导入各个 controller 模块
                        init_lines.append("")
                        init_lines.append("")
                        for file_type, class_name, table_info in dir_files[dir_path]:
                            if file_type == 'controller':
                                controller_module_name = f"{to_underscore(table_info.class_name)}_controller"
                                init_lines.append(f"from . import {controller_module_name}")
                else:
                    # 其他目录正常生成导入语句
                    if dir_path in dir_files:
                        imports = []
                        for file_type, class_name, table_info in dir_files[dir_path]:
                            if file_type == 'entity':
                                # Entity 文件在 domain/entity/ 目录下，导入时使用文件名
                                entity_file_name = to_underscore(class_name)
                                imports.append(f"from .{entity_file_name} import {class_name}")
                            elif file_type == 'po':
                                # PO 文件在 domain/po/ 目录下，导入时使用文件名
                                po_file_name = class_name  # class_name 已经是 address_info_po
                                imports.append(f"from .{po_file_name} import {class_name}")
                            elif file_type == 'service':
                                imports.append(f"from .{to_underscore(class_name.replace('Service', ''))}_service import {class_name}")
                            elif file_type == 'mapper':
                                imports.append(f"from .{to_underscore(class_name.replace('Mapper', ''))}_mapper import {class_name}")
                        
                        if imports:
                            init_lines.extend(sorted(set(imports)))
                
                init_content = "\n".join(init_lines) + "\n"
                zip_file.writestr(init_file_path, init_content)
        
        zip_buffer.seek(0)
        return zip_buffer
        
    @staticmethod
    def set_column_list_index(table: GenTable):
        """
        为表的列设置 list_index 属性，用于 Vue 模板中的 columns 数组索引
        
        Args:
            table (GenTable): 表信息
        """
        if not table.columns:
            return
        
        list_index = 0
        for column in table.columns:
            if column.is_list == '1':
                # 使用 setattr 动态添加属性
                setattr(column, 'list_index', list_index)
                list_index += 1

    @staticmethod
    def preview_code(table: GenTable) -> dict:
        """
        预览代码
        
        Args:
            table (GenTable): 表信息
            
        Returns:
            dict: 预览代码
        """
        # 设置列的 list_index 属性
        GenUtils.set_column_list_index(table)
        
        # 设置主键列
        pk_columns = [column for column in table.columns if column.is_pk == '1']
        if pk_columns:
            table.pk_column = pk_columns[0]
        else:
            table.pk_column = None
        
        # 从 options 中解析 parentMenuId
        if table.options:
            import json
            try:
                if isinstance(table.options, str):
                    options_dict = json.loads(table.options)
                else:
                    options_dict = table.options
                # 从 options 中提取 parentMenuId 并设置到 table
                if 'parentMenuId' in options_dict:
                    table.parent_menu_id = options_dict.get('parentMenuId')
            except Exception as e:
                print(f"解析 options 字段出错: {e}")
        
        # 强制使用前端模块名（modelName），而不是 Python 模块名
        # module_name 必须使用 modelName（test），不能使用 pythonModelName（ruoyi_test）
        # 如果 module_name 是空的、等于 python_model_name 或包含 python_model_name，强制替换为 model_name
        original_module_name = table.module_name
        if not table.module_name or table.module_name == GeneratorConfig.python_model_name or (table.module_name and GeneratorConfig.python_model_name in table.module_name):
            table.module_name = GeneratorConfig.model_name
            if original_module_name != table.module_name:
                print(f"警告：table.module_name 从 '{original_module_name}' 强制替换为 '{table.module_name}'（前端模块名）")
        
        # 获取模板目录
        template_dir = os.path.join(os.path.dirname(__file__), 'vm')
        
        # 存储预览代码的字典
        preview_data = {}
        
        # 定义需要预览的核心模板文件
        core_templates = [
            'py/entity.py.vm',
            'py/po.py.vm', 
            'py/controller.py.vm',
            'py/service.py.vm',
            'py/mapper.py.vm',
            'js/api.js.vm',
            'sql/menu.sql.vm'
        ]
        
        # 根据表类型添加相应的Vue模板，但预览时都使用index.vue.vm作为文件名
        if table.tpl_category == 'tree':
            core_templates.append('vue/index-tree.vue.vm')
        else:
            core_templates.append('vue/index.vue.vm')
        
        # 处理每个核心模板文件
        for relative_path in core_templates:
            template_path = os.path.join(template_dir, relative_path)
            if os.path.exists(template_path):
                # 读取模板内容
                try:
                    with open(template_path, 'r', encoding='utf-8') as f:
                        template_content = f.read()
                        
                    # 准备模板上下文
                    # table.module_name 是从数据库读取的前端模块名（真正的模块名，用于权限、前端、SQL）
                    # GeneratorConfig.python_model_name 是 Python 模块名（只用于 Python 后端代码路径）
                    context = {
                        'table': table,
                        'datetime': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
                        'underscore': to_underscore,  # 添加自定义过滤器
                        'get_import_path': GenUtils.get_import_path  # 添加导入路径生成函数
                    }
                    
                    # 使用Jinja2渲染模板
                    template = Template(template_content)
                    rendered_content = template.render(**context)
                    
                    # 存储渲染后的内容
                    preview_data[relative_path] = rendered_content
                except Exception as e:
                    # 如果渲染失败，存储错误信息
                    preview_data[relative_path] = f"模板渲染失败: {str(e)}"
            else:
                preview_data[relative_path] = "模板文件不存在"
        
        return preview_data