Add buffer to avoid forwarding duplicate messages from shoutbox

1 files changed, 24 insertions, 3 deletions

diff --git a/botmanager.py b/botmanager.py
index 48fad3c..e9b65a2 100644
--- a/botmanager.py
+++ b/botmanager.py
@@ -10,12 +10,16 @@ import logging
 from jsonfactory import JSONFactory
 from shoutboxapicommunicator import Communicator
 
+class ObsoleteMessageException(Exception):
+    pass
 
 class BotManager(object):
     shoutbox_api_url = "http://localhost:8000"
     telegram_chat_id = "default_id"
     api_call_interval = 10
 
+    last_message_timestamps = {}
+
     def __init__(self, token):
         self.token = token
 
@@ -44,14 +48,31 @@ class BotManager(object):
             self.forward_to_telegram(Communicator.fetch(self.shoutbox_api_url))
             time.sleep(self.api_call_interval)
 
+            new_message_dict = {}
+
+            for msg_id in self.last_message_timestamps:
+
+                # if the message is older than 2 * call interval,
+                # pop it from the buffer because there is no way
+                # it can be a duplicate anymore
+                timestamp = round(float(self.last_message_timestamps[msg_id]))
+                if not round(time.time()) - timestamp < 2 * self.api_call_interval:
+                    new_message_dict[msg_id] = self.last_message_timestamps[msg_id]
+                else:
+                    logging.info("Popped message from buffer.")
+
+            self.last_message_timestamps = new_message_dict
+
     def forward_to_telegram(self, messages):
         try:
             for message in messages:
-                self.bot.sendMessage(self.telegram_chat_id,
-                                     "{}: {}".format(message["user"], message["text"]))
+                if message["id"] not in self.last_message_timestamps:
+                    self.bot.sendMessage(self.telegram_chat_id,
+                                         "{}: {}".format(message["user"], message["text"]))
+                    self.last_message_timestamps[message["id"]] = message["timestamp"]
+
         except:
             logging.warning("Failed to send message to Telegram!")
-            traceback.print_exc()
 
     def default_action(self, chat_id):
         self.bot.sendMessage(chat_id, "Radio palaa keväällä 2017!")