虽然在 Python 中将“子”模块导入另一个模块并列出其属性相当简单,但当您想要导入所有子模块时会变得稍微困难一些。
我正在为现有的 3D 应用程序构建一个工具库。每个工具都有自己的菜单项和子菜单。我希望该工具负责创建自己的菜单,因为其中许多菜单会根据上下文和模板而变化。我希望我的基本模块能够找到所有子模块并检查一个create_menu()
函数并在找到它时调用它。
发现所有子模块的最简单方法是什么?
虽然在 Python 中将“子”模块导入另一个模块并列出其属性相当简单,但当您想要导入所有子模块时会变得稍微困难一些。
我正在为现有的 3D 应用程序构建一个工具库。每个工具都有自己的菜单项和子菜单。我希望该工具负责创建自己的菜单,因为其中许多菜单会根据上下文和模板而变化。我希望我的基本模块能够找到所有子模块并检查一个create_menu()
函数并在找到它时调用它。
发现所有子模块的最简单方法是什么?
我认为做这种插件的最好方法是使用entry_points
API来查询它们。
当我还是一个刚开始用 Python 编程的人时,我已经为我的模块化 IRC 机器人编写了这个:
# Load plugins
_plugins = []
def ifName(name):
try:
return re.match('([^_.].+)\.[^.]+', a).group(1)
except:
return None
def isValidPlugin(obj):
from common.base import PluginBase
try:
if obj.__base__ == PluginBase:
return True
else:
return False
except:
return False
plugin_names = set(ifilter(lambda a: a!=None, [ifName(a) for a in os.listdir(os.path.join(os.getcwd(), 'plugins'))]))
for plugin_name in plugin_names:
try:
plugin = __import__('plugins.'+plugin_name, fromlist=['plugins'])
valid_plugins = filter(lambda a: isValidPlugin(a), [plugin.__getattribute__(a) for a in dir(plugin)])
_plugins.extend(valid_plugins)
except Exception, e:
logger.exception('Error loading plugin %s', plugin_name)
# Run plugins
_plugins = [klass() for klass in _plugins]
这不是安全或“正确”的方式,但也许它仍然有用。这是非常古老的代码,所以请不要打败我。
只要您将每个插件实现为基于文件系统的模块,上述遍历文件系统以查找子模块的解决方案就可以了。
一种更灵活的方法是在主模块中显式地列出插件列表,并让每个插件(无论是由文件创建的模块、动态创建的模块,还是类的实例)都显式地将自身添加到该列表中。也许通过 registerPlugin 函数。
请记住:“显式胜于隐式”是 python 禅宗的一部分。
pkgutil.walk_packages()
似乎是这样做的正确方法。以下是我如何找到可用模块列表,然后导入其中一个:
$ mkdir foo
$ touch foo/__init__.py
$ touch foo/bar.py
$ python
Python 3.8.2 (default, Jul 16 2020, 14:00:26)
[GCC 9.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import foo
>>> from pkgutil import walk_packages
>>> list(walk_packages(foo.__path__))
[ModuleInfo(module_finder=FileFinder('/home/don/git/zero-play/zero_play/foo'), name='bar', ispkg=False)]
>>> from importlib import import_module
>>> bar = import_module('foo.bar')
您可以尝试glob
bing 目录:
import os
import glob
modules = glob.glob(os.path.join('/some/path/to/modules', '*.py'))
然后你可以尝试导入它们:
checked_modules
for module in modules:
try:
__import__(module, globals(), locals()) # returns module object
except ImportError:
pass
else:
checked_modules.append(module)