90 lines
3.4 KiB
Python
90 lines
3.4 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 functools
|
|
|
|
try:
|
|
import tkinter as tk
|
|
from tkinter import messagebox as tkmessagebox
|
|
except ImportError:
|
|
logging.warning('failed to import tkinter - bumblebee popups won\'t work!')
|
|
|
|
import core.module
|
|
import core.widget
|
|
import core.input
|
|
import core.decorators
|
|
|
|
import util.cli
|
|
import util.popup
|
|
import util.format
|
|
|
|
class Module(core.module.Module):
|
|
@core.decorators.every(minutes=60)
|
|
def __init__(self, config):
|
|
super().__init__(config, core.widget.Widget(self.text))
|
|
|
|
self.__confirm = util.format.asbool(self.parameter('confirm', True))
|
|
|
|
core.input.register(self, button=core.input.LEFT_MOUSE,
|
|
cmd=self.popup)
|
|
|
|
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:
|
|
util.cli.execute(command)
|
|
|
|
|
|
def popup(self, widget):
|
|
menu = util.popup.menu()
|
|
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(util.cli.execute, switch_user_cmd))
|
|
menu.add_menuitem('lock', callback=functools.partial(util.cli.execute, lock_cmd))
|
|
menu.add_menuitem('suspend', callback=functools.partial(util.cli.execute, suspend_cmd))
|
|
menu.add_menuitem('hibernate', callback=functools.partial(util.cli.execute, hibernate_cmd))
|
|
|
|
menu.show(widget, 0, 0)
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|