Fix Calendar and Weather for Inky SSD1608

pimoroni · Oct 30, 2020 · f5b5bb5 · f5b5bb5
1 parent 10862b3
commit f5b5bb5
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 8 deletions.
diff --git a/examples/phat/calendar-phat.py b/examples/phat/calendar-phat.py
@@ -31,12 +31,16 @@
 colour = args.colour
 
 # Set up the display
-if colour == "auto":
+try:
     from inky.auto import auto
     inky_display = auto()
     colour = inky_display.colour
-else:
-    inky_display = InkyPHAT(colour)
+    print("Auto-detected {} Inky pHAT".format(colour))
+except RuntimeError:
+    if colour == "auto":
+        raise RuntimeError("Failed to auto-detect your Inky pHAT, try specifying a colour manually.")
+    else:
+        inky_display = InkyPHAT(colour)
 
 inky_display.set_border(inky_display.BLACK)
 

diff --git a/examples/phat/weather-phat.py b/examples/phat/weather-phat.py
@@ -53,14 +53,17 @@
 args = parser.parse_args()
 
 # Set up the display
-
-colour = args.colour
-if colour == "auto":
+try:
     from inky.auto import auto
     inky_display = auto()
     colour = inky_display.colour
-else:
-    inky_display = InkyPHAT(colour)
+    print("Auto-detected {} Inky pHAT".format(colour))
+except RuntimeError:
+    if colour == "auto":
+        raise RuntimeError("Failed to auto-detect your Inky pHAT, try specifying a colour manually.")
+    else:
+        inky_display = InkyPHAT(colour)
+
 inky_display.set_border(inky_display.BLACK)
 
 # Details to customise your weather display