def parse(self):
args = self.parser.parse_args()
self.session.gui = True
if args.color:
self.session.color_cli = True
os.environ['OA_CLICOLOR'] = '1'
if args.list_plugins:
self.session.gui = False
import pkg_resources
from openalea.core.plugin import iter_groups
if args.list_plugins in ['summary', 'all']:
prefixes = ['oalab', 'vpltk', 'openalea']
else:
prefixes = [args.list_plugins]
for category in sorted(iter_groups()):
match = False
for prefix in prefixes:
if category.startswith(prefix):
match = True
break
if match:
eps = [ep for ep in pkg_resources.iter_entry_points(category)]
if args.list_plugins == 'summary':
print ' - \033[91m%s\033[0m (%d plugins)' % (category, len(eps))
else:
print '\033[91m%s\033[0m' % category
print
for ep in eps:
print_plugin_name(ep)
print
print
if args.debug_plugins:
debug = args.debug_plugins.split(',')
if 'oalab.lab' not in debug:
debug.append('oalab.lab')
from openalea.core.service.plugin import PluginInstanceManager
pim = PluginInstanceManager()
pim.debug = debug
self.session.extension = args.extension
self.args = args
def __init__(self):
import traceback
traceback.print_stack(file=sys.__stdout__)
self._project = None
self._is_proj = False
self._debug = False
self.gui = True
self.tmpdir = path(get_openalea_tmp_dir())
self._config = MainConfig()
self.extension = None
self.applet = {}
self.manager = {}
self.package_manager = package_manager
self.control_manager = ControlManager()
self.project_manager = ProjectManager()
self.plugin_manager = PluginManager()
self.plugin_instance_manager = PluginInstanceManager()
self.manager['control'] = self.control_manager
self.manager['package'] = self.package_manager
self.manager['project'] = self.project_manager
self.manager['plugin'] = self.plugin_manager
self.manager['plugin_instance'] = self.plugin_instance_manager
self.world = World()
self.interpreter = interpreter()
# Hack if interpreter is an object from class TerminalInteractiveShell
if not hasattr(self.interpreter, "shell"):
self.interpreter.shell = self.interpreter
if hasattr(self.interpreter.shell, "events"):
self.interpreter.shell.events.register("post_execute",
self.add_to_history)
else:
print("You need ipython >= 2.0 to use history.")
# self.project_manager.set_shell(self.interpreter.shell)
self.interpreter.locals['session'] = self
self.old_syspath = sys.path
self.load_default()
self.__class__.instantiated = True
def setUp(self):
self.pm = PluginManager()
self.pm.debug = False
self.pim = PluginInstanceManager(self.pm)
class TestPluginManager(unittest.TestCase):
@classmethod
def setupClass(cls):
cls.tmppath, cls.pythonpath = install_package()
import pkg_resources
dist = list(pkg_resources.find_distributions(cls.pythonpath))[0]
pkg_resources.working_set.add(dist)
def setUp(self):
self.pm = PluginManager()
self.pm.debug = False
self.pim = PluginInstanceManager(self.pm)
def test_autoload(self):
pm = PluginManager()
pm.items('test.c1')
assert 'test.c1' in pm._item
assert 'tstpkg1.plugin:C1Plugin1' in pm._item['test.c1']
pm = PluginManager(autoload=[])
pm.items('test.c1')
assert 'test.c1' in pm._item
assert 'tstpkg1.plugin:C1Plugin1' not in pm._item['test.c1']
def test_module_plugin_def(self):
assert self.pm.items('test.c1')
# check if right plugins have been found
assert self.pm.item('MyPlugin1', 'test.c1') is not None
assert self.pm.item('MyPlugin2', 'test.c1') is not None
def test_manual_plugin_def(self):
assert self.pm.items('test.c2')
# check if right plugins have been found
assert self.pm.item('C2Plugin1', 'test.c2') is not None
assert self.pm.item('C2Plugin2', 'test.c2') is not None
def test_interface_filter(self):
plugins = self.pm.items('test.c1')
assert len(plugins) == 2
plugins = self.pm.items('test.c1', criteria=dict(implement='IClass1'))
assert len(plugins) == 1
assert plugins[0].name == 'MyPlugin1'
def test_dynamic_plugin(self):
pl = self.pm.add(AlgoPlugin1, 'test.dynamic3')
self.pm.add(AlgoPlugin2, 'test.dynamic3')
assert len(self.pm.items('test.dynamic3')) == 2
objc3c1 = self.pim.instance('test.dynamic3', 'algo_1')
assert objc3c1
assert isinstance(objc3c1, Algo)
objc3c2 = self.pim.instance('test.dynamic3', 'algo_2')
assert objc3c2
assert isinstance(objc3c1, Algo)
def atest_proxy_plugin(self):
from openalea.core.plugin.manager import SimpleClassPluginProxy
self.pm.add('test.dynamic4', Algo, plugin_proxy=SimpleClassPluginProxy)
objc4c1 = self.pim.instance('test.dynamic4', 'Algo')
objc4c1_2 = self.pim.instance('test.dynamic4', 'Algo')
objc4c1_3 = self.pim.new('test.dynamic4', 'Algo')
assert objc4c1
self.assertIsInstance(objc4c1, Algo)
assert objc4c1 is objc4c1_2
assert objc4c1 is not objc4c1_3
def test_plugin_name(self):
import tstpkg1.plugin
import tstpkg1.impl
# Check manager use plugin name attribute if defined (instead of class name)
c1plugin1 = self.pm.item('MyPlugin1', 'test.c1')
self.assertIsInstance(c1plugin1, tstpkg1.plugin.C1Plugin1)
# Check there is no conflict if two plugins with same alias (in setup.py) but in different categories
# Check plugin name use class name if no attribute "name"
c2plugin1 = self.pm.item('C2Plugin1', 'test.c2')
self.assertIsInstance(c2plugin1, tstpkg1.plugin.C2Plugin1)
def test_new_instance(self):
import tstpkg1.impl
objc1c1 = self.pim.instance('test.c1', 'MyPlugin1')
assert objc1c1
assert isinstance(objc1c1, tstpkg1.impl.C1Class1)
objc1c2 = self.pim.instance('test.c1', 'MyPlugin2')
assert objc1c2
assert isinstance(objc1c2, tstpkg1.impl.C1Class2)
objc2c1 = self.pim.instance('test.c2', 'C2Plugin1')
assert objc2c1
assert isinstance(objc2c1, tstpkg1.impl.C2Class1)
objc2c2 = self.pim.instance('test.c2', 'C2Plugin2')
assert objc2c1
assert isinstance(objc2c2, tstpkg1.impl.C2Class2)
# Check instance return existing instance
objc1c1_2 = self.pim.instance('test.c1', 'MyPlugin1')
assert objc1c1 is objc1c1_2
assert self.pim.instances('test.c1', 'MyPlugin1')[0] is objc1c1_2
def test_multiple_instance(self):
# Assert weakref work correctly and plugin instances have been lost
assert not self.pim.instances('test.c1', 'MyPlugin1')
objc1c1_0 = self.pim.instance('test.c1', 'MyPlugin1')
objc1c1_1 = self.pim.new('test.c1', 'MyPlugin1')
objc1c1_2 = self.pim.new('test.c1', 'MyPlugin1')
# Assert all object have been created correctly
assert objc1c1_0
assert objc1c1_1
assert objc1c1_2
# Assert all instances are different as we use new instead of instance
assert objc1c1_0 is not objc1c1_1
assert objc1c1_1 is not objc1c1_2
objc1c2 = self.pim.new('test.c1', 'MyPlugin2')
assert len(self.pim.instances('test.c1', 'MyPlugin1')) == 3
assert len(self.pim.instances('test.c1')) == 4
del objc1c1_2
assert len(self.pim.instances('test.c1', 'MyPlugin1')) == 2
assert len(self.pim.instances('test.c1')) == 3
del objc1c2
assert len(self.pim.instances('test.c1', 'MyPlugin1')) == 2
assert len(self.pim.instances('test.c1', 'MyPlugin2')) == 0
assert len(self.pim.instances('test.c1')) == 2
def test_debug_ep_load(self):
self.pm.debug = True
with self.assertRaises(ImportError):
self.pm.items('test.err1')
self.pm.clear()
self.pm.debug = False
self.pm.items('test.err1')
@classmethod
def tearDownClass(cls):
cls.tmppath.rmtree()
def setUp(self):
self.pm = PluginManager()
self.pm.debug = False
self.pim = PluginInstanceManager(self.pm)
class TestPluginManager(unittest.TestCase):
@classmethod
def setupClass(cls):
cls.tmppath, cls.pythonpath = install_package()
import pkg_resources
dist = list(pkg_resources.find_distributions(cls.pythonpath))[0]
pkg_resources.working_set.add(dist)
def setUp(self):
self.pm = PluginManager()
self.pm.debug = False
self.pim = PluginInstanceManager(self.pm)
def test_autoload(self):
pm = PluginManager()
pm.items("test.c1")
assert "test.c1" in pm._item
assert "tstpkg1.plugin:C1Plugin1" in pm._item["test.c1"]
pm = PluginManager(autoload=[])
pm.items("test.c1")
assert "test.c1" in pm._item
assert "tstpkg1.plugin:C1Plugin1" not in pm._item["test.c1"]
def test_module_plugin_def(self):
assert self.pm.items("test.c1")
# check if right plugins have been found
assert self.pm.item("MyPlugin1", "test.c1") is not None
assert self.pm.item("MyPlugin2", "test.c1") is not None
def test_manual_plugin_def(self):
assert self.pm.items("test.c2")
# check if right plugins have been found
assert self.pm.item("C2Plugin1", "test.c2") is not None
assert self.pm.item("C2Plugin2", "test.c2") is not None
def test_interface_filter(self):
plugins = self.pm.items("test.c1")
assert len(plugins) == 2
plugins = self.pm.items("test.c1", criteria=dict(implement="IClass1"))
assert len(plugins) == 1
assert plugins[0].name == "MyPlugin1"
def test_dynamic_plugin(self):
pl = self.pm.add(AlgoPlugin1, "test.dynamic3")
self.pm.add(AlgoPlugin2, "test.dynamic3")
assert len(self.pm.items("test.dynamic3")) == 2
objc3c1 = self.pim.instance("test.dynamic3", "algo_1")
assert objc3c1
assert isinstance(objc3c1, Algo)
objc3c2 = self.pim.instance("test.dynamic3", "algo_2")
assert objc3c2
assert isinstance(objc3c1, Algo)
def atest_proxy_plugin(self):
from openalea.core.plugin.manager import SimpleClassPluginProxy
self.pm.add("test.dynamic4", Algo, plugin_proxy=SimpleClassPluginProxy)
objc4c1 = self.pim.instance("test.dynamic4", "Algo")
objc4c1_2 = self.pim.instance("test.dynamic4", "Algo")
objc4c1_3 = self.pim.new("test.dynamic4", "Algo")
assert objc4c1
self.assertIsInstance(objc4c1, Algo)
assert objc4c1 is objc4c1_2
assert objc4c1 is not objc4c1_3
def test_plugin_name(self):
import tstpkg1.plugin
import tstpkg1.impl
# Check manager use plugin name attribute if defined (instead of class name)
c1plugin1 = self.pm.item("MyPlugin1", "test.c1")
self.assertIsInstance(c1plugin1, tstpkg1.plugin.C1Plugin1)
# Check there is no conflict if two plugins with same alias (in setup.py) but in different categories
# Check plugin name use class name if no attribute "name"
c2plugin1 = self.pm.item("C2Plugin1", "test.c2")
self.assertIsInstance(c2plugin1, tstpkg1.plugin.C2Plugin1)
def test_new_instance(self):
import tstpkg1.impl
objc1c1 = self.pim.instance("test.c1", "MyPlugin1")
assert objc1c1
assert isinstance(objc1c1, tstpkg1.impl.C1Class1)
objc1c2 = self.pim.instance("test.c1", "MyPlugin2")
assert objc1c2
assert isinstance(objc1c2, tstpkg1.impl.C1Class2)
objc2c1 = self.pim.instance("test.c2", "C2Plugin1")
assert objc2c1
assert isinstance(objc2c1, tstpkg1.impl.C2Class1)
objc2c2 = self.pim.instance("test.c2", "C2Plugin2")
assert objc2c1
assert isinstance(objc2c2, tstpkg1.impl.C2Class2)
# Check instance return existing instance
objc1c1_2 = self.pim.instance("test.c1", "MyPlugin1")
assert objc1c1 is objc1c1_2
assert self.pim.instances("test.c1", "MyPlugin1")[0] is objc1c1_2
def test_multiple_instance(self):
# Assert weakref work correctly and plugin instances have been lost
assert not self.pim.instances("test.c1", "MyPlugin1")
objc1c1_0 = self.pim.instance("test.c1", "MyPlugin1")
objc1c1_1 = self.pim.new("test.c1", "MyPlugin1")
objc1c1_2 = self.pim.new("test.c1", "MyPlugin1")
# Assert all object have been created correctly
assert objc1c1_0
assert objc1c1_1
assert objc1c1_2
# Assert all instances are different as we use new instead of instance
assert objc1c1_0 is not objc1c1_1
assert objc1c1_1 is not objc1c1_2
objc1c2 = self.pim.new("test.c1", "MyPlugin2")
assert len(self.pim.instances("test.c1", "MyPlugin1")) == 3
assert len(self.pim.instances("test.c1")) == 4
del objc1c1_2
assert len(self.pim.instances("test.c1", "MyPlugin1")) == 2
assert len(self.pim.instances("test.c1")) == 3
del objc1c2
assert len(self.pim.instances("test.c1", "MyPlugin1")) == 2
assert len(self.pim.instances("test.c1", "MyPlugin2")) == 0
assert len(self.pim.instances("test.c1")) == 2
def test_debug_ep_load(self):
self.pm.debug = True
with self.assertRaises(ImportError):
self.pm.items("test.err1")
self.pm.clear()
self.pm.debug = False
self.pm.items("test.err1")
@classmethod
def tearDownClass(cls):
cls.tmppath.rmtree()
