not add ".heic" to EXTENSIONS without pi_heif

Signed-off-by: bigcat88 <[email protected]>
helgeerbe · Jun 12, 2023 · b4ccc0f · b4ccc0f
1 parent 2405c11
commit b4ccc0f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/picframe/interface_http.py b/src/picframe/interface_http.py
@@ -13,7 +13,14 @@
     from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer  # py2
     import urlparse
 
-EXTENSIONS = [".jpg", ".jpeg", ".png", ".heif", ".heic"]
+try:
+    from pi_heif import register_heif_opener
+except ImportError:
+    register_heif_opener = None
+
+EXTENSIONS = [".jpg", ".jpeg", ".png"]
+if register_heif_opener is not None:
+    EXTENSIONS += [".heif", ".heic"]
 
 
 def heif_to_jpg(fname):