From 84c14042b6f2ffce3c3774007624fd3dd9d7106e Mon Sep 17 00:00:00 2001
From: Jan Pochyla <jpochyla@gmail.com>
Date: Tue, 2 Apr 2019 15:34:47 +0200
Subject: [PATCH] ui: save screen after refresh, if env var is set

---
 src/trezor/ui/__init__.py |  2 ++
 src/trezor/utils.py       | 10 ++++++++++
 2 files changed, 12 insertions(+)

diff --git a/src/trezor/ui/__init__.py b/src/trezor/ui/__init__.py
index 91fceee2c..0afc00a33 100644
--- a/src/trezor/ui/__init__.py
+++ b/src/trezor/ui/__init__.py
@@ -13,6 +13,8 @@ if __debug__:
     def debug_display_refresh():
         display.bar(Display.WIDTH - 8, 0, 8, 8, 0xF800)
         display.refresh()
+        if utils.SAVE_SCREEN:
+            display.save("refresh")
 
     loop.after_step_hook = debug_display_refresh
 
diff --git a/src/trezor/utils.py b/src/trezor/utils.py
index 77667fba7..6c6ab6670 100644
--- a/src/trezor/utils.py
+++ b/src/trezor/utils.py
@@ -13,6 +13,16 @@ from trezorutils import (  # noqa: F401
     set_mode_unprivileged,
 )
 
+if __debug__:
+    if EMULATOR:
+        import uos
+
+        TEST = int(uos.getenv("TREZOR_TEST") or "0")
+        SAVE_SCREEN = int(uos.getenv("TREZOR_SAVE_SCREEN") or "0")
+    else:
+        TEST = 0
+        SAVE_SCREEN = 0
+
 
 def unimport_begin():
     return set(sys.modules)