make_latest: true

generate_release_notes: true

Signed-off-by: jayofelony <oudshoorn.jeroen@gmail.com>

.github/FUNDING.yml

@@ -1,3 +1,4 @@
 # These are supported funding model platforms
 
 github: jayofelony
+custom: https://tikkie.me/pay/dubcto94hnskg539kar0

.github/workflows/publish.yml

@@ -20,21 +20,6 @@ jobs:
           VERSION=$(cut -d "'" -f2 < pwnagotchi/_version.py)
           echo "VERSION=$VERSION" >> $GITHUB_ENV
 
-      - name: Get latest tag
-        uses: actions-ecosystem/action-get-latest-tag@v1
-        id: get-latest-tag
-
-      - name: Set LAST_VERSION as an environment variable
-        run: echo "LAST_VERSION=${{ steps.get-latest-tag.outputs.tag }}" >> $GITHUB_ENV
-
-      - name: Generate release notes
-        id: generate_release_notes
-        run: |
-          COMMITS=$(git log --merges --pretty=format:"* %s" $LAST_VERSION--$VERSION | sed 's/$/\\n/g')
-          CONTRIBUTORS=$(git shortlog -sn $LAST_VERSION--$VERSION | awk '{print "* @" $2}' | sed 's/$/\\n/g')
-          RELEASE_BODY="**Full Changelog**: https://github.com/jayofelony/pwnagotchi/compare/$LAST_VERSION...$VERSION"
-          echo "RELEASE_BODY=$RELEASE_BODY" >> $GITHUB_ENV
-
       - name: Install qemu dependencies
         run: sudo apt update && sudo apt install qemu-user-static qemu-utils xz-utils -y
 
@@ -63,10 +48,11 @@ jobs:
       - name: Release
         uses: softprops/action-gh-release@v2
         with:
-          prerelease: true
+          prerelease: false
+          make_latest: true
           tag_name: v${{ env.VERSION }}
           name: Pwnagotchi v${{ env.VERSION }}
           files: |
               pwnagotchi-${{ env.VERSION }}-32bit.img.xz
               pwnagotchi-${{ env.VERSION }}-64bit.img.xz
-          body: ${{ env.RELEASE_BODY }}
+          generate_release_notes: true

Makefile

@@ -51,13 +51,11 @@ image: clean packer
 
 bullseye: clean packer
 	export LC_ALL=en_GB.UTF-8
-	cd builder && sudo /usr/bin/packer init data/32bit/raspberrypi32.json.pkr.hcl && sudo $(UNSHARE) /usr/bin/packer build -var "pwn_hostname=$(PWN_HOSTNAME)" -var "pwn_version=$(PWN_VERSION)" data/32bit/raspberrypi32.json.pkr.hcl
-	sudo pishrink -vaZ pwnagotchi-32bit.img
+	cd builder && sudo /usr/bin/packer init raspberrypi32.json.pkr.hcl && sudo $(UNSHARE) /usr/bin/packer build -var "pwn_hostname=$(PWN_HOSTNAME)" -var "pwn_version=$(PWN_VERSION)" raspberrypi32.json.pkr.hcl
 
 bookworm: clean packer
 	export LC_ALL=en_GB.UTF-8
-	cd builder && sudo /usr/bin/packer init data/64bit/raspberrypi64.json.pkr.hcl && sudo $(UNSHARE) /usr/bin/packer build -var "pwn_hostname=$(PWN_HOSTNAME)" -var "pwn_version=$(PWN_VERSION)" data/64bit/raspberrypi64.json.pkr.hcl
-	sudo pishrink -vaZ pwnagotchi-64bit.img
+	cd builder && sudo /usr/bin/packer init raspberrypi64.json.pkr.hcl && sudo $(UNSHARE) /usr/bin/packer build -var "pwn_hostname=$(PWN_HOSTNAME)" -var "pwn_version=$(PWN_VERSION)" raspberrypi64.json.pkr.hcl
 
 clean:
 	- rm -rf /tmp/packer*

bin/pwnagotchi

@@ -264,8 +264,7 @@ def pwnagotchi_cli():
         sys.exit(0)
 
     if args.donate:
-        print("Donations can made @ \n "
-              "https://www.patreon.com/pwnagotchi_torch \n "
+        print("Donations can be made @ \n "
               "https://github.com/sponsors/jayofelony \n\n"
               "But only if you really want to!")
         sys.exit(0)
@@ -278,8 +277,7 @@ def pwnagotchi_cli():
         local = version_to_tuple(pwnagotchi.__version__)
         remote = version_to_tuple(latest_ver)
         if remote > local:
-            user_input = input("There is a new version available! Update from v%s to v%s?\n[Y/N] "
-                               % (pwnagotchi.__version__, latest_ver))
+            user_input = input("There is a new version available! Update from v%s to v%s?\n[Y/N] " % (pwnagotchi.__version__, latest_ver))
             # input validation
             if user_input.lower() in ('y', 'yes'):
                 if os.path.exists('/root/.auto-update'):
@@ -334,5 +332,6 @@ def pwnagotchi_cli():
     else:
         do_auto_mode(agent)
 
+
 if __name__ == '__main__':
-    pwnagotchi_cli()
+    pwnagotchi_cli()

builder/combined.json.pkr.hcl

@@ -123,7 +123,7 @@ build {
   provisioner "ansible-local" {
     command         = "ANSIBLE_FORCE_COLOR=1 PYTHONUNBUFFERED=1 PWN_VERSION=${var.pwn_version} PWN_HOSTNAME=${var.pwn_hostname} ansible-playbook"
     extra_arguments = ["--extra-vars \"ansible_python_interpreter=/usr/bin/python3\""]
-    playbook_file   = "data/64bit/raspberrypi64.yml"
+    playbook_file   = "raspberrypi64.yml"
   }
 }
 
@@ -167,7 +167,7 @@ build {
   provisioner "ansible-local" {
     command         = "ANSIBLE_FORCE_COLOR=1 PYTHONUNBUFFERED=1 PWN_VERSION=${var.pwn_version} PWN_HOSTNAME=${var.pwn_hostname} ansible-playbook"
     extra_arguments = ["--extra-vars \"ansible_python_interpreter=/usr/bin/python3\""]
-    playbook_dir    = "data/32bit/extras/"
-    playbook_file   = "data/32bit/raspberrypi32.yml"
+    playbook_dir    = "extras/"
+    playbook_file   = "raspberrypi32.yml"
   }
 }

builder/data/32bit/boot/config.txt

@@ -52,12 +52,16 @@ dtparam=i2c_arm=on
 dtparam=spi=on
 gpu_mem=1
 dtoverlay=dwc2
+#dtoverlay=disable-wifi
 
 [pi0]
 dtoverlay=spi1-3cs
+#dtoverlay=disable-wifi
 
 [pi3]
 dtoverlay=spi1-3cs
+#dtoverlay=disable-wifi
 
 [pi4]
 dtoverlay=spi1-3cs
+#dtoverlay=disable-wifi

builder/data/32bit/etc/profile

@@ -0,0 +1,40 @@
+# /etc/profile: system-wide .profile file for the Bourne shell (sh(1))
+# and Bourne compatible shells (bash(1), ksh(1), ash(1), ...).
+
+if [ "$(id -u)" -eq 0 ]; then
+  PATH="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin"
+else
+  PATH="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/games:/usr/games"
+fi
+export PATH
+
+if [ "${PS1-}" ]; then
+  if [ "${BASH-}" ] && [ "$BASH" != "/bin/sh" ]; then
+    # The file bash.bashrc already sets the default PS1.
+    # PS1='\h:\w\$ '
+    if [ -f /etc/bash.bashrc ]; then
+      . /etc/bash.bashrc
+    fi
+  else
+    if [ "$(id -u)" -eq 0 ]; then
+      PS1='# '
+    else
+      PS1='$ '
+    fi
+  fi
+fi
+
+if [ -d /etc/profile.d ]; then
+  for i in /etc/profile.d/*.sh; do
+    if [ -r $i ]; then
+      . $i
+    fi
+  done
+  unset i
+fi
+alias custom='cd /usr/local/share/pwnagotchi/custom-plugins/'
+alias config='sudo nano /etc/pwnagotchi/config.toml'
+alias pwnlog='tail -f -n300 /etc/pwnagotchi/log/pwn*.log | sed --unbuffered "s/,[[:digit:]]\\{3\\}\\]//g" | cut -d " " -f 2-'
+alias pwnver='python3 -c "import pwnagotchi as p; print(p.__version__)"'
+alias pwnkill='sudo killall -USR1 pwnagotchi'
+

builder/data/32bit/etc/update-motd.d/01-motd

@@ -20,6 +20,7 @@ echo "    I'm managed by systemd. Here are some basic commands."
 echo
 echo "    If you want to know what I'm doing, you can check my logs with the command"
 echo "    - pwnlog"
+echo "    - sudo pwnagotchi --wizard, to help set up a config.toml"
 echo "    - sudo pwnagotchi --version, to check the current version"
 echo "    - sudo pwnagotchi --donate, to see how you can donate to this project"
 echo "    - sudo pwnagotchi --check-update, to see if there is a new version available"
@@ -30,4 +31,4 @@ echo
 echo "    You can restart me using"
 echo "    pwnkill"
 echo
-echo "    You learn more about me at https://pwnagotchi.ai/"
+echo "    You can learn more about me at https://pwnagotchi.org/"

builder/data/64bit/boot/firmware/config.txt

@@ -51,15 +51,20 @@ dtparam=i2c_arm=on
 dtparam=spi=on
 gpu_mem=1
 dtoverlay=dwc2
+#dtoverlay=disable-wifi
 
 [pi0]
 dtoverlay=spi0-0cs
+#dtoverlay=disable-wifi
 
 [pi3]
 dtoverlay=spi0-0cs
+#dtoverlay=disable-wifi
 
 [pi4]
 dtoverlay=spi0-0cs
+#dtoverlay=disable-wifi
 
 [pi5]
 dtoverlay=spi0-0cs
+#dtoverlay=disable-wifi

builder/data/64bit/etc/profile

@@ -0,0 +1,40 @@
+# /etc/profile: system-wide .profile file for the Bourne shell (sh(1))
+# and Bourne compatible shells (bash(1), ksh(1), ash(1), ...).
+
+if [ "$(id -u)" -eq 0 ]; then
+  PATH="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin"
+else
+  PATH="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/games:/usr/games"
+fi
+export PATH
+
+if [ "${PS1-}" ]; then
+  if [ "${BASH-}" ] && [ "$BASH" != "/bin/sh" ]; then
+    # The file bash.bashrc already sets the default PS1.
+    # PS1='\h:\w\$ '
+    if [ -f /etc/bash.bashrc ]; then
+      . /etc/bash.bashrc
+    fi
+  else
+    if [ "$(id -u)" -eq 0 ]; then
+      PS1='# '
+    else
+      PS1='$ '
+    fi
+  fi
+fi
+
+if [ -d /etc/profile.d ]; then
+  for i in /etc/profile.d/*.sh; do
+    if [ -r $i ]; then
+      . $i
+    fi
+  done
+  unset i
+fi
+alias custom='cd /usr/local/share/pwnagotchi/custom-plugins/'
+alias config='sudo nano /etc/pwnagotchi/config.toml'
+alias pwnlog='tail -f -n300 /etc/pwnagotchi/log/pwn*.log | sed --unbuffered "s/,[[:digit:]]\\{3\\}\\]//g" | cut -d " " -f 2-'
+alias pwnver='python3 -c "import pwnagotchi as p; print(p.__version__)"'
+alias pwnkill='sudo killall -USR1 pwnagotchi'
+

builder/data/64bit/etc/update-motd.d/01-motd

@@ -2,7 +2,7 @@
 _hostname=$(hostname)
 _version=$(cut -d"'" -f2 < /usr/local/lib/python3.11/dist-packages/pwnagotchi/_version.py)
 echo
-echo "(☉_☉ ) $_hostname"
+echo "(◕‿‿◕) $_hostname"
 echo
 echo "    Hi! I'm a pwnagotchi $_version, please take good care of me!"
 echo "    Here are some basic things you need to know to raise me properly!"

builder/raspberrypi32.json.pkr.hcl

@@ -88,7 +88,7 @@ build {
   provisioner "ansible-local" {
     command         = "ANSIBLE_FORCE_COLOR=1 PYTHONUNBUFFERED=1 PWN_VERSION=${var.pwn_version} PWN_HOSTNAME=${var.pwn_hostname} ansible-playbook"
     extra_arguments = ["--extra-vars \"ansible_python_interpreter=/usr/bin/python3\""]
-    playbook_dir    = "data/32bit/extras/"
-    playbook_file   = "data/32bit/raspberrypi32.yml"
+    playbook_dir    = "extras/"
+    playbook_file   = "raspberrypi32.yml"
   }
 }

builder/raspberrypi32.yml

@@ -97,6 +97,7 @@
           - bc
           - bison
           - bluez
+          - bluez-tools
           - build-essential
           - curl
           - dkms
@@ -158,6 +159,7 @@
           - lsof
           - make
           - ntp
+          - python3-dbus
           - python3-flask
           - python3-flask-cors
           - python3-flaskext.wtf
@@ -369,11 +371,6 @@
         block: |
           export GOPATH=$HOME/go
           export PATH=/usr/local/go/bin:$PATH:$GOPATH/bin
-          alias custom='cd /usr/local/share/pwnagotchi/custom-plugins/'
-          alias config='sudo nano /etc/pwnagotchi/config.toml'
-          alias pwnlog='tail -f -n300 /etc/pwnagotchi/log/pwn*.log | sed --unbuffered "s/,[[:digit:]]\\{3\\}\\]//g" | cut -d " " -f 2-'
-          alias pwnver='python3 -c "import pwnagotchi as p; print(p.__version__)"'
-          alias pwnkill='sudo killall -USR1 pwnagotchi'
       when: golang.changed
 
     - name: download pwngrid

builder/raspberrypi64.json.pkr.hcl

@@ -95,6 +95,6 @@ build {
   provisioner "ansible-local" {
     command         = "ANSIBLE_FORCE_COLOR=1 PYTHONUNBUFFERED=1 PWN_VERSION=${var.pwn_version} PWN_HOSTNAME=${var.pwn_hostname} ansible-playbook"
     extra_arguments = ["--extra-vars \"ansible_python_interpreter=/usr/bin/python3\""]
-    playbook_file   = "data/64bit/raspberrypi64.yml"
+    playbook_file   = "raspberrypi64.yml"
   }
 }

builder/raspberrypi64.yml

@@ -64,6 +64,7 @@
           - bc
           - bison
           - bluez
+          - bluez-tools
           - build-essential
           - curl
           - dkms
@@ -481,16 +482,11 @@
 
     - name: Update .bashrc for go-1.21
       blockinfile:
-        dest: /home/pi/.bashrc
+        dest: /etc/profile
         state: present
         block: |
           export GOPATH=$HOME/go
           export PATH=/usr/local/go/bin:$PATH:$GOPATH/bin
-          alias custom='cd /usr/local/share/pwnagotchi/custom-plugins/'
-          alias config='sudo nano /etc/pwnagotchi/config.toml'
-          alias pwnlog='tail -f -n300 /etc/pwnagotchi/log/pwn*.log | sed --unbuffered "s/,[[:digit:]]\\{3\\}\\]//g" | cut -d " " -f 2-'
-          alias pwnver='python3 -c "import pwnagotchi as p; print(p.__version__)"'
-          alias pwnkill='sudo killall -USR1 pwnagotchi'
       when: golang.changed
 
     - name: download pwngrid

pwnagotchi/_version.py

@@ -1 +1 @@
-__version__ = '2.8.7.2'
+__version__ = '2.8.9'

pwnagotchi/agent.py

@@ -254,7 +254,7 @@ class Agent(Client, Automata, AsyncAdvertiser, AsyncTrainer):
         txt = '%d (%d)' % (len(self._handshakes), tot)
 
         if self._last_pwnd is not None:
-            txt += ' [%s]' % self._last_pwnd[:11]  # So it doesn't overlap with fix_brcmfmac_plugin
+            txt += ' [%s]' % self._last_pwnd
 
         self._view.set('shakes', txt)
 

pwnagotchi/bettercap.py

@@ -100,7 +100,6 @@ class Client(object):
                 await asyncio.sleep(sleep_time)
                 continue
             except OSError:
-                sleep_time = min_sleep + max_sleep * random.random()
                 logging.warning('connection to the bettercap endpoint failed...')
                 pwnagotchi.restart("AUTO")
 

pwnagotchi/defaults.toml

@@ -112,8 +112,6 @@ main.mon_stop_cmd = "/usr/bin/monstop"
 main.mon_max_blind_epochs = 50
 main.no_restart = false
 
-main.filter = ""
-
 main.log.path = "/etc/pwnagotchi/log/pwnagotchi.log"
 main.log.rotation.enabled = true
 main.log.rotation.size = "10M"

pwnagotchi/locale/fy/LC_MESSAGES/voice.po

@@ -248,7 +248,7 @@ msgid "minutes"
 msgstr "minuten"
 
 msgid "seconds"
-msgstr "seconds"
+msgstr "sekondes"
 
 msgid "hour"
 msgstr "oere"

pwnagotchi/plugins/default/auto-update.py

@@ -230,7 +230,7 @@ class AutoUpdate(plugins.Plugin):
                 if num_installed > 0:
                     display.update(force=True, new_data={'status': 'Rebooting ...'})
                     time.sleep(3)
-                    os.system("service pwnagotchi restart")
+                    pwnagotchi.reboot()
 
             except Exception as e:
                 logging.error("[update] %s" % e)

pwnagotchi/plugins/default/fix_services.py

@@ -4,6 +4,7 @@ import subprocess
 import time
 import random
 from io import TextIOWrapper
+import os
 
 import pwnagotchi
 from pwnagotchi import plugins
@@ -32,6 +33,8 @@ class FixServices(plugins.Plugin):
         self.pattern2 = re.compile(r'wifi error while hopping to channel')
         self.pattern3 = re.compile(r'Firmware has halted or crashed')
         self.pattern4 = re.compile(r'error 400: could not find interface wlan0mon')
+        self.pattern5 = re.compile(r'fatal error: concurrent map iteration and map write')
+        self.pattern6 = re.compile(r'panic: runtime error')
         self.isReloadingMon = False
         self.connection = None
         self.LASTTRY = 0
@@ -177,6 +180,25 @@ class FixServices(plugins.Plugin):
                 except Exception as err:
                     logging.error("[Fix_Services monstart]: %s" % repr(err))
 
+            # Look for pattern 5
+            elif len(self.pattern5.findall(other_other_last_lines)) >= 1:
+                logging.debug("[Fix_Services] Bettercap has crashed!")
+                if hasattr(agent, 'view'):
+                    display = agent.view()
+                    display.set('status', 'Restarting pwnagotchi!')
+                    display.update(force=True)
+                os.system("systemctl restart bettercap")
+                pwnagotchi.restart("AUTO")
+
+            # Look for pattern 6
+            elif len(self.pattern6.findall(other_other_last_lines)) >= 1:
+                logging.debug("[Fix_Services] Bettercap has crashed!")
+                if hasattr(agent, 'view'):
+                    display = agent.view()
+                    display.set('status', 'Restarting pwnagotchi!')
+                    display.update(force=True)
+                os.system("systemctl restart bettercap")
+                pwnagotchi.restart("AUTO")
             else:
                 print("logs look good")
 

pwnagotchi/plugins/default/logtail.py

@@ -253,7 +253,6 @@ class Logtail(plugins.Plugin):
         """
         logging.info("Logtail plugin loaded.")
 
-
     def on_webhook(self, path, request):
         if not self.ready:
             return "Plugin not ready"

pwnagotchi/ui/display.py

@@ -28,6 +28,39 @@ class Display(View):
     def is_lcdhat(self):
         return self._implementation.name == 'lcdhat'
 
+    def is_wavesharelcd0in96(self):
+        return self._implementation.name == 'wavesharelcd0in96'
+
+    def is_wavesharelcd1in3(self):
+        return self._implementation.name == 'wavesharelcd1in3'
+
+    def is_wavesharelcd1in8(self):
+        return self._implementation.name == 'wavesharelcd1in8'
+
+    def is_wavesharelcd1in9(self):
+        return self._implementation.name == 'wavesharelcd1in9'
+
+    def is_wavesharelcd1in14(self):
+        return self._implementation.name == 'wavesharelcd1in14'
+
+    def is_wavesharelcd1in28(self):
+        return self._implementation.name == 'wavesharelcd1in28'
+
+    def is_wavesharelcd1in47(self):
+        return self._implementation.name == 'wavesharelcd1in47'
+
+    def is_wavesharelcd1in54(self):
+        return self._implementation.name == 'wavesharelcd1in54'
+
+    def is_wavesharelcd1in69(self):
+        return self._implementation.name == 'wavesharelcd1in69'
+
+    def is_wavesharelcd2in0(self):
+        return self._implementation.name == 'wavesharelcd2in0'
+
+    def is_wavesharelcd2in4(self):
+        return self._implementation.name == 'wavesharelcd2in4'
+
     def is_waveshare144lcd(self):
         return self._implementation.name == 'waveshare144lcd'
 
@@ -202,6 +235,9 @@ class Display(View):
     def is_inky(self):
         return self._implementation.name == 'inky'
 
+    def is_dummy_display(self):
+        return self._implementation.name == 'dummydisplay'
+
     def is_papirus(self):
         return self._implementation.name == 'papirus'
 
@@ -232,6 +268,9 @@ class Display(View):
     def is_waveshare35lcd(self):
         return self._implementation.name == 'waveshare35lcd'
 
+    def is_adfruit213v3(self):
+        return self._implementation.name == 'adafruit2in13_v3'
+
     def is_waveshare_any(self):
         return self.is_waveshare_v1() or self.is_waveshare_v2()
 

pwnagotchi/ui/faces.py

@@ -27,6 +27,7 @@ PNG = False
 POSITION_X = 0
 POSITION_Y = 40
 
+
 def load_from_config(config):
     for face_name, face_value in config.items():
         globals()[face_name.upper()] = face_value

pwnagotchi/ui/hw/__init__.py

@@ -1,283 +1,333 @@
-from pwnagotchi.ui.hw.inky import Inky
-from pwnagotchi.ui.hw.papirus import Papirus
-from pwnagotchi.ui.hw.oledhat import OledHat
-from pwnagotchi.ui.hw.lcdhat import LcdHat
-from pwnagotchi.ui.hw.dfrobot import DFRobotV1
-from pwnagotchi.ui.hw.dfrobot_v2 import DFRobotV2
-from pwnagotchi.ui.hw.waveshare2in13 import WaveshareV1
-from pwnagotchi.ui.hw.waveshare2in13_V2 import WaveshareV2
-from pwnagotchi.ui.hw.waveshare2in13_V3 import WaveshareV3
-from pwnagotchi.ui.hw.waveshare2in13_V4 import WaveshareV4
-from pwnagotchi.ui.hw.waveshare2in7 import Waveshare27inch
-from pwnagotchi.ui.hw.waveshare2in7_V2 import Waveshare27inchV2
-from pwnagotchi.ui.hw.waveshare2in9 import Waveshare29inch
-from pwnagotchi.ui.hw.waveshare2in9_V2 import Waveshare29inchV2
-from pwnagotchi.ui.hw.waveshare1in44lcd import Waveshare144lcd
-from pwnagotchi.ui.hw.waveshare1in54b import Waveshare154inchb
-from pwnagotchi.ui.hw.waveshare2in13bc import Waveshare213bc
-from pwnagotchi.ui.hw.waveshare2in13d import Waveshare213d
-from pwnagotchi.ui.hw.waveshare2in13g import Waveshare2in13g
-from pwnagotchi.ui.hw.waveshare2in13b_V4 import Waveshare213bV4
-from pwnagotchi.ui.hw.waveshare3in5lcd import Waveshare35lcd
-from pwnagotchi.ui.hw.spotpear24in import Spotpear24inch
-from pwnagotchi.ui.hw.displayhatmini import DisplayHatMini
-from pwnagotchi.ui.hw.pirateaudio import PirateAudio
-from pwnagotchi.ui.hw.pitft import Pitft
-from pwnagotchi.ui.hw.tftbonnet import TftBonnet
-from pwnagotchi.ui.hw.waveshareoledlcd import Waveshareoledlcd
-from pwnagotchi.ui.hw.waveshare1in02 import Waveshare1in02
-from pwnagotchi.ui.hw.waveshare1in54 import Waveshare154
-from pwnagotchi.ui.hw.waveshare1in54_V2 import Waveshare154V2
-from pwnagotchi.ui.hw.waveshare1in54b_V2 import Waveshare154bV2
-from pwnagotchi.ui.hw.waveshare1in54c import Waveshare1in54c
-from pwnagotchi.ui.hw.waveshare1in64g import Waveshare1in64g
-from pwnagotchi.ui.hw.waveshare2in7b import Waveshare27b
-from pwnagotchi.ui.hw.waveshare2in7b_V2 import Waveshare27bV2
-from pwnagotchi.ui.hw.waveshare2in9b_V3 import Waveshare29bV3
-from pwnagotchi.ui.hw.waveshare2in9b_V4 import Waveshare29bV4
-from pwnagotchi.ui.hw.waveshare2in9bc import Waveshare2in9bc
-from pwnagotchi.ui.hw.waveshare2in9d import Waveshare2in9d
-from pwnagotchi.ui.hw.waveshare2in13b_V3 import Waveshare2in13bV3
-from pwnagotchi.ui.hw.waveshare2in36g import Waveshare2in36g
-from pwnagotchi.ui.hw.waveshare2in66 import Waveshare2in66
-from pwnagotchi.ui.hw.waveshare2in66b import Waveshare2in66b
-from pwnagotchi.ui.hw.waveshare2in66g import Waveshare2in66g
-from pwnagotchi.ui.hw.waveshare3in0g import Waveshare3in0g
-from pwnagotchi.ui.hw.waveshare3in7 import Waveshare3in7
-from pwnagotchi.ui.hw.waveshare3in52 import Waveshare3in52
-from pwnagotchi.ui.hw.waveshare4in01f import Waveshare4in01f
-from pwnagotchi.ui.hw.waveshare4in2 import Waveshare4in2
-from pwnagotchi.ui.hw.waveshare4in2_V2 import Waveshare4in2V2
-from pwnagotchi.ui.hw.waveshare4in2b_V2 import Waveshare4in2bV2
-from pwnagotchi.ui.hw.waveshare4in2bc import Waveshare4in2bc
-from pwnagotchi.ui.hw.waveshare4in26 import Waveshare4in26
-from pwnagotchi.ui.hw.waveshare4in37g import Waveshare4in37g
-from pwnagotchi.ui.hw.waveshare5in65f import Waveshare5in65f
-from pwnagotchi.ui.hw.waveshare5in79 import Waveshare5in79
-from pwnagotchi.ui.hw.waveshare5in79b import Waveshare5in79b
-from pwnagotchi.ui.hw.waveshare5in83 import Waveshare5in83
-from pwnagotchi.ui.hw.waveshare5in83_V2 import Waveshare5in83V2
-from pwnagotchi.ui.hw.waveshare5in83b_V2 import Waveshare5in83bV2
-from pwnagotchi.ui.hw.waveshare5in83bc import Waveshare5in83bc
-from pwnagotchi.ui.hw.waveshare7in3f import Waveshare7in3f
-from pwnagotchi.ui.hw.waveshare7in3g import Waveshare7in3g
-from pwnagotchi.ui.hw.waveshare7in5 import Waveshare7in5
-from pwnagotchi.ui.hw.waveshare7in5_HD import Waveshare7in5HD
-from pwnagotchi.ui.hw.waveshare7in5_V2 import Waveshare7in5V2
-from pwnagotchi.ui.hw.waveshare7in5b_HD import Waveshare7in5bHD
-from pwnagotchi.ui.hw.waveshare7in5b_V2 import Waveshare7in5bV2
-from pwnagotchi.ui.hw.waveshare7in5bc import Waveshare7in5bc
-from pwnagotchi.ui.hw.waveshare13in3k import Waveshare13in3k
-
-
 def display_for(config):
     # config has been normalized already in utils.load_config
     if config['ui']['display']['type'] == 'inky':
+        from pwnagotchi.ui.hw.inky import Inky
         return Inky(config)
 
+    elif config['ui']['display']['type'] == 'wavesharelcd0in96':
+        from pwnagotchi.ui.hw.wavesharelcd0in96 import Wavesharelcd0in96
+        return Wavesharelcd0in96(config)
+
+    elif config['ui']['display']['type'] == 'wavesharelcd1in3':
+        from pwnagotchi.ui.hw.wavesharelcd1in3 import Wavesharelcd1in3
+        return Wavesharelcd1in3(config)
+
+    elif config['ui']['display']['type'] == 'wavesharelcd1in8':
+        from pwnagotchi.ui.hw.wavesharelcd1in8 import Wavesharelcd1in8
+        return Wavesharelcd1in8(config)
+
+    elif config['ui']['display']['type'] == 'wavesharelcd1in9':
+        from pwnagotchi.ui.hw.wavesharelcd1in9 import Wavesharelcd1in9
+        return Wavesharelcd1in9(config)
+
+    elif config['ui']['display']['type'] == 'wavesharelcd1in14':
+        from pwnagotchi.ui.hw.wavesharelcd1in14 import Wavesharelcd1in14
+        return Wavesharelcd1in14(config)
+
+    elif config['ui']['display']['type'] == 'wavesharelcd1in28':
+        from pwnagotchi.ui.hw.wavesharelcd1in28 import Wavesharelcd1in28
+        return Wavesharelcd1in28(config)
+
+    elif config['ui']['display']['type'] == 'wavesharelcd1in47':
+        from pwnagotchi.ui.hw.wavesharelcd1in47 import Wavesharelcd1in47
+        return Wavesharelcd1in47(config)
+
+    elif config['ui']['display']['type'] == 'wavesharelcd1in54':
+        from pwnagotchi.ui.hw.wavesharelcd1in54 import Wavesharelcd1in54
+        return Wavesharelcd1in54(config)
+
+    elif config['ui']['display']['type'] == 'wavesharelcd1in69':
+        from pwnagotchi.ui.hw.wavesharelcd1in69 import Wavesharelcd1in69
+        return Wavesharelcd1in69(config)
+
+    elif config['ui']['display']['type'] == 'wavesharelcd2in0':
+        from pwnagotchi.ui.hw.wavesharelcd2in0 import Wavesharelcd2in0
+        return Wavesharelcd2in0(config)
+
+    elif config['ui']['display']['type'] == 'wavesharelcd2in4':
+        from pwnagotchi.ui.hw.wavesharelcd2in4 import Wavesharelcd2in4
+        return Wavesharelcd2in4(config)
+
+    elif config['ui']['display']['type'] == 'dummydisplay':
+        from pwnagotchi.ui.hw.dummydisplay import DummyDisplay
+        return DummyDisplay(config)
+
     elif config['ui']['display']['type'] == 'papirus':
+        from pwnagotchi.ui.hw.papirus import Papirus
         return Papirus(config)
 
     elif config['ui']['display']['type'] == 'oledhat':
+        from pwnagotchi.ui.hw.oledhat import OledHat
         return OledHat(config)
 
     elif config['ui']['display']['type'] == 'lcdhat':
+        from pwnagotchi.ui.hw.lcdhat import LcdHat
         return LcdHat(config)
 
     elif config['ui']['display']['type'] == 'dfrobot_1':
+        from pwnagotchi.ui.hw.dfrobot import DFRobotV1
         return DFRobotV1(config)
 
     elif config['ui']['display']['type'] == 'dfrobot_2':
+        from pwnagotchi.ui.hw.dfrobot_v2 import DFRobotV2
         return DFRobotV2(config)
 
     elif config['ui']['display']['type'] == 'waveshare144lcd':
+        from pwnagotchi.ui.hw.waveshare1in44lcd import Waveshare144lcd
         return Waveshare144lcd(config)
 
     elif config['ui']['display']['type'] == 'waveshare35lcd':
+        from pwnagotchi.ui.hw.waveshare3in5lcd import Waveshare35lcd
         return Waveshare35lcd(config)
 
     elif config['ui']['display']['type'] == 'spotpear24inch':
+        from pwnagotchi.ui.hw.spotpear24in import Spotpear24inch
         return Spotpear24inch(config)
 
     elif config['ui']['display']['type'] == 'displayhatmini':
+        from pwnagotchi.ui.hw.displayhatmini import DisplayHatMini
         return DisplayHatMini(config)
-  
+
     elif config['ui']['display']['type'] == 'pirateaudio':
+        from pwnagotchi.ui.hw.pirateaudio import PirateAudio
         return PirateAudio(config)
-        
+
     elif config['ui']['display']['type'] == 'pitft':
+        from pwnagotchi.ui.hw.pitft import Pitft
         return Pitft(config)
 
     elif config['ui']['display']['type'] == 'tftbonnet':
+        from pwnagotchi.ui.hw.tftbonnet import TftBonnet
         return TftBonnet(config)
 
     elif config['ui']['display']['type'] == 'waveshareoledlcd':
+        from pwnagotchi.ui.hw.waveshareoledlcd import Waveshareoledlcd
         return Waveshareoledlcd(config)
-        
+
     elif config['ui']['display']['type'] == 'waveshare1in02':
+        from pwnagotchi.ui.hw.waveshare1in02 import Waveshare1in02
         return Waveshare1in02(config)
 
     elif config['ui']['display']['type'] == 'waveshare1in54':
+        from pwnagotchi.ui.hw.waveshare1in54 import Waveshare154
         return Waveshare154(config)
 
     elif config['ui']['display']['type'] == 'waveshare1in54_v2':
+        from pwnagotchi.ui.hw.waveshare1in54_V2 import Waveshare154V2
         return Waveshare154V2(config)
 
     elif config['ui']['display']['type'] == 'waveshare1in54b':
+        from pwnagotchi.ui.hw.waveshare1in54b import Waveshare154inchb
         return Waveshare154inchb(config)
 
     elif config['ui']['display']['type'] == 'waveshare1in54b_v2':
+        from pwnagotchi.ui.hw.waveshare1in54b_V2 import Waveshare154bV2
         return Waveshare154bV2(config)
 
     elif config['ui']['display']['type'] == 'waveshare1in54c':
+        from pwnagotchi.ui.hw.waveshare1in54c import Waveshare1in54c
         return Waveshare1in54c(config)
 
     elif config['ui']['display']['type'] == 'waveshare1in64g':
+        from pwnagotchi.ui.hw.waveshare1in64g import Waveshare1in64g
         return Waveshare1in64g(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in7':
+        from pwnagotchi.ui.hw.waveshare2in7 import Waveshare27inch
         return Waveshare27inch(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in7_v2':
+        from pwnagotchi.ui.hw.waveshare2in7_V2 import Waveshare27inchV2
         return Waveshare27inchV2(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in7b':
+        from pwnagotchi.ui.hw.waveshare2in7b import Waveshare27b
         return Waveshare27b(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in7b_v2':
+        from pwnagotchi.ui.hw.waveshare2in7b_V2 import Waveshare27bV2
         return Waveshare27bV2(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in9':
+        from pwnagotchi.ui.hw.waveshare2in9 import Waveshare29inch
         return Waveshare29inch(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in9bc':
+        from pwnagotchi.ui.hw.waveshare2in9bc import Waveshare2in9bc
         return Waveshare2in9bc(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in9d':
+        from pwnagotchi.ui.hw.waveshare2in9d import Waveshare2in9d
         return Waveshare2in9d(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in9_v2':
+        from pwnagotchi.ui.hw.waveshare2in9_V2 import Waveshare29inchV2
         return Waveshare29inchV2(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in9b_v3':
+        from pwnagotchi.ui.hw.waveshare2in9b_V3 import Waveshare29bV3
         return Waveshare29bV3(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in9b_v4':
+        from pwnagotchi.ui.hw.waveshare2in9b_V4 import Waveshare29bV4
         return Waveshare29bV4(config)
 
     elif config['ui']['display']['type'] == 'waveshare_1':
+        from pwnagotchi.ui.hw.waveshare2in13 import WaveshareV1
         return WaveshareV1(config)
 
     elif config['ui']['display']['type'] == 'waveshare_2':
+        from pwnagotchi.ui.hw.waveshare2in13_V2 import WaveshareV2
         return WaveshareV2(config)
 
     elif config['ui']['display']['type'] == 'waveshare_3':
+        from pwnagotchi.ui.hw.waveshare2in13_V3 import WaveshareV3
         return WaveshareV3(config)
 
     elif config['ui']['display']['type'] == 'waveshare_4':
+        from pwnagotchi.ui.hw.waveshare2in13_V4 import WaveshareV4
         return WaveshareV4(config)
 
+    elif config['ui']['display']['type'] == 'adafruit2in13_v3':
+        from pwnagotchi.ui.hw.adafruit2in13 import Adafruit2in13V3
+        return Adafruit2in13V3(config)
+
     elif config['ui']['display']['type'] == 'waveshare2in13bc':
+        from pwnagotchi.ui.hw.waveshare2in13bc import Waveshare213bc
         return Waveshare213bc(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in13d':
+        from pwnagotchi.ui.hw.waveshare2in13d import Waveshare213d
         return Waveshare213d(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in13b_v3':
+        from pwnagotchi.ui.hw.waveshare2in13b_V3 import Waveshare2in13bV3
         return Waveshare2in13bV3(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in13b_v4':
+        from pwnagotchi.ui.hw.waveshare2in13b_V4 import Waveshare213bV4
         return Waveshare213bV4(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in13g':
+        from pwnagotchi.ui.hw.waveshare2in13g import Waveshare2in13g
         return Waveshare2in13g(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in36g':
+        from pwnagotchi.ui.hw.waveshare2in36g import Waveshare2in36g
         return Waveshare2in36g(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in66':
+        from pwnagotchi.ui.hw.waveshare2in66 import Waveshare2in66
         return Waveshare2in66(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in66b':
+        from pwnagotchi.ui.hw.waveshare2in66b import Waveshare2in66b
         return Waveshare2in66b(config)
 
     elif config['ui']['display']['type'] == 'waveshare2in66g':
+        from pwnagotchi.ui.hw.waveshare2in66g import Waveshare2in66g
         return Waveshare2in66g(config)
 
     elif config['ui']['display']['type'] == 'waveshare3in0g':
+        from pwnagotchi.ui.hw.waveshare3in0g import Waveshare3in0g
         return Waveshare3in0g(config)
 
     elif config['ui']['display']['type'] == 'waveshare3in7':
+        from pwnagotchi.ui.hw.waveshare3in7 import Waveshare3in7
         return Waveshare3in7(config)
 
     elif config['ui']['display']['type'] == 'waveshare3in52':
+        from pwnagotchi.ui.hw.waveshare3in52 import Waveshare3in52
         return Waveshare3in52(config)
 
     elif config['ui']['display']['type'] == 'waveshare4in01f':
+        from pwnagotchi.ui.hw.waveshare4in01f import Waveshare4in01f
         return Waveshare4in01f(config)
 
     elif config['ui']['display']['type'] == 'waveshare4in2':
+        from pwnagotchi.ui.hw.waveshare4in2 import Waveshare4in2
         return Waveshare4in2(config)
 
     elif config['ui']['display']['type'] == 'waveshare4in2_v2':
+        from pwnagotchi.ui.hw.waveshare4in2_V2 import Waveshare4in2V2
         return Waveshare4in2V2(config)
 
     elif config['ui']['display']['type'] == 'waveshare4in2b_v2':
+        from pwnagotchi.ui.hw.waveshare4in2b_V2 import Waveshare4in2bV2
         return Waveshare4in2bV2(config)
 
     elif config['ui']['display']['type'] == 'waveshare4in2bc':
+        from pwnagotchi.ui.hw.waveshare4in2bc import Waveshare4in2bc
         return Waveshare4in2bc(config)
 
     elif config['ui']['display']['type'] == 'waveshare4in26':
+        from pwnagotchi.ui.hw.waveshare4in26 import Waveshare4in26
         return Waveshare4in26(config)
 
     elif config['ui']['display']['type'] == 'waveshare4in37g':
+        from pwnagotchi.ui.hw.waveshare4in37g import Waveshare4in37g
         return Waveshare4in37g(config)
 
     elif config['ui']['display']['type'] == 'waveshare5in65f':
+        from pwnagotchi.ui.hw.waveshare5in65f import Waveshare5in65f
         return Waveshare5in65f(config)
 
     elif config['ui']['display']['type'] == 'waveshare5in79':
+        from pwnagotchi.ui.hw.waveshare5in79 import Waveshare5in79
         return Waveshare5in79(config)
 
     elif config['ui']['display']['type'] == 'waveshare5in79b':
+        from pwnagotchi.ui.hw.waveshare5in79b import Waveshare5in79b
         return Waveshare5in79b(config)
 
     elif config['ui']['display']['type'] == 'waveshare5in83':
+        from pwnagotchi.ui.hw.waveshare5in83 import Waveshare5in83
         return Waveshare5in83(config)
 
     elif config['ui']['display']['type'] == 'waveshare5in83_v2':
+        from pwnagotchi.ui.hw.waveshare5in83_V2 import Waveshare5in83V2
         return Waveshare5in83V2(config)
 
     elif config['ui']['display']['type'] == 'waveshare5in83b_v2':
+        from pwnagotchi.ui.hw.waveshare5in83b_V2 import Waveshare5in83bV2
         return Waveshare5in83bV2(config)
 
     elif config['ui']['display']['type'] == 'waveshare5in83bc':
+        from pwnagotchi.ui.hw.waveshare5in83bc import Waveshare5in83bc
         return Waveshare5in83bc(config)
 
     elif config['ui']['display']['type'] == 'waveshare7in3f':
+        from pwnagotchi.ui.hw.waveshare7in3f import Waveshare7in3f
         return Waveshare7in3f(config)
 
     elif config['ui']['display']['type'] == 'waveshare7in3g':
+        from pwnagotchi.ui.hw.waveshare7in3g import Waveshare7in3g
         return Waveshare7in3g(config)
 
     elif config['ui']['display']['type'] == 'waveshare7in5':
+        from pwnagotchi.ui.hw.waveshare7in5 import Waveshare7in5
         return Waveshare7in5(config)
 
     elif config['ui']['display']['type'] == 'waveshare7in5_HD':
+        from pwnagotchi.ui.hw.waveshare7in5_HD import Waveshare7in5HD
         return Waveshare7in5HD(config)
 
     elif config['ui']['display']['type'] == 'waveshare7in5_v2':
+        from pwnagotchi.ui.hw.waveshare7in5_V2 import Waveshare7in5V2
         return Waveshare7in5V2(config)
 
     elif config['ui']['display']['type'] == 'waveshare7in5b_HD':
+        from pwnagotchi.ui.hw.waveshare7in5b_HD import Waveshare7in5bHD
         return Waveshare7in5bHD(config)
 
     elif config['ui']['display']['type'] == 'waveshare7in5b_v2':
+        from pwnagotchi.ui.hw.waveshare7in5b_V2 import Waveshare7in5bV2
         return Waveshare7in5bV2(config)
 
     elif config['ui']['display']['type'] == 'waveshare7in5bc':
+        from pwnagotchi.ui.hw.waveshare7in5bc import Waveshare7in5bc
         return Waveshare7in5bc(config)
 
     elif config['ui']['display']['type'] == 'waveshare13in3k':
+        from pwnagotchi.ui.hw.waveshare13in3k import Waveshare13in3k
         return Waveshare13in3k(config)

pwnagotchi/ui/hw/adafruit2in13.py

@@ -0,0 +1,45 @@
+import logging
+
+import pwnagotchi.ui.fonts as fonts
+from pwnagotchi.ui.hw.base import DisplayImpl
+
+
+class Adafruit2in13V3(DisplayImpl):
+    def __init__(self, config):
+        super(Adafruit2in13V3, self).__init__(config, 'adafruit2in13_v3')
+
+    def layout(self):
+        fonts.setup(10, 9, 10, 35, 25, 9)
+        self._layout['width'] = 250
+        self._layout['height'] = 122
+        self._layout['face'] = (0, 40)
+        self._layout['name'] = (5, 20)
+        self._layout['channel'] = (0, 0)
+        self._layout['aps'] = (28, 0)
+        self._layout['uptime'] = (185, 0)
+        self._layout['line1'] = [0, 14, 250, 14]
+        self._layout['line2'] = [0, 108, 250, 108]
+        self._layout['friend_face'] = (0, 92)
+        self._layout['friend_name'] = (40, 94)
+        self._layout['shakes'] = (0, 109)
+        self._layout['mode'] = (225, 109)
+        self._layout['status'] = {
+            'pos': (125, 20),
+            'font': fonts.status_font(fonts.Medium),
+            'max': 20
+        }
+        return self._layout
+
+    def initialize(self):
+        logging.info("initializing adafruit 2in13 V3 display")
+        from pwnagotchi.ui.hw.libs.adafruit.v2in13_v3.epd2in13_v3 import EPD
+        self._display = EPD()
+        self._display.init()
+        self._display.Clear(0xFF)
+
+    def render(self, canvas):
+        buf = self._display.getbuffer(canvas)
+        self._display.displayPartial(buf)
+
+    def clear(self):
+        self._display.Clear(0xFF)

pwnagotchi/ui/hw/displayhatmini.py

@@ -34,10 +34,10 @@ class DisplayHatMini(DisplayImpl):
     def initialize(self):
         logging.info("initializing Display Hat Mini")
         from pwnagotchi.ui.hw.libs.pimoroni.displayhatmini.ST7789 import ST7789
-        self._display = ST7789(0,1,9,13)
+        self._display = ST7789(0, 1, 9, 13)
 
     def render(self, canvas):
         self._display.display(canvas)
 
     def clear(self):
-        self._display.clear()
+        self._display.clear()

pwnagotchi/ui/hw/dummydisplay.py

@@ -0,0 +1,43 @@
+import logging
+
+import pwnagotchi.ui.fonts as fonts
+from pwnagotchi.ui.hw.base import DisplayImpl
+
+
+class DummyDisplay(DisplayImpl):
+    def __init__(self, config):
+        super(DummyDisplay, self).__init__(config, 'DummyDisplay')
+
+    def layout(self):
+        width = 480 if 'width' not in self.config else self.config['width']
+        height = 720 if 'height' not in self.config else self.config['height']
+        fonts.setup(int(height/30), int(height/40), int(height/30), int(height/6), int(height/30), int(height/35))
+        self._layout['width'] = width
+        self._layout['height'] = height
+        self._layout['face'] = (0, int(width/12))
+        self._layout['name'] = (5, int(width/25))
+        self._layout['channel'] = (0, 0)
+        self._layout['aps'] = (int(width/8), 0)
+        self._layout['uptime'] = (width-int(width/12), 0)
+        self._layout['line1'] = [0, int(height/32), width, int(height/32)]
+        self._layout['line2'] = [0, height-int(height/25)-1, width, height-int(height/25)-1]
+        self._layout['friend_face'] = (0, int(height/10))
+        self._layout['friend_name'] = (int(width/12), int(height/10))
+        self._layout['shakes'] = (0, height-int(height/25))
+        self._layout['mode'] = (width-int(width/8), height - int (height/25))
+        lw, lh = fonts.Small.getsize("W")
+        self._layout['status'] = {
+            'pos': (int(width/48), int(height/3)),
+            'font': fonts.status_font(fonts.Small),
+            'max': int(width / lw)
+        }
+        return self._layout
+
+    def initialize(self):
+        return
+
+    def render(self, canvas):
+        return
+
+    def clear(self):
+        return

pwnagotchi/ui/hw/lcdhat.py

@@ -33,7 +33,7 @@ class LcdHat(DisplayImpl):
 
     def initialize(self):
         logging.info("initializing lcdhat display")
-        from pwnagotchi.ui.hw.libs.waveshare.lcdhat.epd import EPD
+        from pwnagotchi.ui.hw.libs.waveshare.lcd.lcdhat.epd import EPD
         self._display = EPD()
         self._display.init()
         self._display.clear()

pwnagotchi/ui/hw/libs/adafruit/epdconfig.py

@@ -0,0 +1,132 @@
+# /*****************************************************************************
+# * | File        :	  epdconfig.py
+# * | Author      :   Waveshare team
+# * | Function    :   Hardware underlying interface
+# * | Info        :
+# *----------------
+# * | This version:   V1.2
+# * | Date        :   2022-10-29
+# * | Info        :
+# ******************************************************************************
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documnetation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to  whom the Software is
+# furished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS OR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+
+import os
+import logging
+import sys
+import time
+import subprocess
+
+logger = logging.getLogger(__name__)
+
+
+class RaspberryPi:
+    # Pin definition
+    RST_PIN = 27
+    DC_PIN = 22
+    CS_PIN = 8
+    BUSY_PIN = 17
+    PWR_PIN = 18
+
+    def __init__(self):
+        import spidev
+        import gpiozero
+
+        self.SPI = spidev.SpiDev()
+        self.GPIO_RST_PIN = gpiozero.LED(self.RST_PIN)
+        self.GPIO_DC_PIN = gpiozero.LED(self.DC_PIN)
+        self.GPIO_CS_PIN = gpiozero.LED(self.CS_PIN)
+        self.GPIO_PWR_PIN = gpiozero.LED(self.PWR_PIN)
+        self.GPIO_BUSY_PIN = gpiozero.Button(self.BUSY_PIN, pull_up=False)
+
+    def digital_write(self, pin, value):
+        if pin == self.RST_PIN:
+            if value:
+                self.GPIO_RST_PIN.on()
+            else:
+                self.GPIO_RST_PIN.off()
+        elif pin == self.DC_PIN:
+            if value:
+                self.GPIO_DC_PIN.on()
+            else:
+                self.GPIO_DC_PIN.off()
+        elif pin == self.CS_PIN:
+            if value:
+                self.GPIO_CS_PIN.on()
+            else:
+                self.GPIO_CS_PIN.off()
+        elif pin == self.PWR_PIN:
+            if value:
+                self.GPIO_PWR_PIN.on()
+            else:
+                self.GPIO_PWR_PIN.off()
+
+    def digital_read(self, pin):
+        if pin == self.BUSY_PIN:
+            return self.GPIO_BUSY_PIN.value
+        elif pin == self.RST_PIN:
+            return self.RST_PIN.value
+        elif pin == self.DC_PIN:
+            return self.DC_PIN.value
+        elif pin == self.CS_PIN:
+            return self.CS_PIN.value
+        elif pin == self.PWR_PIN:
+            return self.PWR_PIN.value
+
+    def delay_ms(self, delaytime):
+        time.sleep(delaytime / 1000.0)
+
+    def spi_writebyte(self, data):
+        self.SPI.writebytes(data)
+
+    def spi_writebyte2(self, data):
+        self.SPI.writebytes2(data)
+
+    def module_init(self):
+        self.GPIO_PWR_PIN.on()
+
+        # SPI device, bus = 0, device = 0
+        self.SPI.open(0, 0)
+        self.SPI.max_speed_hz = 4000000
+        self.SPI.mode = 0b00
+        return 0
+
+    def module_exit(self, cleanup=False):
+        logger.debug("spi end")
+        self.SPI.close()
+
+        self.GPIO_RST_PIN.off()
+        self.GPIO_DC_PIN.off()
+        self.GPIO_PWR_PIN.off()
+        logger.debug("close 5V, Module enters 0 power consumption ...")
+
+        if cleanup:
+            self.GPIO_RST_PIN.close()
+            self.GPIO_DC_PIN.close()
+            self.GPIO_CS_PIN.close()
+            self.GPIO_PWR_PIN.close()
+            self.GPIO_BUSY_PIN.close()
+
+
+implementation = RaspberryPi()
+
+for func in [x for x in dir(implementation) if not x.startswith('_')]:
+    setattr(sys.modules[__name__], func, getattr(implementation, func))
+
+### END OF FILE ###

pwnagotchi/ui/hw/libs/adafruit/pitft/ILI9341.py

@@ -93,6 +93,7 @@ ILI9341_GMCTRN1 = 0xE1
 
 ILI9341_PWCTR6 = 0xFC
 
+
 class ILI9341(object):
     """Representation of an ILI9341 TFT LCD."""
 
@@ -349,7 +350,6 @@ class ILI9341(object):
         # Rotate the image
         pb = np.rot90(image, rotation // 90).astype('uint16')
 
-
         # Mask and shift the 888 RGB into 565 RGB
         red   = (pb[..., [0]] & 0xf8) << 8
         green = (pb[..., [1]] & 0xfc) << 3

pwnagotchi/ui/hw/libs/fb/fb.py

@@ -1,144 +1,164 @@
-FBIOGET_VSCREENINFO=0x4600
-FBIOPUT_VSCREENINFO=0x4601
-FBIOGET_FSCREENINFO=0x4602
-FBIOGETCMAP=0x4604
-FBIOPUTCMAP=0x4605
-FBIOPAN_DISPLAY=0x4606
+FBIOGET_VSCREENINFO = 0x4600
+FBIOPUT_VSCREENINFO = 0x4601
+FBIOGET_FSCREENINFO = 0x4602
+FBIOGETCMAP = 0x4604
+FBIOPUTCMAP = 0x4605
+FBIOPAN_DISPLAY = 0x4606
 
-FBIOGET_CON2FBMAP=0x460F
-FBIOPUT_CON2FBMAP=0x4610
-FBIOBLANK=0x4611
-FBIO_ALLOC=0x4613
-FBIO_FREE=0x4614
-FBIOGET_GLYPH=0x4615
-FBIOGET_HWCINFO=0x4616
-FBIOPUT_MODEINFO=0x4617
-FBIOGET_DISPINFO=0x4618
+FBIOGET_CON2FBMAP = 0x460F
+FBIOPUT_CON2FBMAP = 0x4610
+FBIOBLANK = 0x4611
+FBIO_ALLOC = 0x4613
+FBIO_FREE = 0x4614
+FBIOGET_GLYPH = 0x4615
+FBIOGET_HWCINFO = 0x4616
+FBIOPUT_MODEINFO = 0x4617
+FBIOGET_DISPINFO = 0x4618
 
 from mmap import mmap
 from fcntl import ioctl
 import struct
 
 mm = None
-bpp, w, h = 0, 0, 0 # framebuffer bpp and size
+bpp, w, h = 0, 0, 0  # framebuffer bpp and size
 bytepp = 0
-vx, vy, vw, vh = 0, 0, 0, 0 #virtual window offset and size
+vx, vy, vw, vh = 0, 0, 0, 0  # virtual window offset and size
 vi, fi = None, None
-_fb_cmap = 'IIPPPP' # start, len, r, g, b, a
+_fb_cmap = 'IIPPPP'  # start, len, r, g, b, a
 RGB = False
 _verbose = False
 msize_kb = 0
 
+
 def report_fb(i=0, layer=0):
-  with open('/dev/fb'+str(i), 'r+b')as f:
-    vi = ioctl(f, FBIOGET_VSCREENINFO, bytes(160))
-    vi = list(struct.unpack('I'*40, vi))
-    ffm = 'c'*16+'L'+'I'*4+'H'*3+'ILIIHHH'
-    fic = struct.calcsize(ffm)
-    fi = struct.unpack(ffm, ioctl(f, FBIOGET_FSCREENINFO, bytes(fic)))
+    with open('/dev/fb' + str(i), 'r+b') as f:
+        vi = ioctl(f, FBIOGET_VSCREENINFO, bytes(160))
+        vi = list(struct.unpack('I' * 40, vi))
+        ffm = 'c' * 16 + 'L' + 'I' * 4 + 'H' * 3 + 'ILIIHHH'
+        fic = struct.calcsize(ffm)
+        fi = struct.unpack(ffm, ioctl(f, FBIOGET_FSCREENINFO, bytes(fic)))
+
 
 def ready_fb(_bpp=None, i=0, layer=0, _win=None):
-  global mm, bpp, w, h, vi, fi, RGB, msize_kb, vx, vy, vw, vh, bytepp
-  if mm and bpp == _bpp: return mm, w, h, bpp
-  with open('/dev/fb'+str(i), 'r+b')as f:
-    vi = ioctl(f, FBIOGET_VSCREENINFO, bytes(160))
-    vi = list(struct.unpack('I'*40, vi))
-    bpp = vi[6]
-    bytepp = bpp//8
-    if _bpp:
-      vi[6] = _bpp # 24 bit = BGR 888 mode
-      try:
-        vi = ioctl(f, FBIOPUT_VSCREENINFO, struct.pack('I'*40, *vi)) # fb_var_screeninfo
-        vi = struct.unpack('I'*40,vi)
+    global mm, bpp, w, h, vi, fi, RGB, msize_kb, vx, vy, vw, vh, bytepp
+    if mm and bpp == _bpp:
+        return mm, w, h, bpp
+    with open('/dev/fb' + str(i), 'r+b') as f:
+        vi = ioctl(f, FBIOGET_VSCREENINFO, bytes(160))
+        vi = list(struct.unpack('I' * 40, vi))
         bpp = vi[6]
-        bytepp = bpp//8
-      except:
-        pass
-    
-    if vi[8] == 0 : RGB = True
-    
-    ffm = 'c'*16+'L'+'I'*4+'H'*3+'ILIIHHH'
-    fic = struct.calcsize(ffm)
-    fi = struct.unpack(ffm, ioctl(f, FBIOGET_FSCREENINFO, bytes(fic)))
-    msize = fi[17] # = w*h*bpp//8
-    ll, start = fi[-7:-5]
-    w, h = ll//bytepp, vi[1] # when screen is vertical, width becomes wrong. ll//3 is more accurate at such time.
-    if _win and len(_win)==4: # virtual window settings
-      vx, vy, vw, vh = _win
-      if vw == 'w': vw = w
-      if vh == 'h': vh = h
-      vx, vy, vw, vh = map(int, (vx, vy, vw, vh))
-      if vx>=w: vx = 0
-      if vy>=h: vy = 0
-      if vx>w: vw = w - vx
-      else: vw -= vx
-      if vy>h: vh = h - vy
-      else: vh -= vy
-    else:
-      vx, vy, vw, vh = 0,0,w,h
-    msize_kb = vw*vh*bytepp//1024 # more accurate FB memory size in kb
+        bytepp = bpp // 8
+        if _bpp:
+            vi[6] = _bpp  # 24 bit = BGR 888 mode
+            try:
+                vi = ioctl(f, FBIOPUT_VSCREENINFO, struct.pack('I' * 40, *vi))  # fb_var_screeninfo
+                vi = struct.unpack('I' * 40, vi)
+                bpp = vi[6]
+                bytepp = bpp // 8
+            except:
+                pass
 
-    mm = mmap(f.fileno(), msize, offset=start)
-    return mm, w, h, bpp#ll//(bpp//8), h
+        if vi[8] == 0:
+            RGB = True
+
+        ffm = 'c' * 16 + 'L' + 'I' * 4 + 'H' * 3 + 'ILIIHHH'
+        fic = struct.calcsize(ffm)
+        fi = struct.unpack(ffm, ioctl(f, FBIOGET_FSCREENINFO, bytes(fic)))
+        msize = fi[17]  # = w*h*bpp//8
+        ll, start = fi[-7:-5]
+        w, h = ll // bytepp, vi[1]  # when screen is vertical, width becomes wrong. ll//3 is more accurate at such time.
+        if _win and len(_win) == 4:  # virtual window settings
+            vx, vy, vw, vh = _win
+            if vw == 'w':
+                vw = w
+            if vh == 'h':
+                vh = h
+            vx, vy, vw, vh = map(int, (vx, vy, vw, vh))
+            if vx >= w:
+                vx = 0
+            if vy >= h:
+                vy = 0
+            if vx > w:
+                vw = w - vx
+            else:
+                vw -= vx
+            if vy > h:
+                vh = h - vy
+            else:
+                vh -= vy
+        else:
+            vx, vy, vw, vh = 0, 0, w, h
+        msize_kb = vw * vh * bytepp // 1024  # more accurate FB memory size in kb
+
+        mm = mmap(f.fileno(), msize, offset=start)
+        return mm, w, h, bpp  # ll//(bpp//8), h
+
+
+def fill_scr(r, g, b):
+    if bpp == 32:
+        seed = struct.pack('BBBB', b, g, r, 255)
+    elif bpp == 24:
+        seed = struct.pack('BBB', b, g, r)
+    elif bpp == 16:
+        seed = struct.pack('H', r >> 3 << 11 | g >> 2 << 5 | b >> 3)
+    mm.seek(0)
+    show_img(seed * vw * vh)
 
-def fill_scr(r,g,b):
-  if bpp == 32:
-    seed = struct.pack('BBBB', b, g, r, 255)
-  elif bpp == 24:
-    seed = struct.pack('BBB', b, g, r)
-  elif bpp == 16:
-    seed = struct.pack('H', r>>3<<11 | g>>2<<5 | b>>3)
-  mm.seek(0)
-  show_img(seed * vw * vh)
 
 def black_scr():
-  fill_scr(0,0,0)
+    fill_scr(0, 0, 0)
+
 
 def white_scr():
-  fill_scr(255,255,255)
+    fill_scr(255, 255, 255)
+
+
+def mmseekto(x, y):
+    mm.seek((x + y * w) * bytepp)
 
-def mmseekto(x,y):
-  mm.seek((x + y*w) * bytepp)
 
 def dot(x, y, r, g, b):
-  mmseekto(x,y)
-  mm.write(struct.pack('BBB',*((r,g,b) if RGB else (b,g,r))))
+    mmseekto(x, y)
+    mm.write(struct.pack('BBB', *((r, g, b) if RGB else (b, g, r))))
+
+
+def get_pixel(x, y):
+    mmseekto(x, y)
+    return mm.read(bytepp)
 
-def get_pixel(x,y):
-  mmseekto(x,y)
-  return mm.read(bytepp)
 
 def _888_to_565(bt):
-  b = b''
-  for i in range(0, len(bt),3):
-    b += int.to_bytes(bt[i]>>3<<11|bt[i+1]>>2<<5|bt[i+2]>>3, 2, 'little')
-  return b
+    b = b''
+    for i in range(0, len(bt), 3):
+        b += int.to_bytes(bt[i] >> 3 << 11 | bt[i + 1] >> 2 << 5 | bt[i + 2] >> 3, 2, 'little')
+    return b
+
 
 def numpy_888_565(bt):
-  import numpy as np
-  arr = np.fromstring(bt, dtype=np.uint32)
-  return (((0xF80000 & arr)>>8)|((0xFC00 & arr)>>5)|((0xF8 & arr)>>3)).astype(np.uint16).tostring()
+    import numpy as np
+    arr = np.fromstring(bt, dtype=np.uint32)
+    return (((0xF80000 & arr) >> 8) | ((0xFC00 & arr) >> 5) | ((0xF8 & arr) >> 3)).astype(np.uint16).tostring()
+
 
 def show_img(img):
-  if not type(img) is bytes:
-    if not RGB:
-      if bpp == 24: # for RPI
-        img = img.tobytes('raw', 'BGR')
-      else:
-        img = img.convert('RGBA').tobytes('raw', 'BGRA')
-        if bpp == 16:
-          img = numpy_888_565(img)
-    else:
-      if bpp == 24:
-        img = img.tobytes()
-      else:
-        img = img.convert('RGBA').tobytes()
-        if bpp == 16:
-          img = numpy_888_565(img)
-  from io import BytesIO
-  b = BytesIO(img)
-  s = vw*bytepp
-  for y in range(vh): # virtual window drawing
-    mmseekto(vx,vy+y)
-    mm.write(b.read(s))
-
+    if not type(img) is bytes:
+        if not RGB:
+            if bpp == 24:  # for RPI
+                img = img.tobytes('raw', 'BGR')
+            else:
+                img = img.convert('RGBA').tobytes('raw', 'BGRA')
+                if bpp == 16:
+                    img = numpy_888_565(img)
+        else:
+            if bpp == 24:
+                img = img.tobytes()
+            else:
+                img = img.convert('RGBA').tobytes()
+                if bpp == 16:
+                    img = numpy_888_565(img)
+    from io import BytesIO
+    b = BytesIO(img)
+    s = vw * bytepp
+    for y in range(vh):  # virtual window drawing
+        mmseekto(vx, vy + y)
+        mm.write(b.read(s))

pwnagotchi/ui/hw/libs/pimoroni/displayhatmini/ST7789.py

@@ -90,8 +90,8 @@ ST7789_PWCTR6 = 0xFC
 class ST7789(object):
     """Representation of an ST7789 TFT LCD."""
 
-    def __init__(self, port, cs, dc, backlight, rst=None, width=320,
-                 height=240, rotation=0, invert=True, spi_speed_hz=60 * 1000 * 1000,
+    def __init__(self, port, cs, dc, backlight=None, rst=None, width=240,
+                 height=240, rotation=90, invert=True, spi_speed_hz=4000000,
                  offset_left=0,
                  offset_top=0):
         """Create an instance of the display using SPI communication.

pwnagotchi/ui/hw/libs/waveshare/epaper/v13in3k/epd13in3k.py

@@ -29,7 +29,7 @@
 
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 960

pwnagotchi/ui/hw/libs/waveshare/epaper/v1in02/epd1in02.py

@@ -29,7 +29,7 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 80

pwnagotchi/ui/hw/libs/waveshare/epaper/v1in54/epd1in54.py

@@ -80,7 +80,7 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 200

pwnagotchi/ui/hw/libs/waveshare/epaper/v1in54_v2/epd1in54_V2.py

@@ -28,7 +28,7 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 200

pwnagotchi/ui/hw/libs/waveshare/epaper/v1in54b/epd1in54b.py

@@ -28,7 +28,7 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH       = 200

pwnagotchi/ui/hw/libs/waveshare/epaper/v1in54b_v2/epd1in54b_v2.py

@@ -28,7 +28,7 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 200

pwnagotchi/ui/hw/libs/waveshare/epaper/v1in54c/epd1in54c.py

@@ -27,7 +27,7 @@
 # THE SOFTWARE.
 #
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 152

pwnagotchi/ui/hw/libs/waveshare/epaper/v1in64g/epd1in64g.py

@@ -29,11 +29,9 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
-import PIL
 from PIL import Image
-import io
 
 # Display resolution
 EPD_WIDTH = 168

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in13_V1/epd2in13.py

@@ -29,7 +29,7 @@
 
 
 import logging
-from . import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH       = 122

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in13_V1/epd2in13bc.py

@@ -27,7 +27,7 @@
 # THE SOFTWARE.
 #
 
-from . import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 import RPi.GPIO as GPIO
 # import numpy as np
 

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in13_V1/epd2in13bcFAST.py

@@ -40,9 +40,8 @@
 
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 from PIL import Image
-import RPi.GPIO as GPIO
 
 # Display resolution
 EPD_WIDTH       = 104

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in13_V2/epd2in13_V2.py

@@ -29,7 +29,7 @@
 
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 122

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in13_V3/epd2in13_V3.py

@@ -4,8 +4,8 @@
 # * | Function    :   Electronic paper driver
 # * | Info        :
 # *----------------
-# * | This version:   V1.1
-# * | Date        :   2021-10-30
+# * | This version:   V1.2
+# * | Date        :   2022-08-9
 # # | Info        :   python demo
 # -----------------------------------------------------------------------------
 # Permission is hereby granted, free of charge, to any person obtaining a copy
@@ -29,15 +29,15 @@
 
 
 import logging
-from . import epdconfig
-import numpy as np
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
-EPD_WIDTH       = 122
-EPD_HEIGHT      = 250
+EPD_WIDTH = 122
+EPD_HEIGHT = 250
 
 logger = logging.getLogger(__name__)
 
+
 class EPD:
     def __init__(self):
         self.reset_pin = epdconfig.RST_PIN
@@ -46,68 +46,70 @@ class EPD:
         self.cs_pin = epdconfig.CS_PIN
         self.width = EPD_WIDTH
         self.height = EPD_HEIGHT
-        
-    lut_partial_update= [
-        0x0,0x40,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x80,0x80,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x40,0x40,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x80,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x14,0x0,0x0,0x0,0x0,0x0,0x0,  
-        0x1,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x1,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x22,0x22,0x22,0x22,0x22,0x22,0x0,0x0,0x0,
-        0x22,0x17,0x41,0x00,0x32,0x36,
+
+    lut_partial_update = [
+        0x0, 0x40, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x80, 0x80, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x40, 0x40, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x80, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x14, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x1, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x1, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x0, 0x0, 0x0,
+        0x22, 0x17, 0x41, 0x00, 0x32, 0x36,
     ]
 
-    lut_full_update = [ 
-        0x80,0x4A,0x40,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x40,0x4A,0x80,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x80,0x4A,0x40,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x40,0x4A,0x80,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0xF,0x0,0x0,0x0,0x0,0x0,0x0,
-        0xF,0x0,0x0,0xF,0x0,0x0,0x2,
-        0xF,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x1,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x0,0x0,0x0,0x0,0x0,0x0,0x0,
-        0x22,0x22,0x22,0x22,0x22,0x22,0x0,0x0,0x0,
-        0x22,0x17,0x41,0x0,0x32,0x36,
+    lut_full_update = [
+        0x80, 0x4A, 0x40, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x40, 0x4A, 0x80, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x80, 0x4A, 0x40, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x40, 0x4A, 0x80, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0xF, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0xF, 0x0, 0x0, 0xF, 0x0, 0x0, 0x2,
+        0xF, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x1, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+        0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x0, 0x0, 0x0,
+        0x22, 0x17, 0x41, 0x0, 0x32, 0x36,
     ]
-        
+
     '''
     function :Hardware reset
     parameter:
     '''
+
     def reset(self):
         epdconfig.digital_write(self.reset_pin, 1)
-        epdconfig.delay_ms(20) 
+        epdconfig.delay_ms(20)
         epdconfig.digital_write(self.reset_pin, 0)
         epdconfig.delay_ms(2)
         epdconfig.digital_write(self.reset_pin, 1)
-        epdconfig.delay_ms(20)   
+        epdconfig.delay_ms(20)
 
     '''
     function :send command
     parameter:
      command : Command register
     '''
+
     def send_command(self, command):
         epdconfig.digital_write(self.dc_pin, 0)
         epdconfig.digital_write(self.cs_pin, 0)
@@ -119,71 +121,84 @@ class EPD:
     parameter:
      data : Write data
     '''
+
     def send_data(self, data):
         epdconfig.digital_write(self.dc_pin, 1)
         epdconfig.digital_write(self.cs_pin, 0)
         epdconfig.spi_writebyte([data])
         epdconfig.digital_write(self.cs_pin, 1)
-    
+
+    # send a lot of data
+    def send_data2(self, data):
+        epdconfig.digital_write(self.dc_pin, 1)
+        epdconfig.digital_write(self.cs_pin, 0)
+        epdconfig.spi_writebyte2(data)
+        epdconfig.digital_write(self.cs_pin, 1)
+
     '''
     function :Wait until the busy_pin goes LOW
     parameter:
     '''
+
     def ReadBusy(self):
         logger.debug("e-Paper busy")
-        while(epdconfig.digital_read(self.busy_pin) == 1):      # 0: idle, 1: busy
-            epdconfig.delay_ms(10)  
+        while (epdconfig.digital_read(self.busy_pin) == 1):  # 0: idle, 1: busy
+            epdconfig.delay_ms(10)
         logger.debug("e-Paper busy release")
 
     '''
     function : Turn On Display
     parameter:
     '''
+
     def TurnOnDisplay(self):
-        self.send_command(0x22) # Display Update Control
+        self.send_command(0x22)  # Display Update Control
         self.send_data(0xC7)
-        self.send_command(0x20) # Activate Display Update Sequence
+        self.send_command(0x20)  # Activate Display Update Sequence
         self.ReadBusy()
-    
+
     '''
     function : Turn On Display Part
     parameter:
     '''
+
     def TurnOnDisplayPart(self):
-        self.send_command(0x22) # Display Update Control
-        self.send_data(0x0f)    # fast:0x0c, quality:0x0f, 0xcf
-        self.send_command(0x20) # Activate Display Update Sequence
+        self.send_command(0x22)  # Display Update Control
+        self.send_data(0x0f)  # fast:0x0c, quality:0x0f, 0xcf
+        self.send_command(0x20)  # Activate Display Update Sequence
         self.ReadBusy()
-    
+
     '''
     function : Set lut
     parameter:
         lut : lut data
-    '''    
+    '''
+
     def Lut(self, lut):
         self.send_command(0x32)
         for i in range(0, 153):
             self.send_data(lut[i])
         self.ReadBusy()
-    
+
     '''
     function : Send lut data and configuration
     parameter:
         lut : lut data 
     '''
+
     def SetLut(self, lut):
         self.Lut(lut)
         self.send_command(0x3f)
         self.send_data(lut[153])
-        self.send_command(0x03)     # gate voltage
+        self.send_command(0x03)  # gate voltage
         self.send_data(lut[154])
-        self.send_command(0x04)     # source voltage
-        self.send_data(lut[155])    # VSH
-        self.send_data(lut[156])    # VSH2
-        self.send_data(lut[157])    # VSL
-        self.send_command(0x2c)     # VCOM
+        self.send_command(0x04)  # source voltage
+        self.send_data(lut[155])  # VSH
+        self.send_data(lut[156])  # VSH2
+        self.send_data(lut[157])  # VSL
+        self.send_command(0x2c)  # VCOM
         self.send_data(lut[158])
-    
+
     '''
     function : Setting the display window
     parameter:
@@ -192,13 +207,14 @@ class EPD:
         xend : End position of X-axis
         yend : End position of Y-axis
     '''
+
     def SetWindow(self, x_start, y_start, x_end, y_end):
-        self.send_command(0x44) # SET_RAM_X_ADDRESS_START_END_POSITION
+        self.send_command(0x44)  # SET_RAM_X_ADDRESS_START_END_POSITION
         # x point must be the multiple of 8 or the last 3 bits will be ignored
-        self.send_data((x_start>>3) & 0xFF)
-        self.send_data((x_end>>3) & 0xFF)
-        
-        self.send_command(0x45) # SET_RAM_Y_ADDRESS_START_END_POSITION
+        self.send_data((x_start >> 3) & 0xFF)
+        self.send_data((x_end >> 3) & 0xFF)
+
+        self.send_command(0x45)  # SET_RAM_Y_ADDRESS_START_END_POSITION
         self.send_data(y_start & 0xFF)
         self.send_data((y_start >> 8) & 0xFF)
         self.send_data(y_end & 0xFF)
@@ -210,52 +226,54 @@ class EPD:
         x : X-axis starting position
         y : Y-axis starting position
     '''
+
     def SetCursor(self, x, y):
-        self.send_command(0x4E) # SET_RAM_X_ADDRESS_COUNTER
+        self.send_command(0x4E)  # SET_RAM_X_ADDRESS_COUNTER
         # x point must be the multiple of 8 or the last 3 bits will be ignored
         self.send_data(x & 0xFF)
-        
-        self.send_command(0x4F) # SET_RAM_Y_ADDRESS_COUNTER
+
+        self.send_command(0x4F)  # SET_RAM_Y_ADDRESS_COUNTER
         self.send_data(y & 0xFF)
         self.send_data((y >> 8) & 0xFF)
-    
+
     '''
     function : Initialize the e-Paper register
     parameter:
     '''
+
     def init(self):
         if (epdconfig.module_init() != 0):
             return -1
         # EPD hardware init start
         self.reset()
-        
-        self.ReadBusy()
-        self.send_command(0x12)  #SWRESET
-        self.ReadBusy() 
 
-        self.send_command(0x01) #Driver output control      
+        self.ReadBusy()
+        self.send_command(0x12)  # SWRESET
+        self.ReadBusy()
+
+        self.send_command(0x01)  # Driver output control
         self.send_data(0xf9)
         self.send_data(0x00)
         self.send_data(0x00)
-    
-        self.send_command(0x11) #data entry mode       
+
+        self.send_command(0x11)  # data entry mode
         self.send_data(0x03)
 
-        self.SetWindow(0, 0, self.width-1, self.height-1)
+        self.SetWindow(0, 0, self.width - 1, self.height - 1)
         self.SetCursor(0, 0)
-        
+
         self.send_command(0x3c)
         self.send_data(0x05)
 
-        self.send_command(0x21) #  Display update control
+        self.send_command(0x21)  # Display update control
         self.send_data(0x00)
         self.send_data(0x80)
-    
+
         self.send_command(0x18)
         self.send_data(0x80)
-        
+
         self.ReadBusy()
-        
+
         self.SetLut(self.lut_full_update)
         return 0
 
@@ -264,54 +282,52 @@ class EPD:
     parameter:
         image : Image data
     '''
+
     def getbuffer(self, image):
         img = image
         imwidth, imheight = img.size
-        if(imwidth == self.width and imheight == self.height):
+        if (imwidth == self.width and imheight == self.height):
             img = img.convert('1')
-        elif(imwidth == self.height and imheight == self.width):
+        elif (imwidth == self.height and imheight == self.width):
             # image has correct dimensions, but needs to be rotated
             img = img.rotate(90, expand=True).convert('1')
         else:
             logger.warning("Wrong image dimensions: must be " + str(self.width) + "x" + str(self.height))
             # return a blank buffer
-            return [0x00] * (int(self.width/8) * self.height)
+            return [0x00] * (int(self.width / 8) * self.height)
 
         buf = bytearray(img.tobytes('raw'))
         return buf
-        
+
     '''
     function : Sends the image buffer in RAM to e-Paper and displays
     parameter:
         image : Image data
     '''
+
     def display(self, image):
-        if self.width%8 == 0:
-            linewidth = int(self.width/8)
+        if self.width % 8 == 0:
+            linewidth = int(self.width / 8)
         else:
-            linewidth = int(self.width/8) + 1
+            linewidth = int(self.width / 8) + 1
 
         self.send_command(0x24)
         for j in range(0, self.height):
             for i in range(0, linewidth):
-                self.send_data(image[i + j * linewidth])   
+                self.send_data(image[i + j * linewidth])
         self.TurnOnDisplay()
-    
+
     '''
     function : Sends the image buffer in RAM to e-Paper and partial refresh
     parameter:
         image : Image data
     '''
-    def displayPartial(self, image):
-        if self.width%8 == 0:
-            linewidth = int(self.width/8)
-        else:
-            linewidth = int(self.width/8) + 1
 
+    def displayPartial(self, image):
         epdconfig.digital_write(self.reset_pin, 0)
         epdconfig.delay_ms(1)
-        epdconfig.digital_write(self.reset_pin, 1)  
-        
+        epdconfig.digital_write(self.reset_pin, 1)
+
         self.SetLut(self.lut_partial_update)
         self.send_command(0x37)
         self.send_data(0x00)
@@ -322,24 +338,25 @@ class EPD:
         self.send_data(0x40)
         self.send_data(0x00)
         self.send_data(0x00)
-        self.send_data(0x00)  
+        self.send_data(0x00)
         self.send_data(0x00)
 
-        self.send_command(0x3C) #BorderWavefrom
+        self.send_command(0x3C)  # BorderWavefrom
         self.send_data(0x80)
 
-        self.send_command(0x22) 
+        self.send_command(0x22)
         self.send_data(0xC0)
         self.send_command(0x20)
         self.ReadBusy()
 
         self.SetWindow(0, 0, self.width - 1, self.height - 1)
         self.SetCursor(0, 0)
-        
-        self.send_command(0x24) # WRITE_RAM
-        for j in range(0, self.height):
-            for i in range(0, linewidth):
-                self.send_data(image[i + j * linewidth])   
+
+        self.send_command(0x24)  # WRITE_RAM
+        # for j in range(0, self.height):
+        #     for i in range(0, linewidth):
+        #         self.send_data(image[i + j * linewidth])
+        self.send_data2(image)
         self.TurnOnDisplayPart()
 
     '''
@@ -347,51 +364,41 @@ class EPD:
     parameter:
         image : Image data
     '''
-    def displayPartBaseImage(self, image):
-        if self.width%8 == 0:
-            linewidth = int(self.width/8)
-        else:
-            linewidth = int(self.width/8) + 1
 
+    def displayPartBaseImage(self, image):
         self.send_command(0x24)
-        for j in range(0, self.height):
-            for i in range(0, linewidth):
-                self.send_data(image[i + j * linewidth])   
-                
+        self.send_data2(image)
+
         self.send_command(0x26)
-        for j in range(0, self.height):
-            for i in range(0, linewidth):
-                self.send_data(image[i + j * linewidth])  
+        self.send_data2(image)
         self.TurnOnDisplay()
-    
+
     '''
     function : Clear screen
     parameter:
     '''
-    def Clear(self, color):
-        if self.width%8 == 0:
-            linewidth = int(self.width/8)
+
+    def Clear(self, color=0xFF):
+        if self.width % 8 == 0:
+            linewidth = int(self.width / 8)
         else:
-            linewidth = int(self.width/8) + 1
+            linewidth = int(self.width / 8) + 1
         # logger.debug(linewidth)
-        
+
         self.send_command(0x24)
-        for j in range(0, self.height):
-            for i in range(0, linewidth):
-                self.send_data(color)
-                
+        self.send_data2([color] * int(self.height * linewidth))
         self.TurnOnDisplay()
 
     '''
     function : Enter sleep mode
     parameter:
     '''
+
     def sleep(self):
-        self.send_command(0x10) #enter deep sleep
+        self.send_command(0x10)  # enter deep sleep
         self.send_data(0x01)
-        
+
         epdconfig.delay_ms(2000)
         epdconfig.module_exit()
 
 ### END OF FILE ###
-

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in13_V4/epd2in13_V4.py

@@ -29,7 +29,7 @@
 
 
 import logging
-from . import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 122

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in13b_v3/epd2in13b_V3.py

@@ -28,7 +28,7 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 104

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in13b_v4/epd2in13b_V4.py

@@ -28,7 +28,7 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 122

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in13bc/epd2in13bc.py

@@ -28,7 +28,7 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 from PIL import Image
 
 # Display resolution

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in13d/epd2in13d.py

@@ -29,9 +29,8 @@
 
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 from PIL import Image
-import RPi.GPIO as GPIO
 
 # Display resolution
 EPD_WIDTH       = 104

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in13g/epd2in13g.py

@@ -29,11 +29,9 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
-import PIL
 from PIL import Image
-import io
 
 # Display resolution
 EPD_WIDTH = 122

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in23g/epd2in13g.py

@@ -29,11 +29,9 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
-import PIL
 from PIL import Image
-import io
 
 # Display resolution
 EPD_WIDTH = 122

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in36g/epd2in36g.py

@@ -29,11 +29,9 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
-import PIL
 from PIL import Image
-import io
 
 # Display resolution
 EPD_WIDTH = 168

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in66/epd2in66.py

@@ -28,7 +28,7 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 152

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in66b/epd2in66b.py

@@ -28,7 +28,7 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 152

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in66g/epd2in66g.py

@@ -29,11 +29,9 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
-import PIL
 from PIL import Image
-import io
 
 # Display resolution
 EPD_WIDTH = 184

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in7/epd2in7.py

@@ -28,7 +28,7 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH       = 176

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in7_v2/epd2in7_V2.py

@@ -28,7 +28,7 @@
 #
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 176

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in7b/epd2in7b.py

@@ -29,7 +29,7 @@
 
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 176

pwnagotchi/ui/hw/libs/waveshare/epaper/v2in7b_v2/epd2in7b_V2.py

@@ -29,7 +29,7 @@
 
 
 import logging
-from .. import epdconfig
+from pwnagotchi.ui.hw.libs.waveshare.epaper import epdconfig
 
 # Display resolution
 EPD_WIDTH = 176