popup.py

import curses
import logging
from enum import Enum

from widget import WidgetBase
from helpers import *

log = logging.getLogger('puzzle')

button_w = 20

class Popup(WidgetBase):

    def __init__(self, app, parent, title, text, buttons=['OK']):
        assert len(buttons) > 0

        h, w = parent.getmaxyx()
        parent_size = Vector(w, h)

        self.layout_text = layout_text(text, 100) # max width: 100
        self.height = len(self.layout_text) + 2 # title, buttons

        # set size according to contents of popup
        self.width = max(max(len(l) for l in self.layout_text), len(buttons)*(button_w+1))

        size = Vector(self.width+4, self.height+4)

        # center in middle of parent
        super(Popup, self).__init__(app, parent_size/2-size/2, size)

        self.title = title
        self.app = app
        self.buttons = buttons
        self.selected_button = 0

        curses.init_pair(30, curses.COLOR_WHITE, curses.COLOR_BLACK)
        curses.init_pair(31, curses.COLOR_WHITE, curses.COLOR_RED)
        self.col_unselected = 30
        self.col_selected   = 31

        self.callback = None

    def handle_input(self, key):
        if   key == curses.KEY_LEFT  or key == curses.KEY_BTAB: self.selected_button -= 1
        elif key == curses.KEY_RIGHT or key == ord('\t'):       self.selected_button += 1
        elif key == curses.KEY_ENTER or key == ord(' ') or key == ord('\n') or key == ord('\r'):
            if self.callback:
                log.info('Popup: calling callback')
                self.app.mi.send_packet({
                    'command':  'popup_closed',
                    'button':   str(self.buttons[self.selected_button]),
                    'title':    self.title
                })
                self.callback(self.buttons[self.selected_button])
                return True
            else:
                log.warn('Popup: no callback defined!')
        else:
            return False

        self.selected_button %= len(self.buttons)
        return True


    def draw(self):
        #self.screen.clear() # everything gets painted over anyways
        self.screen.border()

        # title
        self.screen.addstr(1,2, self.title, curses.A_BOLD)

        # text
        for y, line in enumerate(self.layout_text):
            self.screen.addstr(3+y, 2, line)

        # buttons
        for i, button in enumerate(self.buttons):
            attr = curses.A_BOLD | curses.color_pair(self.col_selected if i == self.selected_button else self.col_unselected)
            self.screen.addstr(self.height+2,
                    self.width+3 - (button_w+1)*(len(self.buttons) - i),
                    f"{button:^{button_w}}", attr)