source: prevent to skip the drivers screen but allow to go back

Signed-off-by: Olivier Gayot <olivier.gayot@canonical.com>
This commit is contained in:
Olivier Gayot 2022-03-16 16:20:41 +01:00
parent 1ffff94eff
commit 57d5aef575
1 changed files with 5 additions and 5 deletions

View File

@ -26,7 +26,7 @@ from urwid import (
Text, Text,
) )
from subiquitycore.ui.buttons import ok_btn from subiquitycore.ui.buttons import back_btn, ok_btn
from subiquitycore.ui.form import ( from subiquitycore.ui.form import (
Form, Form,
BooleanField, BooleanField,
@ -81,10 +81,10 @@ class DriversView(BaseView):
Text(""), Text(""),
self.spinner, self.spinner,
] ]
self.cont_btn = ok_btn( self.back_btn = back_btn(
_("Continue"), _("Back"),
on_press=lambda sender: self.done(False)) on_press=lambda sender: self.cancel())
self._w = screen(rows, [self.cont_btn]) self._w = screen(rows, [self.back_btn])
asyncio.create_task(self._wait(install)) asyncio.create_task(self._wait(install))
self.status = DriversViewStatus.WAITING self.status = DriversViewStatus.WAITING