修改sub_applications/handle.py，实现一个自动扫描子应用目录并动态挂载控制器的机制，避免手动修改。

定义了_discover_controllers函数，它会：
扫描sub_applications目录下的所有子目录
检查每个子目录中是否有controller文件夹
导入每个控制器模块并查找其中的APIRouter实例
自动为每个控制器生成标签名
修改了handle_sub_applications函数，使用自动发现的方式挂载控制器
这样，每次添加新的模块或控制器时，系统会自动扫描并挂载它们，无需手动修改handle.py文件。
要使用此修改，只需替换当前的handle.py文件内容即可。重启应用后，所有子应用的控制器将被自动发现并挂载。

```
import os
import importlib
import inspect
from pathlib import Path
from typing import List, Dict, Any
from fastapi import FastAPI, APIRouter
from sub_applications.staticfiles import mount_staticfiles

def _discover_controllers() -> List[Dict[str, Any]]:
    """
    自动发现子应用中的所有控制器
    返回格式: [{'router': router_instance, 'tags': ['标签名']}]
    """
    controllers = []
    # 获取sub_applications目录的绝对路径
    base_dir = Path(__file__).parent
    
    # 遍历sub_applications下的所有目录
    for item in os.listdir(base_dir):
        # 跳过非目录项和以_开头的目录
        item_path = base_dir / item
        if not item_path.is_dir() or item.startswith('__'):
            continue
        
        # 检查是否存在controller目录
        controller_dir = item_path / 'controller'
        if not controller_dir.is_dir():
            continue
        
        # 遍历controller目录中的所有Python文件
        for controller_file in controller_dir.glob('*.py'):
            if controller_file.name.startswith('__'):
                continue
                
            # 构建模块路径
            module_name = f"sub_applications.{item}.controller.{controller_file.stem}"
            
            try:
                # 导入模块
                module = importlib.import_module(module_name)
                
                # 查找模块中的所有APIRouter实例
                for name, obj in inspect.getmembers(module):
                    if isinstance(obj, APIRouter):
                        # 根据模块名和控制器名生成tag
                        tag_name = f"{item.replace('_', '-')}-{name.replace('Controller', '')}"
                        # 将控制器添加到列表中
                        controllers.append({
                            'router': obj, 
                            'tags': [tag_name]
                        })
            except (ImportError, AttributeError) as e:
                print(f"无法导入模块 {module_name}: {e}")
    
    return controllers

def handle_sub_applications(app: FastAPI):
    """
    全局处理子应用挂载
    """
    # 挂载静态文件
    mount_staticfiles(app)
    
    # 自动发现并挂载所有控制器
    controllers = _discover_controllers()
    for controller in controllers:
        app.include_router(
            router=controller['router'], 
            tags=controller['tags']
        )
        print(f"已挂载控制器: {controller['router'].prefix} -> {controller['tags']}")

```

insistence/RuoYi-Vue3-FastAPI

内容风险标识

评论 (0)

insistence/RuoYi-Vue3-FastAPI .gitee-modal { width: 500px !important; }

内容风险标识