stevedore动态加载模块

时间:2022-04-15 04:59:00

stevedore动态加载模块,stevedore使用setuptools的entry points来定义并加载插件。entry point引用的是定义在模块中的对象,比如类、函数、实例等,只要在import模块时能够被创建的对象都可以,但stevedore的优良之处在于,其可以动态调用所有插件的某一相同方法,将返回结果存入list中。这对于处理具有相同行为的类,起到极大的便利。

如何构建插件?

先定义一个基类,所有继承该基类的子类既为各个独立的插件。

import abc
import six
# 定义基类,只包含format方法,文件路径 plugins/
@six.add_metaclass(abc.ABCMeta)
class FormatterBase(object): @abc.abstractmethod
def format(self, data):
pass

定义一个插件Simple,实现format方法

 from plugins import plugin_base
# 文件路径 plugins/simple.py
class Simple(plugin_base.FormatterBase): data = {'ca':'cangyue', "name":"caesar"} def format(self, data):
lines = []
for name, value in sorted(data.items()):
line = '{name}={value}'.format(
name = name,
value = value
)
lines.append(line)
return ''.join(lines)

编写setup.py文件,用于发布包。 包名demo,版本0.1, 组名stevedore.formatter  两个插件simple和plain都对应Plugins.simple路径中的Simple类。文件路径与plugins同路径,注意plugins中有空__init__.py文件,

表示是一个python包。

 from setuptools import setup, find_packages
setup(
name = "demo",
version = "0.1",
packages = find_packages(),
entry_points={
'stevedore.formatter': [
'simple = plugins.simple:Simple',
'plain = plugins.simple:Simple',
],
},
)

使用命令发布包 python setup.py install 安装,安装后会在egg文件夹下生成entry_points.txt文件,描述组名和插件

[stevedore.formatter]
simple = plugins.simple:Simple
plain = plugins.simple:Simple

stevedore动态加载模块

如何调用插件中方法?

1.加载某一插件,并调用其中方法

 from stevedore import driver
mgr = driver.DriverManager(
namespace='stevedore.formatter',
name='simple',
invoke_on_load=True
) # 加载组名或者说命名空间为stevedore.formetter下name为simple的插件
>>> st = mgr.driver.format({'caesa':'sss'})
>>> data = mgr.driver.data # mgr.driver实例化simple对象并调用其data属性
>>> print data
10 {'ca': 'cangyue', 'name': 'caesar'}

2.加载命名空间下所有插件,并调用所有插件的format方法

 from stevedore import extension
mgr = extension.ExtensionManager(
namespace='stevedore.formatter',
invoke_on_load=True) # 加载stevedore.formatter命名空间下所有插件 def d(ext):
return ext.obj.data >>> dat = mgr.map(d) #对所有插件调用d方法,d方法中ext为插件的extension对象,调用其obj即可实例化该对象
>>> print dat
[{'ca': 'cangyue', 'name': 'caesar'}, {'ca': 'cangyue', 'name': 'caesar'}] def m(ext):
data = ext.obj.data #获取插件对象的data属性值
return ext.obj.format(data) 调用对象的format方法,将属性值传入 >>> ret = mgr.map(m)
>>> print ret
['ca=cangyuename=caesar', 'ca=cangyuename=caesar']

特记:

当想重新开发一个插件时,需要在编写完成后,重新安装,否则不生效。比如鄙人想重新开发一个complex,在相应路径下添加complex.py

stevedore动态加载模块

在entry_point中添加插件

stevedore动态加载模块

但重新执行时,加载的仍然是两个插件

stevedore动态加载模块

参考文献: http://blog.csdn.net/qingyuanluofeng/article/details/72853465