From fc80c1cbe29aa46eb581efbce9b7231aa424df7a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Torma=20Krist=C3=B3f?= <tormakristof@tormakristof.eu>
Date: Thu, 30 Jul 2020 15:36:09 +0200
Subject: [PATCH] some validation

---
 src/app.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/app.py b/src/app.py
index 80adcd5..ca2ebf1 100644
--- a/src/app.py
+++ b/src/app.py
@@ -53,9 +53,13 @@ def setup_rabbit() -> None:
 
 def on_message(channel, method_frame, header_frame, body):
     msg_json = json.loads(body)
+    if 'probability' not in msg_json:
+        logging.error("Malformed message from broker")
     if msg_json['probability'] > 0.5:
         r = requests.get(f"http://{config.INPUT_HOSTNAME}/sample/{msg_json['tag']}")
         r.raise_for_status()
+        if 'device_id' not in r.json():
+            logging.error("Input-service response invalid")
         mqtt.publish(subtopic=r.json()['device_id'],
                      message=json.dumps({"command": "doAlert"}))