bumblebee-status/modules/contrib/system.py

# -*- 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