From 247f3a18f0aa2cd2ca784bf3bb2b8eeb22fecc28 Mon Sep 17 00:00:00 2001 From: Rykov7 Date: Sat, 25 Feb 2023 22:31:25 +0300 Subject: [PATCH] add checking antispam setting --- app/bot.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/app/bot.py b/app/bot.py index 2064fca..ac0b13e 100644 --- a/app/bot.py +++ b/app/bot.py @@ -7,6 +7,8 @@ from . import rules from .helpers import represent_as_get, detect_args, update_stats from .filters import * from .config import * +from .database import session +from .models import Chat """ [ ANTISPAM ] """ @@ -39,10 +41,11 @@ def catch_videos(message: types.Message): @bot.edited_message_handler(func=in_delete_list) @bot.message_handler(func=in_delete_list) def delete_message(message: types.Message): - """Delete unwanted message.""" - bot.delete_message(message.chat.id, message.id) - with shelve.open(f"{DATA}{message.chat.id}") as s: - s["Deleted"] += 1 + if session.query(Chat.antispam).filter_by(chat_id=message.chat.id).first()[0]: + """Delete unwanted message.""" + bot.delete_message(message.chat.id, message.id) + with shelve.open(f"{DATA}{message.chat.id}") as s: + s["Deleted"] += 1 """ [ COMMANDS ] """