125 lines
3.9 KiB
Python
125 lines
3.9 KiB
Python
# pylint: disable=C0111,R0903
|
|
|
|
import pulsectl
|
|
|
|
import core.module
|
|
import core.widget
|
|
import core.input
|
|
import core.event
|
|
|
|
import util.cli
|
|
import util.graph
|
|
import util.format
|
|
|
|
class Module(core.module.Module):
|
|
def __init__(self, config, theme, type):
|
|
super().__init__(config, theme, core.widget.Widget(self.display))
|
|
self.background = True
|
|
|
|
self.__type = type
|
|
self.__volume = "n/a"
|
|
self.__devicename = "n/a"
|
|
self.__muted = False
|
|
self.__showbars = util.format.asbool(self.parameter("showbars", False))
|
|
self.__show_device_name = util.format.asbool(
|
|
self.parameter("showdevicename", False)
|
|
)
|
|
|
|
self.__change = util.format.asint(
|
|
self.parameter("percent_change", "2%").strip("%"), 0, 100
|
|
)
|
|
self.__limit = util.format.asint(self.parameter("limit", "0%").strip("%"), 0)
|
|
|
|
events = [
|
|
{
|
|
"type": "mute",
|
|
"action": self.toggle_mute,
|
|
"button": core.input.LEFT_MOUSE
|
|
},
|
|
{
|
|
"type": "volume",
|
|
"action": self.increase_volume,
|
|
"button": core.input.WHEEL_UP,
|
|
},
|
|
{
|
|
"type": "volume",
|
|
"action": self.decrease_volume,
|
|
"button": core.input.WHEEL_DOWN,
|
|
},
|
|
]
|
|
|
|
for event in events:
|
|
core.input.register(self, button=event["button"], cmd=event["action"])
|
|
|
|
if util.format.asbool(self.parameter("autostart", False)):
|
|
util.cli.execute("pulseaudio --start", ignore_errors=True)
|
|
|
|
self.process(None)
|
|
|
|
def display(self, _):
|
|
res = f"{int(self.__volume*100)}%"
|
|
if self.__showbars:
|
|
res = f"{res} {util.graph.hbar(self.__volume*100)}"
|
|
|
|
if self.__show_device_name:
|
|
friendly_name = self.__devicename
|
|
icon = self.parameter("icon." + self.__devicename, "")
|
|
res = (
|
|
icon + " " + friendly_name + " | " + res
|
|
if icon != ""
|
|
else friendly_name + " | " + res
|
|
)
|
|
return res
|
|
|
|
def toggle_mute(self, _):
|
|
with pulsectl.Pulse(self.id + "vol") as pulse:
|
|
dev = self.get_device(pulse)
|
|
pulse.mute(dev, not self.__muted)
|
|
|
|
def change_volume(self, amount):
|
|
with pulsectl.Pulse(self.id + "vol") as pulse:
|
|
dev = self.get_device(pulse)
|
|
vol = dev.volume
|
|
vol.value_flat += amount
|
|
if self.__limit > 0 and vol.value_flat > self.__limit/100:
|
|
vol.value_flat = self.__limit/100
|
|
pulse.volume_set(dev, vol)
|
|
|
|
def increase_volume(self, _):
|
|
self.change_volume(self.__change/100.0)
|
|
|
|
def decrease_volume(self, _):
|
|
self.change_volume(-self.__change/100.0)
|
|
|
|
def get_device(self, pulse):
|
|
devs = pulse.sink_list() if self.__type == "sink" else pulse.source_list()
|
|
default = pulse.server_info().default_sink_name if self.__type == "sink" else pulse.server_info().default_source_name
|
|
|
|
for dev in devs:
|
|
if dev.name == default:
|
|
return dev
|
|
return devs[0] # fallback
|
|
|
|
|
|
|
|
def process(self, _):
|
|
with pulsectl.Pulse(self.id + "proc") as pulse:
|
|
dev = self.get_device(pulse)
|
|
self.__volume = dev.volume.value_flat
|
|
self.__muted = dev.mute
|
|
self.__devicename = dev.name
|
|
core.event.trigger("update", [self.id], redraw_only=True)
|
|
core.event.trigger("draw")
|
|
|
|
def update(self):
|
|
with pulsectl.Pulse(self.id) as pulse:
|
|
pulse.event_mask_set(self.__type)
|
|
pulse.event_callback_set(self.process)
|
|
pulse.event_listen()
|
|
|
|
def state(self, _):
|
|
if self.__muted:
|
|
return ["warning", "muted"]
|
|
return ["unmuted"]
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|