add config to disable filter

SENERGY-Platform · Nov 19, 2020 · 116422c · 116422c
1 parent d108ffa
commit 116422c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/main.py b/main.py
@@ -15,7 +15,7 @@
 import uuid
 from datetime import datetime
 
-from senergy_local_analytics import App, Input, Output
+from senergy_local_analytics import App, Input, Output, Config
 
 
 class Adder:
@@ -25,20 +25,24 @@ def __init__(self):
         self.sum_total = 0
         self.message_no = 0
 
-    def process(self, inputs: typing.List[Input]):
+    def process(self, inputs: typing.List[Input], config: Config):
         for inp in inputs:
             if inp.current_value is not None:
                 self.values[inp.current_topic] = inp.current_value
         self.message_no = self.message_no + 1
         value_sum = sum(self.values.values())
-        if self.sum_total != value_sum:
-            self.sum_total = value_sum
-            return Output(True, {"sum": sum(self.values.values()), "message_no": self.message_no,
-                                 "message_id": str(uuid.uuid4()),
-                                 "timestamp": '{}Z'.format(datetime.utcnow().isoformat())})
+        if config.get_config_value("filter", "True") == "True":
+            if self.sum_total != value_sum:
+                self.sum_total = value_sum
+                return Output(True, {"sum": sum(self.values.values()), "message_no": self.message_no,
+                                     "message_id": str(uuid.uuid4()),
+                                     "timestamp": '{}Z'.format(datetime.utcnow().isoformat())})
+            else:
+                return Output(False, {"sum": sum(self.values.values()), "message_id": str(uuid.uuid4()),
+                                      "timestamp": '{}Z'.format(datetime.utcnow().isoformat())})
         else:
-            return Output(False, {"sum": sum(self.values.values()), "message_id": str(uuid.uuid4()),
-                                  "timestamp": '{}Z'.format(datetime.utcnow().isoformat())})
+            return Output(True, {"sum": sum(self.values.values()), "message_id": str(uuid.uuid4()),
+                                 "timestamp": '{}Z'.format(datetime.utcnow().isoformat())})
 
 
 if __name__ == '__main__':

diff --git a/test/test.py b/test/test.py
@@ -30,7 +30,7 @@ def test_process(self):
         input2.current_topic = "test2"
         output = adder.process([input1, input2])
         self.assertTrue(output.send)
-        self.assertEqual({'sum', 'message_id', 'timestamp'}, output.values.keys())
+        self.assertEqual({'sum', 'message_no', 'message_id', 'timestamp'}, output.values.keys())
         input1.current_value = 7
         input1.current_topic = "test1"
         input2.current_value = 4