Skip to content

T
telegram_bot_antispammer
Commit 9119e4c5 authored by Иванова Мария Кирилловна's avatar Иванова Мария Кирилловна
Browse files
Options

Update bot.py

parent 4507e6eb
Hide whitespace changes
Inline Side-by-side
Showing with 33 additions and 48 deletions
bot.py
View file @ 9119e4c5
......@@ -4,6 +4,7 @@ from aiogram.types import ChatMemberUpdated, Message
from aiogram.exceptions import AiogramError
from pymorphy2 import MorphAnalyzer
from aiogram.types import ChatMember
from itertools import chain
import random
import time
import emoji
......@@ -46,12 +47,12 @@ SPAM_KEYWORDS = [
"спецпредложение","эксклюзивный предложение","мой тг канал", "уникальное предложение", "только сегодня", "лютая жиза", "мерч", "продукция", "коллекция",
# Работа
"удалённая работа","удалённая работа", "бакс", "доллар", "$", "сумма", "выручка", "прибыль", "нужен человек", "нужен помощник", "вакансии", "удаленка","условия труда","лс" , "писать лс", "писать в лс", "бабки", "пасивный заработок","без вложений",
"удалённая работа","удалённая работа", "бакс", "доллар", "$", "нужен человек", "нужен помощник", "вакансии", "удаленка","условия труда", "писать лс", "писать в лс", "лс", "партнер", "сотрудничество", "прибыль", "бабки", "пасивный заработок","без вложений", "подробности лс",
# Мошенничество
"лотерея", "быстрый выигрыш", "выигрыш", "приз", "деньги", "халява", "прибыль", "заработок", "зп", "подработка", "легкий заработок",
"мгновенно", "доход", "доход онлайн", "как заработать", "схема заработка",
"пассивный доход", "быстро заработать", "богатство", "всего за", "инвестировать", "инвестиция", "партнер",
"пассивный доход", "быстро заработать", "богатство", "всего за", "инвестировать", "инвестиция",
"биткоин", "криптовалюта", "forex", "деньги на карта", "дивиденд", "процент", "депозит", "выплата", "невероятный доход", "статус","гарантированный доход", "секрет богатство", "без риск",
# Психология, манипуляции и самопомощь
......@@ -177,6 +178,10 @@ def extract_emojis(text):
return ''.join(c for c in text if emoji.is_emoji(c))
def is_spam(text):
if text is None: # Проверяем, что текст не None
return False
# Нормализуем текст
normalized_words = normalize_text_to_infinitive(text)
......@@ -206,11 +211,7 @@ def is_spam(text):
spam_count += 1
matched_keywords.append(keyword) # Добавляем совпавшее слово в список
spam_count += emoji_count # Учитываем только спамные эмодзи
# Выводим совпавшие ключевые слова для диагностики
if matched_keywords:
print(f"Обнаружены спамные ключевые слова: {', '.join(matched_keywords)}")
spam_count += emoji_count # Учитываем только спам-эмодзи
# Возвращаем True, если найдено два или более совпадений
return spam_count >= 2
......@@ -236,7 +237,7 @@ async def welcome_new_user(event: ChatMemberUpdated):
chat_id=event.chat.id,
text=f"Здравствуйте, {new_user.first_name}! Добро пожаловать в нашу группу!\n\n"
f"Нам необходимо удостовериться, что вы человек.\n"
f"Решите математический пример в течение 1 минуты! В качестве ответа отправьте одно число.\n\n{math_problem}\n"
f"Решите математический пример в течение 2-х минут! В качестве ответа отправьте одно число.\n\n{math_problem}\n"
)
# Сохраняем данные о пользователе
......@@ -249,16 +250,14 @@ async def welcome_new_user(event: ChatMemberUpdated):
"chat_id": event.chat.id,
}
print(f"Приветственное сообщение отправлено пользователю {new_user.id} с примером {math_problem}")
except Exception as e:
print(f"Не удалось отправить сообщение в группу {event.chat.id}. Ошибка: {e}")
print(f"Failed sending a message to the group {event.chat.id}. Ошибка: {e}")
# Ожидание ответа пользователя
await check_answer(new_user.id)
async def check_answer(user_id):
"""Проверка ответа пользователя через 1 минуту."""
"""Проверка ответа пользователя через 1 минуты."""
await asyncio.sleep(60)
user_data = active_users.get(user_id)
......@@ -269,11 +268,10 @@ async def check_answer(user_id):
# Удаляем пользователя из группы
await bot.ban_chat_member(chat_id=user_data["chat_id"], user_id=user_id)
print(f"Пользователь {user_id} не ответил вовремя. Удален из группы.")
except AiogramError as e:
print(f"Ошибка при удалении пользователя {user_id}. Ошибка: {e}")
print(f"User deletion error {user_id}. Error: {e}")
except Exception as e:
print(f"Ошибка при удалении пользователя {user_id}. Ошибка: {e}")
print(f"User deletion error {user_id}. Error: {e}")
async def cleanup_messages(user_data):
"""Удаление всех сообщений пользователя и бота."""
......@@ -281,22 +279,19 @@ async def cleanup_messages(user_data):
# Удаляем все сообщения бота
for message_id in user_data["message_ids"]:
try:
print(f"Удаляю сообщение бота: chat_id={user_data['chat_id']}, message_id={message_id}")
await bot.delete_message(chat_id=user_data["chat_id"], message_id=message_id)
print(f"Сообщение бота для пользователя {user_data['chat_id']} удалено.")
except Exception as e:
print(f"Ошибка при удалении сообщения бота. Ошибка: {e}")
print(f"Bot message deletion error. Error: {e}")
# Удаляем сообщения пользователя
for user_message in user_data["user_messages"]:
try:
if user_message:
await bot.delete_message(chat_id=user_data["chat_id"], message_id=user_message.message_id)
print(f"Сообщение пользователя {user_message.message_id} удалено.")
except Exception as e:
print(f"Ошибка при удалении сообщения пользователя. Ошибка: {e}")
print(f"User message deletion error. Error: {e}")
except Exception as e:
print(f"Ошибка при удалении сообщений. Ошибка: {e}")
print(f"Message deletion error. Error: {e}")
@router.message()
async def handle_message(message: Message):
......@@ -308,10 +303,9 @@ async def handle_message(message: Message):
if chat_member.status not in ["member", "administrator", "creator"]:
# Если пользователь не участник, удаляем его сообщение
await message.delete()
print(f"Удалено сообщение от пользователя {message.from_user.id}, который не является участником группы.")
return
except AiogramError as e:
print(f"Ошибка при проверке статуса пользователя {message.from_user.id}. Ошибка: {e}")
print(f"User status verification error {message.from_user.id}. Error: {e}")
text = message.text or message.caption or ""
text = " ".join(normalize_text_to_infinitive(text)) # Убедимся, что это строка
......@@ -320,42 +314,35 @@ async def handle_message(message: Message):
if text and has_mixed_layout(text):
try:
await message.delete()
print(f"Удалено сообщение с смешанной раскладкой от пользователя {message.from_user.id}: {text}")
return
except Exception as e:
print(f"Ошибка при удалении сообщения с смешанной раскладкой. Ошибка: {e}")
print(f"Error deleting a message with a mixed layout. Error: {e}")
# Проверка на мультимедиа (фото/видео) и его подпись
if message.photo or message.video or message.document:
if text.strip(): # Проверяем, есть ли текст
try:
if is_spam(text): # Проверяем подпись к мультимедиа
await message.delete()
print(f"Удалено спамное мультимедиа сообщение от пользователя {message.from_user.id}.")
return
except Exception as e:
print(f"Ошибка при удалении мультимедиа сообщения. Ошибка: {e}")
if (message.photo or message.video or message.document) and text.strip() and is_spam(text):
try:
await message.delete()
return
except Exception as e:
print(f"Error deleting a multimedia message. Error: {e}")
# Проверка на кнопки/ссылки
if message.reply_markup and hasattr(message.reply_markup, 'inline_keyboard'):
for row in message.reply_markup.inline_keyboard:
for button in row:
if button.url and any(spam_word in button.url.lower() for spam_word in SPAM_KEYWORDS):
try:
await message.delete()
print(f"Удалено сообщение с подозрительной кнопкой от пользователя {message.from_user.id}.")
return
except Exception as e:
print(f"Ошибка при удалении сообщения с кнопкой. Ошибка: {e}")
for button in chain(*message.reply_markup.inline_keyboard):
if button.url and any(spam_word in button.url.lower() for spam_word in SPAM_KEYWORDS):
try:
await message.delete()
return
except Exception as e:
print(f"Error when deleting a message with a button. Error: {e}")
# Проверка на спам
if is_spam(message.text):
try:
await message.delete()
print(f"Удалено спамное сообщение от пользователя {message.from_user.id}: {message.text}")
return
except Exception as e:
print(f"Ошибка при удалении спамного сообщения. Ошибка: {e}")
print(f"Error when trying to delete spam messages. Error: {e}")
user_data = active_users.get(message.from_user.id)
......@@ -370,23 +357,21 @@ async def handle_message(message: Message):
# Ответ правильный — сохраняем сообщение и удаляем все
correct_message = await message.reply("Благодарим!")
user_data["message_ids"].append(correct_message.message_id) # Добавляем в список ID сообщений
print(f"Пользователь {message.from_user.id} ответил правильно.")
# Удаляем сообщения после правильного ответа
await cleanup_messages(user_data)
else:
# Ответ неправильный, добавляем новое сообщение бота
incorrect_message = await message.reply("Неправильный ответ. Попробуйте еще раз.")
user_data["message_ids"].append(incorrect_message.message_id)
print(f"Отправлено сообщение о неправильном ответе для пользователя {message.from_user.id}")
else:
# Если ответ не число
invalid_message = await message.reply("Пожалуйста, отправьте числовой ответ.")
user_data["message_ids"].append(invalid_message.message_id)
print(f"Отправлено сообщение с просьбой ввести числовой ответ для пользователя {message.from_user.id}")
async def main():
"""Запуск бота."""
print("bot is running")
await dp.start_polling(bot)
if __name__ == "__main__":
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment