[modules] re-enable bluetooth

modules/contrib/bluetooth.py

@@ -0,0 +1,131 @@
+"""Displays bluetooth status (Bluez). Left mouse click launches manager app,
+right click toggles bluetooth. Needs dbus-send to toggle bluetooth state.
+
+Parameters:
+    * bluetooth.device : the device to read state from (default is hci0)
+    * bluetooth.manager : application to launch on click (blueman-manager)
+    * bluetooth.dbus_destination : dbus destination (defaults to org.blueman.Mechanism)
+    * bluetooth.dbus_destination_path : dbus destination path (defaults to /)
+    * bluetooth.right_click_popup : use popup menu when right-clicked (defaults to True)
+
+"""
+
+
+import os
+import re
+import bumblebee.input
+import bumblebee.output
+import bumblebee.engine
+import bumblebee.util
+import bumblebee.popup
+import logging
+
+
+class Module(bumblebee.engine.Module):
+    """Bluetooth module."""
+
+    def __init__(self, engine, config):
+        """Initialize."""
+        super(Module, self).__init__(engine, config,
+                                     bumblebee.output.Widget(
+                                         full_text=self.status))
+
+        device = self.parameter("device", "hci0")
+        self.manager = self.parameter("manager", "blueman-manager")
+        self._path = "/sys/class/bluetooth/{}".format(device)
+        self._status = "Off"
+
+        engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
+                                       cmd=self.manager)
+
+        # determine whether to use pop-up menu or simply toggle the device on/off
+        right_click_popup = bumblebee.util.asbool(
+            self.parameter("right_click_popup", True))
+
+        if right_click_popup:
+            engine.input.register_callback(self,
+                                           button=bumblebee.input.RIGHT_MOUSE,
+                                           cmd=self.popup)
+        else:
+            engine.input.register_callback(self,
+                                           button=bumblebee.input.RIGHT_MOUSE,
+                                           cmd=self._toggle)
+
+    def status(self, widget):
+        """Get status."""
+        return self._status
+
+    def update(self, widgets):
+        """Update current state."""
+        if not os.path.exists(self._path):
+            self._status = "?"
+            return
+
+        # search for whichever rfkill directory available
+        try:
+            dirnames = next(os.walk(self._path))[1]
+            for dirname in dirnames:
+                m = re.match(r"rfkill[0-9]+", dirname)
+                if m is not None:
+                    with open(os.path.join(self._path,
+                                           dirname,
+                                           'state'), 'r') as f:
+                        state = int(f.read())
+                        if state == 1:
+                            self._status = "On"
+                        else:
+                            self._status = "Off"
+                    return
+
+        except IOError:
+            self._status = "?"
+
+    def manager(self, widget):
+        """Launch manager."""
+        bumblebee.util.execute(self.manager)
+
+    def popup(self, widget):
+        """Show a popup menu."""
+        menu = bumblebee.popup.PopupMenu()
+        if self._status == "On":
+            menu.add_menuitem('Disable Bluetooth')
+        elif self._status == "Off":
+            menu.add_menuitem('Enable Bluetooth')
+        else:
+            return
+
+        # show menu and get return code
+        ret = menu.show(widget)
+        if ret == 0:
+            # first (and only) item selected.
+            self._toggle()
+
+    def _toggle(self, widget=None):
+        """Toggle bluetooth state."""
+        if self._status == "On":
+            state = "false"
+        else:
+            state = "true"
+
+        dst = self.parameter("dbus_destination", "org.blueman.Mechanism")
+        dst_path = self.parameter("dbus_destination_path", "/")
+
+        cmd = "dbus-send --system --print-reply --dest={}"\
+              " {} org.blueman.Mechanism.SetRfkillState"\
+              " boolean:{}".format(dst, dst_path, state)
+
+        logging.debug('bt: toggling bluetooth')
+        bumblebee.util.execute(cmd)
+
+    def state(self, widget):
+        """Get current state."""
+        state = []
+
+        if self._status == "?":
+            state = ["unknown"]
+        elif self._status == "On":
+            state = ["ON"]
+        else:
+            state = ["OFF"]
+
+        return state