38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
# -*- coding: utf-8 -*-
|
||
import importlib
|
||
import pkgutil
|
||
|
||
|
||
class ModuleHelper:
|
||
"""
|
||
模块动态加载
|
||
"""
|
||
|
||
@classmethod
|
||
def load(cls, package_path, filter_func=lambda name, obj: True):
|
||
"""
|
||
导入子模块
|
||
:param package_path: 父包名
|
||
:param filter_func: 子模块过滤函数，入参为模块名和模块对象，返回True则导入，否则不导入
|
||
:return:
|
||
"""
|
||
|
||
submodules: list = []
|
||
packages = importlib.import_module(package_path)
|
||
for importer, package_name, _ in pkgutil.iter_modules(packages.__path__):
|
||
try:
|
||
if package_name.startswith('_'):
|
||
continue
|
||
full_package_name = f'{package_path}.{package_name}'
|
||
module = importlib.import_module(full_package_name)
|
||
importlib.reload(module)
|
||
for name, obj in module.__dict__.items():
|
||
if name.startswith('_'):
|
||
continue
|
||
if isinstance(obj, type) and filter_func(name, obj):
|
||
submodules.append(obj)
|
||
except Exception as err:
|
||
print(f'加载模块 {package_name} 失败：{err}')
|
||
|
||
return submodules
|