From 5826aaf8f84acca340f06b6224a4b19edc5dc1ce Mon Sep 17 00:00:00 2001
From: ChronosX88 <chronosx88@gmail.com>
Date: Thu, 21 Jul 2022 19:01:57 +0300
Subject: [PATCH] Implement interval setting as runtime argument

---
 bot.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/bot.py b/bot.py
index 1acc6e7..231ff8f 100644
--- a/bot.py
+++ b/bot.py
@@ -1,6 +1,7 @@
 import argparse
 import asyncio
 import queue
+import sys
 import threading
 import time
 
@@ -14,15 +15,19 @@ import utils
 
 parser = argparse.ArgumentParser()
 parser.add_argument("-c", "--config", help="Config path")
+parser.add_argument("-i", "--interval", type=int, help="Polling interval", required=True)
 args = parser.parse_args()
 
+if args.interval <= 0:
+    print("interval must be greater than 0")
+    sys.exit(0)
+
 config = toml.load(args.config)
 
 vk_session = vk_api.VkApi(token=config["vk"]["access_token"])
 vk = vk_session.get_api()
 
 mastodon_clients = {}
-bots_longpoll = {}
 bot_threads = {}
 q = queue.Queue()
 
@@ -102,7 +107,7 @@ print("Bot has been set up, listening events...")
 while True:
     try:
         asyncio.run(listen_new_posts())
-        time.sleep(5)
+        time.sleep(args.interval)
     except KeyboardInterrupt:
         db.close()
         break