subiquity/subiquitycore/ui/views/installprogress.py

# Copyright 2015 Canonical, Ltd.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import logging
from urwid import (Text, Filler,
                   Pile)
from subiquitycore.view import ViewPolicy
from subiquitycore.ui.buttons import confirm_btn
from subiquitycore.ui.utils import Padding, Color

log = logging.getLogger("subiquitycore.views.installprogress")


class ProgressView(ViewPolicy):
    def __init__(self, model, signal):
        """
        :param output_w: Filler widget to display updated status text
        """
        self.model = model
        self.signal = signal
        self.text = Text("Wait for it ...", align="left")
        self.body = [
            Padding.center_79(self.text),
            Padding.line_break(""),
        ]
        self.pile = Pile(self.body)
        super().__init__(Filler(self.pile, valign="middle"))

    def show_finished_button(self):
        w = Padding.fixed_20(
            Color.button(confirm_btn(label="Reboot now",
                                     on_press=self.reboot),
                         focus_map='button focus'))

        z = Padding.fixed_20(
            Color.button(confirm_btn(label="Quit Installer",
                                     on_press=self.quit),
                         focus_map='button focus'))

        self.pile.contents.append((w, self.pile.options()))
        self.pile.contents.append((z, self.pile.options()))
        self.pile.focus_position = 2

    def reboot(self, btn):
        self.signal.emit_signal('installprogress:curtin-reboot')

    def quit(self, btn):
        self.signal.emit_signal('quit')