78 lines
2 KiB
Python
78 lines
2 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.
|
||
|
|
|
||
|
|
Paramters:
|
||
|
|
* system.confirm: show confirmation dialog before performing any action (default: true)
|
||
|
|
"""
|
||
|
|
|
||
|
|
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)
|
||
|
|
|
||
|
|
if do_it:
|
||
|
|
bumblebee.util.execute(command)
|
||
|
|
root.destroy()
|
||
|
|
|
||
|
|
|
||
|
|
def popup(self, widget):
|
||
|
|
menu = bumblebee.popup_v2.PopupMenu()
|
||
|
|
menu.add_menuitem("shutdown", callback=functools.partial(self._on_command, "Shutdown", "Shutdown?", "shutdown -h now"))
|
||
|
|
menu.add_menuitem("reboot", callback=functools.partial(self._on_command, "Reboot", "Reboot?", "reboot"))
|
||
|
|
|
||
|
|
menu.show(widget)
|
||
|
|
|
||
|
|
def state(self, widget):
|
||
|
|
return []
|