From a62891eb6aab7de816b6c687dcb002cf8415fc7c Mon Sep 17 00:00:00 2001
From: Romain Bazile <gromain.baz@gmail.com>
Date: Fri, 27 Nov 2020 11:30:29 +0100
Subject: [PATCH] light: catch exceptions

---
 scripts/planktoscope/light.py | 38 ++++++++++++++++++++++++++---------
 1 file changed, 28 insertions(+), 10 deletions(-)

diff --git a/scripts/planktoscope/light.py b/scripts/planktoscope/light.py
index e55c09e..8024e04 100644
--- a/scripts/planktoscope/light.py
+++ b/scripts/planktoscope/light.py
@@ -5,6 +5,9 @@
 # Turn off using this command line :
 # python3.7 path/to/file/light.py off
 
+# Logger library compatible with multiprocessing
+from loguru import logger
+
 # Library to send command over I2C for the light module on the fan
 import smbus
 import RPi.GPIO
@@ -31,17 +34,23 @@ def i2c_update():
 
 def setRGB(R, G, B):
     """Update all LED at the same time"""
-    bus.write_byte_data(DEVICE_ADDRESS, 0x00, 0xFF)
-    bus.write_byte_data(DEVICE_ADDRESS, 0x01, R & 0xFF)
-    bus.write_byte_data(DEVICE_ADDRESS, 0x02, G & 0xFF)
-    bus.write_byte_data(DEVICE_ADDRESS, 0x03, B & 0xFF)
-    i2c_update()
+    try:
+        bus.write_byte_data(DEVICE_ADDRESS, 0x00, 0xFF)
+        bus.write_byte_data(DEVICE_ADDRESS, 0x01, R & 0xFF)
+        bus.write_byte_data(DEVICE_ADDRESS, 0x02, G & 0xFF)
+        bus.write_byte_data(DEVICE_ADDRESS, 0x03, B & 0xFF)
+        i2c_update()
+    except Exception as e:
+        logger.exception(f"An Exception has occured in the light library at {e}")
 
 
 def setRGBOff():
     """Turn off the RGB LED"""
-    bus.write_byte_data(DEVICE_ADDRESS, 0x07, 0x00)
-    i2c_update()
+    try:
+        bus.write_byte_data(DEVICE_ADDRESS, 0x07, 0x00)
+        i2c_update()
+    except Exception as e:
+        logger.exception(f"An Exception has occured in the light library at {e}")
 
 
 def setRGBEffect(effect):
@@ -55,13 +64,19 @@ def setRGBEffect(effect):
     """
 
     if effect >= 0 and effect <= 4:
-        bus.write_byte_data(DEVICE_ADDRESS, rgb_effect_reg, effect & 0xFF)
+        try:
+            bus.write_byte_data(DEVICE_ADDRESS, rgb_effect_reg, effect & 0xFF)
+        except Exception as e:
+            logger.exception(f"An Exception has occured in the light library at {e}")
 
 
 def setRGBSpeed(speed):
     """Set the effect speed, 1-3, 3 being the fastest speed"""
     if speed >= 1 and speed <= 3:
-        bus.write_byte_data(DEVICE_ADDRESS, rgb_speed_reg, speed & 0xFF)
+        try:
+            bus.write_byte_data(DEVICE_ADDRESS, rgb_speed_reg, speed & 0xFF)
+        except Exception as e:
+            logger.exception(f"An Exception has occured in the light library at {e}")
 
 
 def setRGBColor(color):
@@ -77,7 +92,10 @@ def setRGBColor(color):
     """
 
     if color >= 0 and color <= 6:
-        bus.write_byte_data(DEVICE_ADDRESS, rgb_color_reg, color & 0xFF)
+        try:
+            bus.write_byte_data(DEVICE_ADDRESS, rgb_color_reg, color & 0xFF)
+        except Exception as e:
+            logger.exception(f"An Exception has occured in the light library at {e}")
 
 
 def light(state):