arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-04 11:51:50 +00:00
Code Issues Releases Wiki Activity

trezor.ui.confirm: support Widget content

Browse Source
This commit is contained in:
Jan Pochyla 2017-02-08 16:20:20 +01:00
parent 910033d71c
commit e4af95e11b
1 changed files with 6 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/trezor/ui/confirm.py
View File

@ -1,5 +1,6 @@
from micropython import const from micropython import const
from trezor import loop from trezor import loop
from trezor.ui import Widget
from .button import Button, BTN_CLICKED, BTN_STARTED from .button import Button, BTN_CLICKED, BTN_STARTED
from .button import CONFIRM_BUTTON, CONFIRM_BUTTON_ACTIVE from .button import CONFIRM_BUTTON, CONFIRM_BUTTON_ACTIVE
from .button import CANCEL_BUTTON, CANCEL_BUTTON_ACTIVE from .button import CANCEL_BUTTON, CANCEL_BUTTON_ACTIVE
@ -10,7 +11,7 @@ CONFIRMED = const(1)
CANCELLED = const(2) CANCELLED = const(2)
class ConfirmDialog(): class ConfirmDialog(Widget):
def __init__(self, content=None, confirm='Confirm', cancel='Cancel'): def __init__(self, content=None, confirm='Confirm', cancel='Cancel'):
self.content = content self.content = content
@ -22,26 +23,17 @@ class ConfirmDialog():
active_style=CANCEL_BUTTON_ACTIVE) active_style=CANCEL_BUTTON_ACTIVE)
def render(self): def render(self):
if self.content is not None:
self.content.render()
self.confirm.render() self.confirm.render()
self.cancel.render() self.cancel.render()
def send(self, event, pos): def touch(self, event, pos):
if self.confirm.send(event, pos) == BTN_CLICKED: if self.confirm.touch(event, pos) == BTN_CLICKED:
return CONFIRMED return CONFIRMED
if self.cancel.send(event, pos) == BTN_CLICKED: if self.cancel.touch(event, pos) == BTN_CLICKED:
return CANCELLED return CANCELLED
if self.content is not None:
return self.content.send(event, pos)
def __iter__(self): def __iter__(self):
while True: yield loop.Wait((super().__iter__(), self.content))
self.render()
event, *pos = yield loop.Select(loop.TOUCH)
result = self.send(event, pos)
if result is not None:
return result
class HoldToConfirmDialog(): class HoldToConfirmDialog():