Python 动态加载某模块基础框架

小鱼儿 2022-09-15 11:59 208阅读 0赞

先看目录结构:

Reflect是一个基础框架，里面包含一个apps模块和启动加载文件load\_file.py

apps下包含蓝牙测试和wifi测试模块，他们里面的对象继承自app基类，

![watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBAMTAyNOWwj-elng_size_18_color_FFFFFF_t_70_g_se_x_16][]

load\_file.py 中会自动扫描apps里面的所有py文件，并加载到容器里面：

import os
    import sys
    import inspect
    from apps.app import Person
    
    
    def load_model(file, init_params=None):
        # Ignore anything that isn't a .py file
        module_name = ''
        if len(file) > 3 and (file[-3:] == '.py' or file[-4:] == '.pyc'):
            if file[-3:] == '.py':
                module_name = file[:-3]
            if file[-4:] == '.pyc':
                module_name = file[:-4]
        __import__(module_name)
        module = sys.modules[module_name]
        module_attrs = dir(module)
        app_dict = {}
        # 遍历模块下面所有的属性
        for name in module_attrs:
            var_obj = getattr(module, name)
            # 判断是不是对象
            if inspect.isclass(var_obj):
                if issubclass(var_obj, Person) and var_obj.__name__ != Person.__name__:
                    if app_dict.get(name) is None:
                        if init_params is None:
                            app_dict[name] = var_obj()
                        else:
                            app_dict[name] = var_obj()
                        print("注入 %s 模块 %s 成功" % (Person.__name__, var_obj.__name__))
        return app_dict
    
    
    def recursive_dir(path, f, file_list):
        """
        递归获取文件夹中所有的文件
        :param path:根目录
        :param f:子目录
        :param file_list:文件列表
        :return:
        """
        file_names = os.listdir(os.path.join(path, f))  # 获取当前路径下的文件名，返回List
        for file in file_names:
            newDir = path + '/' + f + '/' + file  # 将文件命加入到当前文件路径后面
            if os.path.isfile(newDir):  # 如果是文件
                if "pycache" not in f and "pycache" not in file:
                    if not f.startswith("apps"):
                        f = f"apps.{f}"
                    file_list.append(f"{f}.{file}")
            else:
                if "__pycache__" not in file:
                    recursive_dir("/".join(newDir.split("/")[0:-1]), newDir.split("/")[-1],
                                  file_list)  # 如果不是文件，递归这个文件夹的路径
    
    
    def search_file():
        app_dicts = {}
        app_files = []
        recursive_dir(os.getcwd(), "apps", app_files)
        print(app_files)
        for app in app_files:
            if app.endswith(".py"):
                app_dicts.update(load_model(app))
        print(f"所有模块已加载完成: {app_dicts}")
    
    
    if __name__ == '__main__':
        search_file()

![watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBAMTAyNOWwj-elng_size_20_color_FFFFFF_t_70_g_se_x_16][]

[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBAMTAyNOWwj-elng_size_18_color_FFFFFF_t_70_g_se_x_16]: /images/20220828/27e0abe4c8ee42d998921574d60f09cc.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBAMTAyNOWwj-elng_size_20_color_FFFFFF_t_70_g_se_x_16]: /images/20220828/abca7981742b463fa03bf9df03a81dc9.png