Add cleanup function to modules, called upon exit. #16

pi_mqtt_gpio/modules/__init__.py

@@ -44,3 +44,9 @@ class GenericGPIO(object):
     @abc.abstractmethod
     def get_pin(self, pin):
         pass
+
+    def cleanup(self):
+        """
+        Called when closing the program to handle any cleanup operations.
+        """
+        pass

pi_mqtt_gpio/modules/raspberrypi.py

@@ -43,3 +43,6 @@ class GPIO(GenericGPIO):
 
     def get_pin(self, pin):
         return self.io.input(pin)
+
+    def cleanup(self):
+        self.io.cleanup()

pi_mqtt_gpio/server.py

@@ -488,3 +488,9 @@ if __name__ == "__main__":
     finally:
         client.disconnect()
         client.loop_stop()
+        for name, gpio in GPIO_MODULES.items():
+            try:
+                gpio.cleanup()
+            except Exception:
+                _LOG.exception(
+                    "Unable to execute cleanup routine for module %r:", name)