88 lines
3 KiB
Python
88 lines
3 KiB
Python
"""Displays bluetooth status. Left mouse click launches manager app `blueman-manager`,
|
|
right click toggles bluetooth. Needs dbus-send to toggle bluetooth state and
|
|
python-dbus to count the number of connections
|
|
|
|
Parameters:
|
|
* bluetooth.manager : application to launch on click (blueman-manager)
|
|
|
|
contributed by `martindoublem <https://github.com/martindoublem>`_ - many thanks!
|
|
"""
|
|
|
|
import os
|
|
import re
|
|
import subprocess
|
|
import dbus
|
|
import dbus.mainloop.glib
|
|
import logging
|
|
|
|
import core.module
|
|
import core.widget
|
|
import core.input
|
|
|
|
import util.cli
|
|
|
|
class Module(core.module.Module):
|
|
def __init__(self, config, theme):
|
|
super().__init__(config, theme, core.widget.Widget(self.status))
|
|
|
|
self.manager = self.parameter("manager", "blueman-manager")
|
|
self._status = "Off"
|
|
dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
|
|
self._bus = dbus.SystemBus()
|
|
|
|
core.input.register(self, button=core.input.LEFT_MOUSE, cmd=self.manager)
|
|
core.input.register(self, button=core.input.RIGHT_MOUSE, cmd=self._toggle)
|
|
|
|
def status(self, widget):
|
|
"""Get status."""
|
|
return self._status if self._status.isdigit() and int(self._status) > 1 else ""
|
|
|
|
def update(self):
|
|
"""Update current state."""
|
|
state = len(
|
|
subprocess.run(["bluetoothctl", "list"], stdout=subprocess.PIPE).stdout
|
|
)
|
|
if state > 0:
|
|
connected_devices = self.get_connected_devices()
|
|
self._status = "{}".format(connected_devices)
|
|
else:
|
|
self._status = "Off"
|
|
adapters_cmd = "rfkill list | grep Bluetooth"
|
|
if not len(
|
|
subprocess.run(adapters_cmd, shell=True, stdout=subprocess.PIPE).stdout
|
|
):
|
|
self._status = "No Adapter Found"
|
|
return
|
|
|
|
def _toggle(self, widget=None):
|
|
"""Toggle bluetooth state."""
|
|
logging.debug("bt: toggling bluetooth")
|
|
|
|
SetRfkillState = self._bus.get_object("org.blueman.Mechanism", "/org/blueman/mechanism").get_dbus_method("SetRfkillState", dbus_interface="org.blueman.Mechanism")
|
|
SetRfkillState(self._status == "Off")
|
|
|
|
def state(self, widget):
|
|
"""Get current state."""
|
|
state = []
|
|
|
|
if self._status in [ "No Adapter Found", "Off" ]:
|
|
state.append("critical")
|
|
elif self._status == "0":
|
|
state.append("warning")
|
|
else:
|
|
state.append("ON")
|
|
|
|
return state
|
|
|
|
def get_connected_devices(self):
|
|
devices = 0
|
|
objects = dbus.Interface(
|
|
self._bus.get_object("org.bluez", "/"), "org.freedesktop.DBus.ObjectManager"
|
|
).GetManagedObjects()
|
|
for path, interfaces in objects.items():
|
|
if "org.bluez.Device1" in interfaces:
|
|
if dbus.Interface(self._bus.get_object("org.bluez", path), "org.freedesktop.DBus.Properties", ).Get("org.bluez.Device1", "Connected"):
|
|
devices += 1
|
|
return devices
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|