wicd/curses/configscript_curses.py

#!/usr/bin/env python3

"""configscript_curses.py
Kind of like configscript.py, except written using urwid.

Also recycles a lot of configscript.py, too. :-)
"""

#       Copyright (C) 2009 Andrew Psaltis

#       This program is free software; you can redistribute it and/or modify
#       it under the terms of the GNU General Public License as published by
#       the Free Software Foundation; either version 2 of the License, or
#       (at your option) any later version.
#
#       This program is distributed in the hope that it will be useful,
#       but WITHOUT ANY WARRANTY; without even the implied warranty of
#       MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#       GNU General Public License for more details.
#
#       You should have received a copy of the GNU General Public License
#       along with this program; if not, write to the Free Software
#       Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
#       MA 02110-1301, USA.

import sys
import os

import urwid
import urwid.curses_display

from wicd.translations import _

from configscript import write_scripts, get_script_info
from configscript import none_to_blank, blank_to_none

ui = None
frame = None
pre_entry = None
post_entry = None
pre_disconnect_entry = None
post_disconnect_entry = None


def main(argv):
    """Main function."""
    global ui, frame
    if len(argv) < 2:
        print('Network id to configure is missing, aborting.')
        sys.exit(1)

    ui = urwid.curses_display.Screen()
    ui.register_palette([('body', 'default', 'default'),
                         ('focus', 'dark magenta', 'light gray'),
                         ('editcp', 'default', 'default', 'standout'),
                         ('editbx', 'light gray', 'dark blue'),
                         ('editfc', 'white', 'dark blue', 'bold')])

    network = argv[1]
    network_type = argv[2]

    script_info = get_script_info(network, network_type)

    blank = urwid.Text('')
    pre_entry_t = ('body', _('Pre-connection Script') + ': ')
    post_entry_t = ('body', _('Post-connection Script') + ': ')
    pre_disconnect_entry_t = ('body', _('Pre-disconnection Script') + ': ')
    post_disconnect_entry_t = ('body', _('Post-disconnection Script') + ': ')

    global pre_entry, post_entry, pre_disconnect_entry, post_disconnect_entry
    pre_entry = urwid.AttrWrap(urwid.Edit(pre_entry_t,
                                          none_to_blank(script_info.
                                                        get('pre_entry'))),
                               'editbx', 'editfc')
    post_entry = urwid.AttrWrap(urwid.Edit(post_entry_t,
                                           none_to_blank(script_info.
                                                         get('post_entry'))),
                                'editbx', 'editfc')

    pre_disconnect_entry = urwid.AttrWrap(
        urwid.Edit(pre_disconnect_entry_t,
                   none_to_blank(script_info.get('pre_disconnect_entry'))),
        'editbx', 'editfc')
    post_disconnect_entry = urwid.AttrWrap(
        urwid.Edit(post_disconnect_entry_t,
                   none_to_blank(script_info.get('post_disconnect_entry'))),
        'editbx', 'editfc')

    # The buttons
    ok_button = urwid.AttrWrap(urwid.Button(_('OK'), ok_callback), 'body',
                               'focus')
    cancel_button = urwid.AttrWrap(urwid.Button(_('Cancel'), cancel_callback),
                                   'body', 'focus')

    button_cols = urwid.Columns([ok_button, cancel_button], dividechars=1)

    lbox = urwid.Pile([('fixed', 2, urwid.Filler(pre_entry)),
                       # ('fixed', urwid.Filler(blank), 1),
                       ('fixed', 2, urwid.Filler(post_entry)),
                       ('fixed', 2, urwid.Filler(pre_disconnect_entry)),
                       ('fixed', 2, urwid.Filler(post_disconnect_entry)),
                       # blank, blank, blank, blank, blank,
                       urwid.Filler(button_cols, 'bottom')])
    frame = urwid.Frame(lbox)
    result = ui.run_wrapper(run)

    if result:
        script_info["pre_entry"] = blank_to_none(pre_entry.get_edit_text())
        script_info["post_entry"] = blank_to_none(post_entry.get_edit_text())
        script_info["pre_disconnect_entry"] = blank_to_none(
            pre_disconnect_entry.get_edit_text())
        script_info["post_disconnect_entry"] = blank_to_none(
            post_disconnect_entry.get_edit_text())
        write_scripts(network, network_type, script_info)


OK_PRESSED = False
CANCEL_PRESSED = False


def ok_callback(button_object, user_data=None):
    """Callback."""
    global OK_PRESSED
    OK_PRESSED = True


def cancel_callback(button_object, user_data=None):
    """Callback."""
    global CANCEL_PRESSED
    CANCEL_PRESSED = True


def run():
    """Run the UI."""
    dim = ui.get_cols_rows()
    ui.set_mouse_tracking()

    keys = True
    while True:
        if keys:
            ui.draw_screen(dim, frame.render(dim, True))
        keys = ui.get_input()

        if "window resize" in keys:
            dim = ui.get_cols_rows()
        if "esc" in keys or 'Q' in keys:
            return False
        for k in keys:
            # Send key to underlying widget:
            if urwid.is_mouse_event(k):
                event, button, col, row = k
                frame.mouse_event(dim, event, button, col, row, focus=True)
            else:
                frame.keypress(dim, k)
            # Check if buttons are pressed.
        if CANCEL_PRESSED:
            return False
        if OK_PRESSED or 'meta enter' in keys:
            return True


if __name__ == '__main__':
    if os.getuid() != 0:
        print("Root privileges are required to configure scripts.  Exiting.")
        sys.exit(0)
    main(sys.argv)