From ad701ace7df78c1f004539d8150b1b25176156a2 Mon Sep 17 00:00:00 2001
From: LmeSzinc <37934724+LmeSzinc@users.noreply.github.com>
Date: Sat, 17 Apr 2021 12:09:34 +0800
Subject: [PATCH] Dev: Easier way to set image for development purpose

---
 module/base/base.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/module/base/base.py b/module/base/base.py
index 1fa5a4b6c..4f7d102b4 100644
--- a/module/base/base.py
+++ b/module/base/base.py
@@ -1,3 +1,5 @@
+from PIL import Image
+
 from module.base.button import Button
 from module.base.timer import Timer
 from module.base.utils import *
@@ -140,3 +142,23 @@ class ModuleBase:
     def interval_reset(self, button):
         if button.name in self.interval_timer:
             self.interval_timer[button.name].reset()
+
+    _image_file = ''
+
+    @property
+    def image_file(self):
+        return self._image_file
+
+    @image_file.setter
+    def image_file(self, value):
+        """
+        For development.
+        Load image from local file system and set it to self.device.image
+        Test an image without taking a screenshot from emulator.
+        """
+        if isinstance(value, np.ndarray):
+            value = Image.fromarray(value)
+        elif isinstance(value, str):
+            value = Image.open(value).convert('RGB')
+
+        self.device.image = value