izzzzzi
142 lines
5.0 KiB
Python
142 lines
5.0 KiB
Python
from datetime import datetime
|
|
from typing import Any
|
|
|
|
from aiogram import F, Router
|
|
from aiogram.fsm.context import FSMContext
|
|
from aiogram.fsm.state import State, StatesGroup
|
|
from aiogram.types import CallbackQuery, Message
|
|
|
|
from filters.admin import IsAdminFilter
|
|
from keyboards.admin.panel_kb import AdminPanelCallback, build_admin_back_kb
|
|
from keyboards.admin.sender_kb import AdminSenderCallback, build_clusters_kb, build_sender_kb
|
|
from logger import logger
|
|
|
|
|
|
router = Router()
|
|
|
|
|
|
class AdminSender(StatesGroup):
|
|
waiting_for_message = State()
|
|
|
|
|
|
@router.callback_query(
|
|
AdminPanelCallback.filter(F.action == "sender"),
|
|
IsAdminFilter(),
|
|
)
|
|
async def handle_sender(callback_query: CallbackQuery):
|
|
await callback_query.message.edit_text(
|
|
text="✍️ Выберите группу пользователей для рассылки:",
|
|
reply_markup=build_sender_kb(),
|
|
)
|
|
|
|
|
|
@router.callback_query(
|
|
AdminSenderCallback.filter(F.type != "cluster-select"),
|
|
IsAdminFilter(),
|
|
)
|
|
async def handle_sender_callback_text(
|
|
callback_query: CallbackQuery, callback_data: AdminSenderCallback, state: FSMContext
|
|
):
|
|
await callback_query.message.edit_text(
|
|
text="✍️ Введите текст сообщения для рассылки:",
|
|
reply_markup=build_admin_back_kb("sender"),
|
|
)
|
|
await state.update_data(type=callback_data.type, cluster_name=callback_data.data)
|
|
await state.set_state(AdminSender.waiting_for_message)
|
|
|
|
|
|
@router.callback_query(
|
|
AdminSenderCallback.filter(F.type == "cluster-select"),
|
|
IsAdminFilter(),
|
|
)
|
|
async def handle_sender_callback(callback_query: CallbackQuery, session: Any):
|
|
clusters = await session.fetch("SELECT DISTINCT cluster_name FROM servers")
|
|
await callback_query.message.answer(
|
|
"✍️ Выберите кластер для рассылки сообщений:",
|
|
reply_markup=build_clusters_kb(clusters),
|
|
)
|
|
|
|
|
|
@router.message(AdminSender.waiting_for_message, IsAdminFilter())
|
|
async def handle_message_input(message: Message, state: FSMContext, session: Any):
|
|
"""
|
|
Обрабатывает ввод сообщения для рассылки (поддержка текста + фото).
|
|
"""
|
|
text_message = message.html_text if message.text else None
|
|
photo = message.photo[-1].file_id if message.photo else None
|
|
photo_url = message.caption if message.photo and message.caption and message.caption.startswith("http") else None
|
|
|
|
if not text_message and message.caption:
|
|
text_message = message.caption
|
|
|
|
if not text_message and not photo and not photo_url:
|
|
await message.answer("⚠ Ошибка! Отправьте текст или изображение для рассылки.")
|
|
return
|
|
|
|
state_data = await state.get_data()
|
|
send_to = state_data.get("type", "all")
|
|
|
|
if send_to == "subscribed":
|
|
tg_ids = await session.fetch(
|
|
"""
|
|
SELECT DISTINCT c.tg_id
|
|
FROM connections c
|
|
JOIN keys k ON c.tg_id = k.tg_id
|
|
WHERE k.expiry_time > $1
|
|
""",
|
|
int(datetime.utcnow().timestamp() * 1000),
|
|
)
|
|
elif send_to == "unsubscribed":
|
|
tg_ids = await session.fetch(
|
|
"""
|
|
SELECT c.tg_id
|
|
FROM connections c
|
|
LEFT JOIN keys k ON c.tg_id = k.tg_id
|
|
GROUP BY c.tg_id
|
|
HAVING COUNT(k.tg_id) = 0 OR MAX(k.expiry_time) <= $1
|
|
""",
|
|
int(datetime.utcnow().timestamp() * 1000),
|
|
)
|
|
elif send_to == "cluster":
|
|
cluster_name = state_data.get("cluster_name")
|
|
tg_ids = await session.fetch(
|
|
"""
|
|
SELECT DISTINCT c.tg_id
|
|
FROM connections c
|
|
JOIN keys k ON c.tg_id = k.tg_id
|
|
JOIN servers s ON k.server_id = s.cluster_name
|
|
WHERE s.cluster_name = $1
|
|
""",
|
|
cluster_name,
|
|
)
|
|
else:
|
|
tg_ids = await session.fetch("SELECT DISTINCT tg_id FROM connections")
|
|
|
|
total_users = len(tg_ids)
|
|
success_count = 0
|
|
|
|
for record in tg_ids:
|
|
tg_id = record["tg_id"]
|
|
try:
|
|
if photo or photo_url:
|
|
await message.bot.send_photo(
|
|
chat_id=tg_id, photo=photo if photo else photo_url, caption=text_message, parse_mode="HTML"
|
|
)
|
|
else:
|
|
await message.bot.send_message(chat_id=tg_id, text=text_message, parse_mode="HTML")
|
|
|
|
success_count += 1
|
|
except Exception as e:
|
|
logger.error(f"❌ Ошибка отправки пользователю {tg_id}: {e}")
|
|
|
|
text = (
|
|
f"📤 <b>Рассылка завершена!</b>\n\n"
|
|
f"👥 <b>Всего пользователей:</b> {total_users}\n"
|
|
f"✅ <b>Доставлено:</b> {success_count}\n"
|
|
f"❌ <b>Не доставлено:</b> {total_users - success_count}"
|
|
)
|
|
|
|
await message.answer(text=text, reply_markup=build_admin_back_kb("stats"), parse_mode="HTML")
|
|
|
|
await state.clear()
|