badge.py: add png support

pimoroni · Jun 21, 2024 · ae04ac9 · ae04ac9
1 parent 21e27e8
commit ae04ac9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/badger_os/examples/badge.py b/badger_os/examples/badge.py
@@ -57,11 +57,11 @@ def draw_badge():
     display.set_pen(0)
     display.clear()
 
-    if badge_image.endswith('.png' or '.PNG' or '.Png'):
+    try:
         # Draw badge image
         png.open_file(badge_image)
         png.decode(WIDTH - IMAGE_WIDTH, 0)
-    else:
+    except (OSError, RuntimeError):
         # Draw badge image
         jpeg.open_file(badge_image)
         jpeg.decode(WIDTH - IMAGE_WIDTH, 0)