PyBitmessage/src/plugins/plugin.py

36 lines
1.1 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
import pkg_resources
2017-03-10 15:45:46 +01:00
def get_plugins(group, point='', name=None, fallback=None):
"""
Iterate through plugins (`connect_plugin` attribute of entry point)
which name starts with `point` or equals to `name`.
If `fallback` kwarg specified, plugin with that name yield last.
"""
for ep in pkg_resources.iter_entry_points('bitmessage.' + group):
2017-03-10 15:45:46 +01:00
if name and ep.name == name or ep.name.startswith(point):
try:
2017-03-10 15:45:46 +01:00
plugin = ep.load().connect_plugin
if ep.name == fallback:
_fallback = plugin
else:
yield plugin
except (AttributeError,
2017-03-10 15:45:46 +01:00
ImportError,
ValueError,
pkg_resources.DistributionNotFound,
pkg_resources.UnknownExtra):
continue
2017-03-10 15:45:46 +01:00
try:
yield _fallback
except NameError:
pass
def get_plugin(*args, **kwargs):
"""Returns first available plugin `from get_plugins()` if any."""
for plugin in get_plugins(*args, **kwargs):
return plugin