From 771eaf331ef3b66c389976f16fe63a7ab3d53ad7 Mon Sep 17 00:00:00 2001
From: tobi-wan-kenobi <tobi-wan-kenobi@users.noreply.github.com>
Date: Tue, 21 Apr 2020 20:37:54 +0200
Subject: [PATCH] [modules/shortcut] update to latest API

---
 bumblebee-status            |  1 +
 modules/contrib/shortcut.py | 41 +++++++++++++++++--------------------
 2 files changed, 20 insertions(+), 22 deletions(-)

diff --git a/bumblebee-status b/bumblebee-status
index 1b7039b..d448773 100755
--- a/bumblebee-status
+++ b/bumblebee-status
@@ -64,6 +64,7 @@ def main():
     core.event.trigger('stop')
 
 if __name__ == "__main__":
+    main()
     try:
         main()
     except Exception as e:
diff --git a/modules/contrib/shortcut.py b/modules/contrib/shortcut.py
index 489741b..b038f37 100644
--- a/modules/contrib/shortcut.py
+++ b/modules/contrib/shortcut.py
@@ -18,32 +18,33 @@ Parameters:
 """
 
 import logging
-import bumblebee.engine
-import bumblebee.output
-import bumblebee.input
 
 LINK = 'https://github.com/tobi-wan-kenobi/bumblebee-status/wiki'
 LABEL = 'Click me'
 
-class Module(bumblebee.engine.Module):
-    """ Shortcut module."""
+import core.module
+import core.widget
+import core.input
+import core.decorators
 
-    def __init__(self, engine, config):
-        widgets = []
-        self._engine = engine
-        super(Module, self).__init__(engine, config, widgets)
+class Module(core.module.Module):
+    @core.decorators.every(minutes=60)
+    def __init__(self, config):
+        super().__init__(config, [])
 
-        self._labels = self.parameter('labels', '{}'.format(LABEL))
-        self._cmds = self.parameter('cmds', 'firefox {}'.format(LINK))
-        self._delim = self.parameter('delim', ';')
+        self.__labels = self.parameter('labels', '{}'.format(LABEL))
+        self.__cmds = self.parameter('cmds', 'firefox {}'.format(LINK))
+        self.__delim = self.parameter('delim', ';')
 
-        self.update_widgets(widgets)
+        self.update_widgets()
 
-    def update_widgets(self, widgets):
+    def update_widgets(self):
         """ Creates a set of widget per user define shortcut."""
 
-        cmds = self._cmds.split(self._delim)
-        labels = self._labels.split(self._delim)
+        widgets = self.widgets()
+
+        cmds = self.__cmds.split(self.__delim)
+        labels = self.__labels.split(self.__delim)
 
         # to be on the safe side create as many widgets as there are data (cmds or labels)
         num_shortcuts = min(len(cmds), len(labels))
@@ -58,13 +59,9 @@ class Module(bumblebee.engine.Module):
             cmd = cmds[idx]
             label = labels[idx]
 
-            widget = bumblebee.output.Widget(full_text=label)
-            self._engine.input.register_callback(widget, button=bumblebee.input.LEFT_MOUSE, cmd=cmd)
+            widget = core.widget.Widget(full_text=label)
+            core.input.register(widget, button=core.input.LEFT_MOUSE, cmd=cmd)
 
             widgets.append(widget)
 
-    def update(self, widgets):
-        if len(widgets) <= 0:
-            self.update_widgets(widgets)
-
 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4