Add option to change API call interval

2 files changed, 6 insertions, 1 deletions

diff --git a/botmanager.py b/botmanager.py
index cd74f46..48fad3c 100644
--- a/botmanager.py
+++ b/botmanager.py
@@ -14,6 +14,7 @@ from shoutboxapicommunicator import Communicator
 class BotManager(object):
     shoutbox_api_url = "http://localhost:8000"
     telegram_chat_id = "default_id"
+    api_call_interval = 10
 
     def __init__(self, token):
         self.token = token
@@ -41,7 +42,7 @@ class BotManager(object):
         logging.info("Listening for messages...")
         while True:
             self.forward_to_telegram(Communicator.fetch(self.shoutbox_api_url))
-            time.sleep(10)
+            time.sleep(self.api_call_interval)
 
     def forward_to_telegram(self, messages):
         try:
diff --git a/radiodiodibot.py b/radiodiodibot.py
index 190cac0..022e685 100755
--- a/radiodiodibot.py
+++ b/radiodiodibot.py
@@ -20,6 +20,7 @@ parser.add_argument("telegram_bot_token", help="Telegram Bot API token")
 parser.add_argument("-U", "--shoutbox-api-url", help="Shoutbox API URL")
 parser.add_argument("-C", "--telegram-chat-id", help="Telegram Chat ID")
 parser.add_argument("-v", "--verbose", help="Verbose output", action="store_true")
+parser.add_argument("-i", "--interval", help="Delay between API update calls in seconds", type=int, default=10)
 args = parser.parse_args()
 
 if args.verbose:
@@ -54,8 +55,11 @@ def main():
     if telegram_chat_id is not None:
         manager.telegram_chat_id = telegram_chat_id
 
+    manager.api_call_interval = args.interval
+
     logging.info("Using Shoutbox API URL: {}".format(manager.shoutbox_api_url))
     logging.info("Using Telegram Chat ID: {}".format(manager.telegram_chat_id))
+    logging.info("Using API call interval of {} seconds.".format(manager.api_call_interval))
 
     manager.start()