Merge pull request #67 from bento-platform/features/info-field

handle multiple response messages
bento-platform · Jan 9, 2024 · 6385ada · 6385ada
2 parents b384e98 + 849812b
commit 6385ada
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/bento_beacon/utils/beacon_response.py b/bento_beacon/utils/beacon_response.py
@@ -10,10 +10,14 @@ def init_response_data():
     g.response_data = {}
     g.response_info = {}
 
-
-# TODO: handle multiple messages
 def add_info_to_response(info):
-    g.response_info["message"] = info
+    add_message({"description": info, "level": "info"}) 
+
+
+def add_message(message_obj):
+    messages = g.response_info.get("messages", [])
+    messages.append(message_obj)
+    g.response_info["messages"] = messages
 
 
 def add_stats_to_response(ids):