put screenshot function into try-except block

wyrnat · Dec 14, 2023 · c774668 · c774668
1 parent 9fdd2bf
commit c774668
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/badger/logbook.py b/src/badger/logbook.py
@@ -89,7 +89,14 @@ def send_to_logbook(routine, data, widget=None):
     xmlFile.write(xmlString)
     xmlFile.write('\n')  # Close with newline so cron job parses correctly
     xmlFile.close()
-    screenshot(widget, f'{fileName}.png')
+
+    # take screenshot of GUI if possible
+    try:
+        screenshot(widget, f'{fileName}.png')
+    except BadgerLogbookError as e:
+        logger.warning(e)
+    except ImportError:
+        logger.warning("Cannot save Screenshot. PIL package missing")
 
 
 def screenshot(widget, filename):