113 lines
3.8 KiB
Python
113 lines
3.8 KiB
Python
from fabric import Application
|
|
from fabric.widgets.box import Box
|
|
from fabric.widgets.label import Label
|
|
from fabric.widgets.image import Image
|
|
from fabric.widgets.button import Button
|
|
from fabric.widgets.window import Window
|
|
from fabric.widgets.centerbox import CenterBox
|
|
from fabric.widgets.scrolledwindow import ScrolledWindow
|
|
from fabric.bluetooth import BluetoothClient, BluetoothDevice
|
|
|
|
|
|
class BluetoothDeviceSlot(CenterBox):
|
|
def __init__(self, device: BluetoothDevice, **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.device = device
|
|
self.device.connect("changed", self.on_changed)
|
|
self.device.connect(
|
|
"notify::closed", lambda *_: self.device.closed and self.destroy()
|
|
)
|
|
|
|
self.connection_label = Label(label="Disconnected")
|
|
self.connect_button = Button(
|
|
label="Connect",
|
|
on_clicked=lambda *_: self.device.set_connecting(not self.device.connected),
|
|
)
|
|
|
|
self.start_children = [
|
|
Image(icon_name=device.icon_name + "-symbolic", size=32),
|
|
Label(label=device.name),
|
|
]
|
|
self.center_children = self.connection_label
|
|
self.end_children = self.connect_button
|
|
|
|
self.device.emit("changed") # to update display status
|
|
|
|
def on_changed(self, *_):
|
|
self.connection_label.set_label(
|
|
"Connected" if self.device.connected else "Disconnected"
|
|
)
|
|
if self.device.connecting:
|
|
self.connect_button.set_label(
|
|
"Connecting..." if not self.device.connecting else "Disconnecting..."
|
|
)
|
|
else:
|
|
self.connect_button.set_label(
|
|
"Connect" if not self.device.connected else "Disconnect"
|
|
)
|
|
return
|
|
|
|
|
|
class BluetoothConnections(Box):
|
|
def __init__(self, **kwargs):
|
|
super().__init__(
|
|
spacing=4,
|
|
orientation="vertical",
|
|
style="margin: 8px",
|
|
**kwargs,
|
|
)
|
|
|
|
self.client = BluetoothClient(on_device_added=self.on_device_added)
|
|
self.scan_button = Button(on_clicked=lambda *_: self.client.toggle_scan())
|
|
self.toggle_button = Button(on_clicked=lambda *_: self.client.toggle_power())
|
|
|
|
self.client.connect(
|
|
"notify::enabled",
|
|
lambda *_: self.toggle_button.set_label(
|
|
"Bluetooth " + ("ON" if self.client.enabled else "OFF")
|
|
),
|
|
)
|
|
self.client.connect(
|
|
"notify::scanning",
|
|
lambda *_: self.scan_button.set_label(
|
|
"Stop" if self.client.scanning else "Scan"
|
|
),
|
|
)
|
|
|
|
self.paired_box = Box(spacing=2, orientation="vertical")
|
|
self.available_box = Box(spacing=2, orientation="vertical")
|
|
|
|
self.children = [
|
|
CenterBox(start_children=self.scan_button, end_children=self.toggle_button),
|
|
Label("Paired Devices"),
|
|
ScrolledWindow(min_content_size=(400, -1), child=self.paired_box),
|
|
Label("Available Devices"),
|
|
ScrolledWindow(min_content_size=(-1, 400), child=self.available_box),
|
|
]
|
|
|
|
# to run notify closures thus display the status
|
|
# without having to wait until an actual change
|
|
self.client.notify("scanning")
|
|
self.client.notify("enabled")
|
|
|
|
def on_device_added(self, client: BluetoothClient, address: str):
|
|
if not (device := client.get_device(address)):
|
|
return
|
|
slot = BluetoothDeviceSlot(device)
|
|
|
|
if device.paired:
|
|
return self.paired_box.add(slot)
|
|
return self.available_box.add(slot)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
bluetooth_widget = Window(
|
|
child=BluetoothConnections(),
|
|
visible=True,
|
|
all_visible=True,
|
|
on_destroy=lambda *_: app.quit(),
|
|
)
|
|
app = Application("bluetooth-widget", bluetooth_widget)
|
|
|
|
app.run()
|