Fix KeyError for gpiozero digital_outputs flyte#414

arturalbov · Oct 20, 2024 · f6838ff · f6838ff
1 parent 678276f
commit f6838ff
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/mqtt_io/modules/gpio/gpiozero.py b/mqtt_io/modules/gpio/gpiozero.py
@@ -80,9 +80,14 @@ def set_pin(self, pin: PinType, value: bool) -> None:
             self._out_pins[pin].on()
         else:
             self._out_pins[pin].off()
-
+    
     def get_pin(self, pin: PinType) -> bool:
-        return cast(bool, self._in_pins[pin].is_active)
+        if pin in self._in_pins:
+            return cast(bool, self._in_pins[pin].is_active)
+        elif pin in self._out_pins:
+            return bool(self._out_pins[pin].value)
+        else:
+            raise ValueError(f"Pin {pin} not found")
 
     def setup_interrupt_callback(
         self,