decorator based subscription

Author: tomvictor

examples/decorator.py

@@ -0,0 +1,43 @@
+class IoTCore:
+    def __init__(self):
+        self.subs = {}  # Dictionary to store topic-callback function pairs
+
+    def accept(self, topic):
+        def decorator(func):
+            self.subs[topic] = func  # Store the topic and callback function in the dictionary
+
+            def wrapper(request):
+                func(request)
+
+            return wrapper
+
+        return decorator
+
+    def temp(self):
+        print("do something")
+
+
+iot = IoTCore()
+
+
+@iot.accept(topic="request/temperature/data")
+def temperature_data_handler(request):
+    print("Handling temperature data request:", request)
+
+
+# Simulate a temperature data request
+request_data = {
+    'topic': 'request/temperature/data',
+    'data': {
+        'temperature': 25.5
+    }
+}
+
+
+def main():
+    temperature_data_handler(request_data)
+    iot.temp()
+
+
+if __name__ == "__main__":
+    main()

examples/fastapi/main.py

@@ -34,3 +34,7 @@ def pub():
     iot.publish("iot", "test")
     return {"response": "published"}
 
+
+@iot.accept(topic="request/temperature/data")
+def temperature_data(request):
+    print(request)

iotcore/mqtt.py

@@ -43,3 +43,15 @@ def iot_core_callback(self, topic, data):
                 subscription.callback(data)
         except KeyError:
             print(f"invalid topic : {topic}")
+
+    def accept(self, topic):
+        def decorator(func):
+            self.subscribed_topics[hash(topic)] = func  # Store the topic and callback function in the dictionary
+
+            def wrapper(request):
+                func(request)
+
+            return wrapper
+
+        return decorator
+

iotcore/request.py

@@ -0,0 +1,14 @@
+
+class Request(object):
+
+    def __int__(self, topic, data):
+        self._topic = topic
+        self._data = data
+
+    def raw(self):
+        return self._data
+
+    def data(self):
+        data_array = bytes(self._data)
+        data_string = data_array.decode('utf-8')
+        return data_string