From 95a05bf886066e26e50771dc4156950407ef3a40 Mon Sep 17 00:00:00 2001
From: Frantisek Sindelar <sindelar.fr@gmail.com>
Date: Sun, 29 Sep 2019 15:27:56 +0200
Subject: [PATCH] Fixing import path for Waveshare V2. Fixing the full refresh
 trigger to enable 0 for every time and -1 for never.

---
 sdcard/rootfs/root/pwnagotchi/config.yml               |  4 ++--
 .../root/pwnagotchi/scripts/pwnagotchi/ui/display.py   | 10 +++++-----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/sdcard/rootfs/root/pwnagotchi/config.yml b/sdcard/rootfs/root/pwnagotchi/config.yml
index dee263c..8d5c3ac 100644
--- a/sdcard/rootfs/root/pwnagotchi/config.yml
+++ b/sdcard/rootfs/root/pwnagotchi/config.yml
@@ -96,8 +96,8 @@ ui:
         type: 'waveshare_2'
         # Possible options red/yellow/black (black used for monocromatic displays)
         color: 'black'
-        # How often to do a full refresh
-        refresh: 10
+        # How often to do a full refresh 0 all the time, -1 never
+        refresh: 50
         video:
             enabled: true
             address: '10.0.0.2'
diff --git a/sdcard/rootfs/root/pwnagotchi/scripts/pwnagotchi/ui/display.py b/sdcard/rootfs/root/pwnagotchi/scripts/pwnagotchi/ui/display.py
index 666dca1..598ed2c 100644
--- a/sdcard/rootfs/root/pwnagotchi/scripts/pwnagotchi/ui/display.py
+++ b/sdcard/rootfs/root/pwnagotchi/scripts/pwnagotchi/ui/display.py
@@ -128,7 +128,7 @@ class Display(View):
             self._display.init(self._display.lut_partial_update)
             self._render_cb = self._waveshare_render
         elif self._is_waveshare2():
-            from pwnagotchi.ui.v2.waveshare import EPD
+            from pwnagotchi.ui.waveshare.v2.waveshare import EPD
             # core.log("display module started")
             self._display = EPD()
             self._display.init(self._display.FULL_UPDATE)
@@ -180,17 +180,17 @@ class Display(View):
     def _waveshare_render(self):
         buf = self._display.getbuffer(self.canvas)
         if self._is_waveshare1:
-            if self.full_refresh_count == self.full_refresh_trigger:
+            if self.full_refresh_trigger >= 0 and self.full_refresh_count == self.full_refresh_trigger:
                 self._display.Clear(0x00)
             self._display.display(buf)
         elif self._is_waveshare2:
-            if self.full_refresh_count == self.full_refresh_trigger:
+            if self.full_refresh_trigger >= 0 and self.full_refresh_count == self.full_refresh_trigger:
                 self._display.Clear(BLACK)
             self._display.displayPartial(buf)
         self._display.sleep()
-        if self.full_refresh_count == self.full_refresh_trigger:
+        if self.full_refresh_trigger >= 0 and self.full_refresh_count == self.full_refresh_trigger:
            self.full_refresh_count = 0
-        else:
+        elif self.full_refresh_trigger >= 0:
            self.full_refresh_count += 1
 
     def _on_view_rendered(self, img):