Implement interval setting as runtime argument
This commit is contained in:
parent
1eb3ec8680
commit
5826aaf8f8
9
bot.py
9
bot.py
@ -1,6 +1,7 @@
|
|||||||
import argparse
|
import argparse
|
||||||
import asyncio
|
import asyncio
|
||||||
import queue
|
import queue
|
||||||
|
import sys
|
||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
|
|
||||||
@ -14,15 +15,19 @@ import utils
|
|||||||
|
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
parser.add_argument("-c", "--config", help="Config path")
|
parser.add_argument("-c", "--config", help="Config path")
|
||||||
|
parser.add_argument("-i", "--interval", type=int, help="Polling interval", required=True)
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
if args.interval <= 0:
|
||||||
|
print("interval must be greater than 0")
|
||||||
|
sys.exit(0)
|
||||||
|
|
||||||
config = toml.load(args.config)
|
config = toml.load(args.config)
|
||||||
|
|
||||||
vk_session = vk_api.VkApi(token=config["vk"]["access_token"])
|
vk_session = vk_api.VkApi(token=config["vk"]["access_token"])
|
||||||
vk = vk_session.get_api()
|
vk = vk_session.get_api()
|
||||||
|
|
||||||
mastodon_clients = {}
|
mastodon_clients = {}
|
||||||
bots_longpoll = {}
|
|
||||||
bot_threads = {}
|
bot_threads = {}
|
||||||
q = queue.Queue()
|
q = queue.Queue()
|
||||||
|
|
||||||
@ -102,7 +107,7 @@ print("Bot has been set up, listening events...")
|
|||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
asyncio.run(listen_new_posts())
|
asyncio.run(listen_new_posts())
|
||||||
time.sleep(5)
|
time.sleep(args.interval)
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
db.close()
|
db.close()
|
||||||
break
|
break
|
||||||
|
Loading…
Reference in New Issue
Block a user