bumblebee-status/bumblebee/modules/bluetooth.py
Pi-Yueh Chuang 1cd5e2edf7
add a parameter to control bluetooth popup menu
The right-click popup menu in the bluetooth module does not work well in
Sway (a i3wm-compatible Wayland WM). This is known issue to
Sway. Given that the popup menu is simply to turn on/off the
bluetooth device, the graphical popup is really not necessary. This
commit adds a parameter to control whether a popup menu should be used
when a right-click happens, or we simply turn on/off the device without
the menu.
2019-12-18 16:53:58 -05:00

131 lines
4.3 KiB
Python

"""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