99 lines
3.6 KiB
Python
99 lines
3.6 KiB
Python
|
# -*- 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 []
|