diff --git a/modules/contrib/system.py b/modules/contrib/system.py new file mode 100644 index 0000000..e2f296a --- /dev/null +++ b/modules/contrib/system.py @@ -0,0 +1,98 @@ +# -*- coding: utf-8 -*- +# pylint: disable=C0111,R0903 + +""" system module + +adds the possibility to + * shutdown + * reboot +the system. + +Per default a confirmation dialog is shown before the actual action is performed. + +Parameters: + * system.confirm: show confirmation dialog before performing any action (default: true) + * system.reboot: specify a reboot command (defaults to 'reboot') + * system.shutdown: specify a shutdown command (defaults to 'shutdown -h now') + * system.logout: specify a logout command (defaults to 'i3exit logout') + * system.switch_user: specify a command for switching the user (defaults to 'i3exit switch_user') + * system.lock: specify a command for locking the screen (defaults to 'i3exit lock') + * system.suspend: specify a command for suspending (defaults to 'i3exit suspend') + * system.hibernate: specify a command for hibernating (defaults to 'i3exit hibernate') +""" + +import logging +import bumblebee.input +import bumblebee.output +import bumblebee.engine +import bumblebee.popup_v2 +import functools + +try: + import Tkinter as tk + import tkMessageBox as tkmessagebox +except ImportError: + # python 3 + try: + import tkinter as tk + from tkinter import messagebox as tkmessagebox + except ImportError: + logging.warning("failed to import tkinter - bumblebee popups won't work!") + + +class Module(bumblebee.engine.Module): + def __init__(self, engine, config): + super(Module, self).__init__(engine, config, + bumblebee.output.Widget(full_text=self.text) + ) + + self._confirm = True + if self.parameter("confirm", "true") == "false": + self._confirm = False + + engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE, + cmd=self.popup) + + def update(self, widgets): + pass + + def text(self, widget): + return "" + + def _on_command(self, header, text, command): + do_it = True + if self._confirm: + root = tk.Tk() + root.withdraw() + root.focus_set() + + do_it = tkmessagebox.askyesno(header, text) + root.destroy() + + if do_it: + bumblebee.util.execute(command) + + + def popup(self, widget): + menu = bumblebee.popup_v2.PopupMenu() + reboot_cmd = self.parameter("reboot", "reboot") + shutdown_cmd = self.parameter("shutdown", "shutdown -h now") + logout_cmd = self.parameter("logout", "i3exit logout") + switch_user_cmd = self.parameter("switch_user", "i3exit switch_user") + lock_cmd = self.parameter("lock", "i3exit lock") + suspend_cmd = self.parameter("suspend", "i3exit suspend") + hibernate_cmd = self.parameter("hibernate", "i3exit hibernate") + + menu.add_menuitem("shutdown", callback=functools.partial(self._on_command, "Shutdown", "Shutdown?", shutdown_cmd)) + menu.add_menuitem("reboot", callback=functools.partial(self._on_command, "Reboot", "Reboot?", reboot_cmd)) + menu.add_menuitem("log out", callback=functools.partial(self._on_command, "Log out", "Log out?", "i3exit logout")) + # don't ask for these + menu.add_menuitem("switch user", callback=functools.partial(bumblebee.util.execute, switch_user_cmd)) + menu.add_menuitem("lock", callback=functools.partial(bumblebee.util.execute, lock_cmd)) + menu.add_menuitem("suspend", callback=functools.partial(bumblebee.util.execute, suspend_cmd)) + menu.add_menuitem("hibernate", callback=functools.partial(bumblebee.util.execute, hibernate_cmd)) + + menu.show(widget) + + def state(self, widget): + return []