diff options
Diffstat (limited to 'portato/plugin.py')
| -rw-r--r-- | portato/plugin.py | 60 |
1 files changed, 41 insertions, 19 deletions
diff --git a/portato/plugin.py b/portato/plugin.py index c823ac7..e627eb1 100644 --- a/portato/plugin.py +++ b/portato/plugin.py @@ -189,7 +189,6 @@ class Plugin (object): :param disable: Forcefully disable the plugin :type disable: bool """ - self.__widgets = [] #: List of `Widget` self.__calls = [] #: List of `Call` self._unresolved_deps = False #: Does this plugin has unresolved dependencies? @@ -258,15 +257,6 @@ class Plugin (object): return getattr(self, "__name__", self.__class__.__name__) @property - def widgets (self): - """ - Returns an iterator over the widgets for this plugin. - - :rtype: iter<`Widget`> - """ - return iter(self.__widgets) - - @property def calls (self): """ Returns an iterator over the registered calls for this plugin. @@ -298,6 +288,27 @@ class Plugin (object): """ return (self.status in (self.STAT_ENABLED, self.STAT_TEMP_ENABLED)) + def add_call (self, hook, callable, type = "before", dep = None): + """ + Adds a new call for this plugin. + + :see: `Call` + """ + self.__calls.append(Call(self, hook, callable, type, dep)) + +class WidgetPlugin (Plugin): + + def __init__ (self, *args, **kwargs): + Plugin.__init__(self, *args, **kwargs) + self.__widgets = [] #: List of `Widget` + + def _widget_init (self): + if self.status == self.STAT_ENABLED and not self._unresolved_deps: + self.widget_init() + + def widget_init (self): + debug("No widgets to initialize. Wrong class used for plugin?") + def add_widget (self, slot, widget): """ Adds a new widget for this plugin. @@ -340,15 +351,15 @@ class Plugin (object): w.connect(k, v) self.add_widget(slot, w) - - def add_call (self, hook, callable, type = "before", dep = None): + + @property + def widgets (self): """ - Adds a new call for this plugin. + Returns an iterator over the widgets for this plugin. - :see: `Call` + :rtype: iter<`Widget`> """ - self.__calls.append(Call(self, hook, callable, type, dep)) - + return iter(self.__widgets) class PluginQueue (object): """ @@ -411,6 +422,7 @@ class PluginQueue (object): plugin_module.__path__.insert(0, PLUGIN_DIR.rstrip("/")) # make the plugins loadable as "portato.plugins.name" # add Plugin and register to the builtins, so the plugins always have the correct version :) plugin_module.__builtins__["Plugin"] = Plugin + plugin_module.__builtins__["WidgetPlugin"] = WidgetPlugin plugin_module.__builtins__["register"] = register for p in plugins: # import them @@ -424,9 +436,12 @@ class PluginQueue (object): self._organize() + def load_widgets(self): for p in self.plugins: - for w in p.widgets: - WidgetSlot.slots[w.slot].add_widget(w) + if isinstance(p, WidgetPlugin): + p._widget_init() + for w in p.widgets: + WidgetSlot.slots[w.slot].add_widget(w) def add (self, plugin, disable = False): """ @@ -443,7 +458,7 @@ class PluginQueue (object): :raise PluginLoadException: passed plugin is not of class `Plugin` """ - if callable(plugin) and Plugin in plugin.__bases__: + if callable(plugin) and issubclass(plugin, Plugin): p = plugin(disable = disable) # need an instance and not the class elif isinstance(plugin, Plugin): p = plugin @@ -617,6 +632,13 @@ def load_plugins(): __plugins.load() +def load_plugin_widgets(): + """ + Loads the widgets of the plugins. + """ + if __plugins is not None: + __plugins.load_widgets() + def get_plugin_queue(): """ Returns the actual `PluginQueue`. If it is ``None``, they are not being loaded yet. |