From 03f722682ac37ee5ea7b16070ae93b77bd34e927 Mon Sep 17 00:00:00 2001 From: Georgios Atheridis Date: Mon, 2 May 2022 18:05:41 +0000 Subject: [PATCH] fixed a bug that only read one auto_message --- skgyorugo/analyze_auto_message.py | 67 +++++++++++++++---------------- skgyorugo/main.py | 6 ++- 2 files changed, 38 insertions(+), 35 deletions(-) diff --git a/skgyorugo/analyze_auto_message.py b/skgyorugo/analyze_auto_message.py index fc3b0a3..f056664 100644 --- a/skgyorugo/analyze_auto_message.py +++ b/skgyorugo/analyze_auto_message.py @@ -35,41 +35,40 @@ def do_auto_message(bot: Bot, message: Message, auto_message_modules: dict): last_used ASC """ ) - fetched = c.fetchone() - if not fetched: - conn.close() - return + while True: + fetched = c.fetchone() + if not fetched: + break - name, cooldown, end_time, last_used, value = fetched - print(fetched) - if time.time() < last_used + cooldown: - return - if time.time() > start_stream_ts + end_time and end_time != 0: - return - if value: - tools.smart_privmsg.send(bot, message, value) - else: - try: - auto_message_modules[name].main(bot, message) - except KeyError: - c.execute( - """ - DELETE FROM - auto_messages - WHERE - name = ? - """, - (name, ) - ) - conn.commit() - return + name, cooldown, end_time, last_used, value = fetched + if time.time() < last_used + cooldown: + continue + if time.time() > start_stream_ts + end_time and end_time != 0: + continue + if value: + tools.smart_privmsg.send(bot, message, value) + else: + try: + auto_message_modules[name].main(bot, message) + except KeyError: + c.execute( + """ + DELETE FROM + auto_messages + WHERE + name = ? + """, + (name, ) + ) + conn.commit() + continue - c.execute( - "UPDATE auto_messages SET last_used = ? WHERE name = ?", - ( - int(time.time()), - name, + c.execute( + "UPDATE auto_messages SET last_used = ? WHERE name = ?", + ( + int(time.time()), + name, + ) ) - ) - conn.commit() + conn.commit() conn.close() diff --git a/skgyorugo/main.py b/skgyorugo/main.py index 01b376b..58994a6 100644 --- a/skgyorugo/main.py +++ b/skgyorugo/main.py @@ -26,7 +26,11 @@ reload(scripts.chatting) reload(database_manager) reload(analyze_auto_message) -logging.basicConfig(filename="/var/log/aptbot/logs.log", level=logging.DEBUG) +logging.basicConfig( + filename="/var/log/aptbot/logs.log", + level=logging.DEBUG, + format="[%(levelname)s] %(asctime)s: %(message)s" +) PATH = os.path.dirname(os.path.realpath(__file__)) logging.info(f"Defined PATH: {PATH}") -- 2.30.2