bumblebee-status/modules/contrib/dunst.py

#pylint: disable=C0111,R0903

'''Toggle dunst notifications.'''

import core.module
import core.widget
import core.input

import util.cli

class Module(core.module.Module):
    def __init__(self, config=None):
        super().__init__(config, core.widget.Widget(''))
        self._paused = False
        # Make sure that dunst is currently not paused
        util.cli.execute('killall -SIGUSR2 dunst', ignore_errors=True)
        core.input.register(self, button=core.input.LEFT_MOUSE,
            cmd=self.toggle_status
        )

    def toggle_status(self, event):
        self._paused = not self._paused
        
        try:
            if self._paused:
                util.cli.execute('killall -SIGUSR1 dunst')
            else:
                util.cli.execute('killall -SIGUSR2 dunst')
        except:
            self._paused = not self._paused # toggling failed

    def state(self, widget):
        if self._paused:
            return ['muted', 'warning']
        return ['unmuted']