diff --git a/pwnagotchi/ui/hw/__init__.py b/pwnagotchi/ui/hw/__init__.py index 696a41f..432cb54 100644 --- a/pwnagotchi/ui/hw/__init__.py +++ b/pwnagotchi/ui/hw/__init__.py @@ -10,6 +10,7 @@ from pwnagotchi.ui.hw.waveshare29inch import Waveshare29inch from pwnagotchi.ui.hw.waveshare144lcd import Waveshare144lcd from pwnagotchi.ui.hw.waveshare154inch import Waveshare154inch from pwnagotchi.ui.hw.waveshare213d import Waveshare213d +from pwnagotchi.ui.hw.waveshare213bc import Waveshare213bc from pwnagotchi.ui.hw.spotpear24inch import Spotpear24inch def display_for(config): @@ -37,10 +38,10 @@ def display_for(config): elif config['ui']['display']['type'] == 'waveshare27inch': return Waveshare27inch(config) - + elif config['ui']['display']['type'] == 'waveshare29inch': return Waveshare29inch(config) - + elif config['ui']['display']['type'] == 'waveshare144lcd': return Waveshare144lcd(config) @@ -50,5 +51,8 @@ def display_for(config): elif config['ui']['display']['type'] == 'waveshare213d': return Waveshare213d(config) + elif config['ui']['display']['type'] == 'waveshare213bc': + return Waveshare213bc(config) + elif config['ui']['display']['type'] == 'spotpear24inch': - return Spotpear24inch(config) \ No newline at end of file + return Spotpear24inch(config) diff --git a/pwnagotchi/utils.py b/pwnagotchi/utils.py index 4d0ba90..d645155 100644 --- a/pwnagotchi/utils.py +++ b/pwnagotchi/utils.py @@ -115,6 +115,9 @@ def load_config(args): elif config['ui']['display']['type'] in ('ws_213d', 'ws213d', 'waveshare_213d', 'waveshare213d'): config['ui']['display']['type'] = 'waveshare213d' + elif config['ui']['display']['type'] in ('ws_213bc', 'ws213bc', 'waveshare_213bc', 'waveshare213bc'): + config['ui']['display']['type'] = 'waveshare213bc' + elif config['ui']['display']['type'] in ('spotpear24inch'): config['ui']['display']['type'] = 'spotpear24inch'