From 92eeac361246105f58a0c4027905deeed97268a1 Mon Sep 17 00:00:00 2001
From: slush0 <slush@satoshilabs.com>
Date: Thu, 28 Apr 2016 23:48:59 +0200
Subject: [PATCH] Handle exceptions in better way, enable to return to main
 screen many times.

---
 src/trezor/layout.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/trezor/layout.py b/src/trezor/layout.py
index 88e1fed74..0fc276e06 100644
--- a/src/trezor/layout.py
+++ b/src/trezor/layout.py
@@ -1,4 +1,5 @@
 import sys
+import utime
 
 _new_layout = None
 _current_layout = None
@@ -15,17 +16,18 @@ def set_main(main_layout):
     global _new_layout
     global _current_layout
 
-    _current_layout = main_layout
+    _current_layout = main_layout()
     while True:
         try:
             _current_layout = yield from _current_layout
         except Exception as e:
             sys.print_exception(e)
-            sys.exit()
+            utime.sleep(1)  # Don't produce wall of exceptions
             # if _current_layout == main_layout:
             #    # Main layout thrown exception, what to do?
-            # _current_layout = main_layout
-            # continue
+            #    sys.exit()
+            _current_layout = main_layout()
+            continue
 
         if _new_layout != None:
             print("Switching to new layout %s" % _new_layout)