From 906013a6d7fd2748d62f7fda8bbd8cd256304445 Mon Sep 17 00:00:00 2001
From: Tomas Susanka <tsusanka@gmail.com>
Date: Thu, 13 Jun 2019 21:18:23 +0200
Subject: [PATCH] ui/mnemonic: small refactor in input's edit

---
 core/src/trezor/ui/mnemonic.py | 27 ++++++++++++---------------
 1 file changed, 12 insertions(+), 15 deletions(-)

diff --git a/core/src/trezor/ui/mnemonic.py b/core/src/trezor/ui/mnemonic.py
index 6153bdbbb0..3da8041d6c 100644
--- a/core/src/trezor/ui/mnemonic.py
+++ b/core/src/trezor/ui/mnemonic.py
@@ -27,8 +27,8 @@ class InputButton(Button):
     def __init__(self, area, content, word):
         super().__init__(area, content)
         self.word = word
-        self.pending = False
-        self.icon = None
+        self.pending = False  # should we draw the pending marker?
+        self.icon = None  # rendered icon
         self.disable()
 
     def edit(self, content, word, pending):
@@ -57,28 +57,25 @@ class InputButton(Button):
         fg_color = s.fg_color
         bg_color = s.bg_color
 
-        p = self.pending  # should we draw the pending marker?
-        t = self.content  # input content
-        w = self.word[len(t) :]  # suggested word
-        i = self.icon  # rendered icon
-
         tx = ax + 24  # x-offset of the content
         ty = ay + ah // 2 + 8  # y-offset of the content
 
-        # input content and the suggested word
-        display.text(tx, ty, t, text_style, fg_color, bg_color)
-        width = display.text_width(t, text_style)
-        display.text(tx + width, ty, w, text_style, ui.GREY, bg_color)
+        # entered content
+        display.text(tx, ty, self.content, text_style, fg_color, bg_color)
+        # word suggestion
+        suggested_word = self.word[len(self.content) :]
+        width = display.text_width(self.content, text_style)
+        display.text(tx + width, ty, suggested_word, text_style, ui.GREY, bg_color)
 
-        if p:  # pending marker
-            pw = display.text_width(t[-1:], text_style)
+        if self.pending:
+            pw = display.text_width(self.content[-1:], text_style)
             px = tx + width - pw
             display.bar(px, ty + 2, pw + 1, 3, fg_color)
 
-        if i:  # icon
+        if self.icon:
             ix = ax + aw - 16 * 2
             iy = ty - 16
-            display.icon(ix, iy, res.load(i), fg_color, bg_color)
+            display.icon(ix, iy, res.load(self.icon), fg_color, bg_color)
 
 
 class Prompt(ui.Control):