Vladless
117 lines
4.3 KiB
Python
117 lines
4.3 KiB
Python
import asyncio
|
|
import os
|
|
|
|
import aiofiles
|
|
|
|
from aiogram import Bot
|
|
from aiogram.exceptions import TelegramForbiddenError, TelegramRetryAfter
|
|
from aiogram.types import BufferedInputFile, InlineKeyboardMarkup
|
|
|
|
from logger import logger
|
|
|
|
|
|
async def send_messages_with_limit(bot: Bot, messages: list[dict], messages_per_second: int = 25):
|
|
"""
|
|
Отправляет сообщения с ограничением по количеству сообщений в секунду.
|
|
"""
|
|
batch_size = messages_per_second
|
|
for i in range(0, len(messages), batch_size):
|
|
batch = messages[i:i + batch_size]
|
|
tasks = []
|
|
for msg in batch:
|
|
tasks.append(send_notification(
|
|
bot,
|
|
msg["tg_id"],
|
|
msg.get("photo"),
|
|
msg["text"],
|
|
msg.get("keyboard")
|
|
))
|
|
try:
|
|
await asyncio.gather(*tasks, return_exceptions=True)
|
|
except Exception as e:
|
|
logger.error(f"⚠ Ошибка при отправке сообщений в батче: {e}")
|
|
await asyncio.sleep(1.0)
|
|
|
|
|
|
def rate_limited_send(func):
|
|
async def wrapper(*args, **kwargs):
|
|
while True:
|
|
try:
|
|
return await func(*args, **kwargs)
|
|
except TelegramRetryAfter as e:
|
|
retry_in = int(e.retry_after) + 1
|
|
logger.warning(f"⚠️ Flood control: повтор через {retry_in} сек.")
|
|
await asyncio.sleep(retry_in)
|
|
except TelegramForbiddenError:
|
|
tg_id = kwargs.get("tg_id") or args[1]
|
|
logger.warning(f"Пользователь {tg_id} заблокировал бота.")
|
|
return False
|
|
except Exception as e:
|
|
tg_id = kwargs.get("tg_id") or args[1]
|
|
logger.error(f"❌ Ошибка отправки сообщения пользователю {tg_id}: {e}")
|
|
return False
|
|
return wrapper
|
|
|
|
|
|
async def send_notification(
|
|
bot: Bot,
|
|
tg_id: int,
|
|
image_filename: str | None,
|
|
caption: str,
|
|
keyboard: InlineKeyboardMarkup | None = None,
|
|
) -> bool:
|
|
"""
|
|
Отправляет уведомление пользователю.
|
|
"""
|
|
if image_filename is None:
|
|
return await _send_text_notification(bot, tg_id, caption, keyboard)
|
|
|
|
photo_path = os.path.join("img", image_filename)
|
|
if os.path.isfile(photo_path):
|
|
return await _send_photo_notification(bot, tg_id, photo_path, image_filename, caption, keyboard)
|
|
else:
|
|
logger.warning(f"Файл с изображением не найден: {photo_path}")
|
|
return await _send_text_notification(bot, tg_id, caption, keyboard)
|
|
|
|
|
|
@rate_limited_send
|
|
async def _send_photo_notification(
|
|
bot: Bot,
|
|
tg_id: int,
|
|
photo_path: str,
|
|
image_filename: str,
|
|
caption: str,
|
|
keyboard: InlineKeyboardMarkup | None = None,
|
|
) -> bool:
|
|
"""Отправляет уведомление с изображением."""
|
|
try:
|
|
async with aiofiles.open(photo_path, "rb") as image_file:
|
|
image_data = await image_file.read()
|
|
buffered_photo = BufferedInputFile(image_data, filename=image_filename)
|
|
await bot.send_photo(tg_id, buffered_photo, caption=caption, reply_markup=keyboard)
|
|
return True
|
|
except TelegramForbiddenError:
|
|
logger.error(f"Пользователь {tg_id} заблокировал бота")
|
|
return False
|
|
except Exception as e:
|
|
logger.error(f"Ошибка отправки фото для пользователя {tg_id}: {e}")
|
|
return await _send_text_notification(bot, tg_id, caption, keyboard)
|
|
|
|
|
|
@rate_limited_send
|
|
async def _send_text_notification(
|
|
bot: Bot,
|
|
tg_id: int,
|
|
caption: str,
|
|
keyboard: InlineKeyboardMarkup | None = None,
|
|
) -> bool:
|
|
"""Отправляет текстовое уведомление."""
|
|
try:
|
|
await bot.send_message(tg_id, caption, reply_markup=keyboard)
|
|
return True
|
|
except TelegramForbiddenError:
|
|
logger.error(f"Пользователь {tg_id} заблокировал бота")
|
|
return False
|
|
except Exception as e:
|
|
logger.error(f"Неизвестная ошибка при отправке сообщения для пользователя {tg_id}: {e}")
|
|
return False |