63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
from asciimatics.exceptions import NextScene
|
|
from asciimatics.widgets import Layout, Label
|
|
import requests
|
|
|
|
from .CeoFrame import CeoFrame
|
|
|
|
|
|
class ResultView(CeoFrame):
|
|
def __init__(self, screen, width, height, model):
|
|
super().__init__(
|
|
screen, height, width, model, 'Result',
|
|
on_load=self._resultview_on_load, title='Result',
|
|
escape_on_q=True,
|
|
)
|
|
|
|
# TODO: deduplicate this from ConfirmView
|
|
def _add_text(self, text: str = '\n', center: bool = False):
|
|
if center:
|
|
layout = Layout([100])
|
|
align = '^'
|
|
col = 0
|
|
else:
|
|
layout = Layout([1, 10])
|
|
align = '<'
|
|
col = 1
|
|
self.add_layout(layout)
|
|
for line in text.splitlines():
|
|
layout.add_widget(Label(line, align=align), col)
|
|
|
|
def _add_pair(self, key: str, val: str):
|
|
layout = Layout([10, 1, 10])
|
|
self.add_layout(layout)
|
|
if key:
|
|
layout.add_widget(Label(key + ':', align='>'), 0)
|
|
else:
|
|
layout.add_widget(Label(''), 0)
|
|
layout.add_widget(Label(val, align='<'), 2)
|
|
|
|
# override this method in child classes if desired
|
|
def show_result(self, resp: requests.Response):
|
|
self._add_text('The operation was successfully performed.', center=True)
|
|
|
|
def _resultview_on_load(self):
|
|
self._add_text()
|
|
resp = self._model.resp
|
|
if not resp.ok:
|
|
self._add_text('An error occurred:')
|
|
if resp.headers.get('content-type') == 'application/json':
|
|
err_msg = resp.json()['error']
|
|
else:
|
|
err_msg = resp.text.rstrip()
|
|
self._add_text(err_msg)
|
|
else:
|
|
self.show_result(resp)
|
|
# fill the rest of the space
|
|
self.add_layout(Layout([100], fill_frame=True))
|
|
self.add_buttons(on_next_excl=self._next)
|
|
self.fix()
|
|
|
|
def _next(self):
|
|
self._model.reset()
|
|
raise NextScene('Welcome')
|