Implement message queue management and automatic sync on bot startup
- Added initialization of the message queue manager during bot startup, enhancing message handling capabilities. - Implemented automatic synchronization of the admin panel on startup, providing real-time updates and improved reliability. - Updated admin handlers to utilize the message queue for broadcasting messages, improving efficiency and error handling. - Introduced a new command for admins to check the status of message queues, enhancing monitoring and management capabilities. - Enhanced localization for new features and messages related to queue management and synchronization.
This commit is contained in:
@@ -1,6 +1,7 @@
|
|||||||
import logging
|
import logging
|
||||||
import asyncio
|
import asyncio
|
||||||
from aiogram import Router, F, types, Bot
|
from aiogram import Router, F, types, Bot
|
||||||
|
from aiogram.exceptions import TelegramRetryAfter
|
||||||
|
|
||||||
from aiogram.fsm.context import FSMContext
|
from aiogram.fsm.context import FSMContext
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
@@ -17,6 +18,7 @@ from bot.keyboards.inline.admin_keyboards import (
|
|||||||
get_admin_panel_keyboard,
|
get_admin_panel_keyboard,
|
||||||
)
|
)
|
||||||
from bot.middlewares.i18n import JsonI18n
|
from bot.middlewares.i18n import JsonI18n
|
||||||
|
from bot.utils.message_queue import get_queue_manager
|
||||||
|
|
||||||
router = Router(name="admin_broadcast_router")
|
router = Router(name="admin_broadcast_router")
|
||||||
|
|
||||||
@@ -171,22 +173,30 @@ async def confirm_broadcast_callback_handler(
|
|||||||
f"Admin {admin_user.id} broadcasting '{text[:50]}...' to {len(user_ids)} users."
|
f"Admin {admin_user.id} broadcasting '{text[:50]}...' to {len(user_ids)} users."
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Get message queue manager
|
||||||
|
queue_manager = get_queue_manager()
|
||||||
|
if not queue_manager:
|
||||||
|
await callback.message.edit_text("❌ Ошибка: система очередей не инициализирована", reply_markup=None)
|
||||||
|
return
|
||||||
|
|
||||||
|
# Queue all messages for sending
|
||||||
for uid in user_ids:
|
for uid in user_ids:
|
||||||
try:
|
try:
|
||||||
await bot.send_message(
|
await queue_manager.send_message(
|
||||||
chat_id=uid,
|
chat_id=uid,
|
||||||
text=text,
|
text=text,
|
||||||
entities=entities,
|
entities=entities,
|
||||||
)
|
)
|
||||||
sent_count += 1
|
sent_count += 1
|
||||||
|
|
||||||
|
# Log successful queuing
|
||||||
await message_log_dal.create_message_log(
|
await message_log_dal.create_message_log(
|
||||||
session,
|
session,
|
||||||
{
|
{
|
||||||
"user_id": admin_user.id,
|
"user_id": admin_user.id,
|
||||||
"telegram_username": admin_user.username,
|
"telegram_username": admin_user.username,
|
||||||
"telegram_first_name": admin_user.first_name,
|
"telegram_first_name": admin_user.first_name,
|
||||||
"event_type": "admin_broadcast_sent",
|
"event_type": "admin_broadcast_queued",
|
||||||
"content": f"To user {uid}: {text[:70]}...",
|
"content": f"To user {uid}: {text[:70]}...",
|
||||||
"is_admin_event": True,
|
"is_admin_event": True,
|
||||||
"target_user_id": uid,
|
"target_user_id": uid,
|
||||||
@@ -195,7 +205,7 @@ async def confirm_broadcast_callback_handler(
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
failed_count += 1
|
failed_count += 1
|
||||||
logging.warning(
|
logging.warning(
|
||||||
f"Failed to send broadcast to {uid}: {type(e).__name__} – {e}"
|
f"Failed to queue broadcast to {uid}: {type(e).__name__} – {e}"
|
||||||
)
|
)
|
||||||
await message_log_dal.create_message_log(
|
await message_log_dal.create_message_log(
|
||||||
session,
|
session,
|
||||||
@@ -209,7 +219,6 @@ async def confirm_broadcast_callback_handler(
|
|||||||
"target_user_id": uid,
|
"target_user_id": uid,
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
await asyncio.sleep(0.05)
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
await session.commit()
|
await session.commit()
|
||||||
@@ -217,7 +226,18 @@ async def confirm_broadcast_callback_handler(
|
|||||||
await session.rollback()
|
await session.rollback()
|
||||||
logging.error(f"Error committing broadcast logs: {e_commit}")
|
logging.error(f"Error committing broadcast logs: {e_commit}")
|
||||||
|
|
||||||
result_message = _("admin_broadcast_finished_stats", sent_count=sent_count, failed_count=failed_count)
|
# Get queue stats for detailed report
|
||||||
|
queue_stats = queue_manager.get_queue_stats()
|
||||||
|
|
||||||
|
result_message = f"""🚀 Рассылка поставлена в очередь!
|
||||||
|
📤 В очередь добавлено: {sent_count}
|
||||||
|
❌ Ошибок: {failed_count}
|
||||||
|
|
||||||
|
📊 Статус очередей:
|
||||||
|
👥 Очередь пользователей: {queue_stats['user_queue_size']} сообщений
|
||||||
|
📢 Очередь групп: {queue_stats['group_queue_size']} сообщений
|
||||||
|
|
||||||
|
ℹ️ Сообщения будут отправлены автоматически с соблюдением лимитов Telegram."""
|
||||||
await callback.message.answer(
|
await callback.message.answer(
|
||||||
result_message,
|
result_message,
|
||||||
reply_markup=get_back_to_admin_panel_keyboard(current_lang, i18n),
|
reply_markup=get_back_to_admin_panel_keyboard(current_lang, i18n),
|
||||||
|
|||||||
@@ -14,6 +14,7 @@ from bot.keyboards.inline.admin_keyboards import (
|
|||||||
from bot.middlewares.i18n import JsonI18n
|
from bot.middlewares.i18n import JsonI18n
|
||||||
from bot.services.panel_api_service import PanelApiService
|
from bot.services.panel_api_service import PanelApiService
|
||||||
from bot.services.subscription_service import SubscriptionService
|
from bot.services.subscription_service import SubscriptionService
|
||||||
|
from bot.utils.message_queue import get_queue_manager
|
||||||
|
|
||||||
from . import broadcast as admin_broadcast_handlers
|
from . import broadcast as admin_broadcast_handlers
|
||||||
from .promo import create as admin_promo_create_handlers
|
from .promo import create as admin_promo_create_handlers
|
||||||
@@ -120,6 +121,8 @@ async def admin_panel_actions_callback_handler(
|
|||||||
panel_service=panel_service,
|
panel_service=panel_service,
|
||||||
session=session)
|
session=session)
|
||||||
await callback.answer(_("admin_sync_initiated_from_panel"))
|
await callback.answer(_("admin_sync_initiated_from_panel"))
|
||||||
|
elif action == "queue_status":
|
||||||
|
await show_queue_status_handler(callback, i18n_data)
|
||||||
elif action == "main":
|
elif action == "main":
|
||||||
try:
|
try:
|
||||||
await callback.message.edit_text(
|
await callback.message.edit_text(
|
||||||
@@ -192,3 +195,52 @@ async def admin_section_handler(callback: types.CallbackQuery, state: FSMContext
|
|||||||
reply_markup=get_admin_panel_keyboard(i18n, current_lang, settings)
|
reply_markup=get_admin_panel_keyboard(i18n, current_lang, settings)
|
||||||
)
|
)
|
||||||
await callback.answer()
|
await callback.answer()
|
||||||
|
|
||||||
|
|
||||||
|
async def show_queue_status_handler(callback: types.CallbackQuery, i18n_data: dict):
|
||||||
|
"""Show message queue status to admin"""
|
||||||
|
current_lang = i18n_data.get("current_language", "ru")
|
||||||
|
i18n: Optional[JsonI18n] = i18n_data.get("i18n_instance")
|
||||||
|
if not i18n or not callback.message:
|
||||||
|
await callback.answer("Error processing request.", show_alert=True)
|
||||||
|
return
|
||||||
|
_ = lambda key, **kwargs: i18n.gettext(current_lang, key, **kwargs)
|
||||||
|
|
||||||
|
queue_manager = get_queue_manager()
|
||||||
|
if not queue_manager:
|
||||||
|
from aiogram.utils.keyboard import InlineKeyboardBuilder
|
||||||
|
await callback.message.edit_text(
|
||||||
|
"❌ Система очередей не инициализирована",
|
||||||
|
reply_markup=InlineKeyboardBuilder().button(
|
||||||
|
text=_("back_to_admin_panel_button"),
|
||||||
|
callback_data="admin_action:main"
|
||||||
|
).as_markup()
|
||||||
|
)
|
||||||
|
await callback.answer()
|
||||||
|
return
|
||||||
|
|
||||||
|
try:
|
||||||
|
stats = queue_manager.get_queue_stats()
|
||||||
|
|
||||||
|
message_text = _(
|
||||||
|
"admin_queue_status_info",
|
||||||
|
user_queue_size=stats['user_queue_size'],
|
||||||
|
user_processing="✅ Да" if stats['user_queue_processing'] else "❌ Нет",
|
||||||
|
user_recent=stats['user_recent_sends'],
|
||||||
|
group_queue_size=stats['group_queue_size'],
|
||||||
|
group_processing="✅ Да" if stats['group_queue_processing'] else "❌ Нет",
|
||||||
|
group_recent=stats['group_recent_sends']
|
||||||
|
)
|
||||||
|
|
||||||
|
from bot.keyboards.inline.admin_keyboards import get_back_to_admin_panel_keyboard
|
||||||
|
|
||||||
|
await callback.message.edit_text(
|
||||||
|
message_text,
|
||||||
|
reply_markup=get_back_to_admin_panel_keyboard(current_lang, i18n),
|
||||||
|
parse_mode="HTML"
|
||||||
|
)
|
||||||
|
await callback.answer()
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logging.error(f"Error getting queue status: {e}")
|
||||||
|
await callback.answer("❌ Ошибка получения статуса очередей", show_alert=True)
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ from datetime import datetime, timedelta, timezone
|
|||||||
from typing import Optional, List
|
from typing import Optional, List
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
|
||||||
from config.settings import Settings
|
from config.settings import Settings, get_settings
|
||||||
from db.dal import promo_code_dal
|
from db.dal import promo_code_dal
|
||||||
from db.models import PromoCode, PromoCodeActivation
|
from db.models import PromoCode, PromoCodeActivation
|
||||||
from bot.states.admin_states import AdminStates
|
from bot.states.admin_states import AdminStates
|
||||||
@@ -19,17 +19,27 @@ from bot.middlewares.i18n import JsonI18n
|
|||||||
router = Router(name="promo_manage_router")
|
router = Router(name="promo_manage_router")
|
||||||
|
|
||||||
|
|
||||||
|
def get_promo_status_emoji_and_text(promo: PromoCode, i18n: JsonI18n, current_lang: str):
|
||||||
|
"""Determine promo code status and return emoji + text"""
|
||||||
|
_ = lambda key, **kwargs: i18n.gettext(current_lang, key, **kwargs)
|
||||||
|
|
||||||
|
if promo.valid_until and promo.valid_until < datetime.now(timezone.utc):
|
||||||
|
return "⏰", _("admin_promo_status_expired")
|
||||||
|
elif promo.current_activations >= promo.max_activations:
|
||||||
|
return "🔄", _("admin_promo_status_used_up")
|
||||||
|
elif promo.is_active:
|
||||||
|
return "✅", _("admin_promo_status_active")
|
||||||
|
else:
|
||||||
|
return "🚫", _("admin_promo_status_inactive")
|
||||||
|
|
||||||
|
|
||||||
async def get_promo_detail_text_and_keyboard(promo_id: int, session: AsyncSession, i18n: JsonI18n, current_lang: str):
|
async def get_promo_detail_text_and_keyboard(promo_id: int, session: AsyncSession, i18n: JsonI18n, current_lang: str):
|
||||||
_ = lambda key, **kwargs: i18n.gettext(current_lang, key, **kwargs)
|
_ = lambda key, **kwargs: i18n.gettext(current_lang, key, **kwargs)
|
||||||
promo = await promo_code_dal.get_promo_code_by_id(session, promo_id)
|
promo = await promo_code_dal.get_promo_code_by_id(session, promo_id)
|
||||||
if not promo:
|
if not promo:
|
||||||
return None, None
|
return None, None
|
||||||
|
|
||||||
status = _("admin_promo_status_active") if promo.is_active else _("admin_promo_status_inactive")
|
_, status = get_promo_status_emoji_and_text(promo, i18n, current_lang)
|
||||||
if promo.valid_until and promo.valid_until < datetime.now(timezone.utc):
|
|
||||||
status = _("admin_promo_status_expired")
|
|
||||||
elif promo.current_activations >= promo.max_activations:
|
|
||||||
status = _("admin_promo_status_used_up")
|
|
||||||
|
|
||||||
validity = _("admin_promo_valid_indefinitely")
|
validity = _("admin_promo_valid_indefinitely")
|
||||||
if promo.valid_until:
|
if promo.valid_until:
|
||||||
@@ -68,7 +78,7 @@ async def view_promo_codes_handler(callback: types.CallbackQuery, i18n_data: dic
|
|||||||
promo_models = await promo_code_dal.get_all_active_promo_codes(session, limit=20, offset=0)
|
promo_models = await promo_code_dal.get_all_active_promo_codes(session, limit=20, offset=0)
|
||||||
text = f"{_('admin_active_promos_list_header')}\n\n{_('admin_no_active_promos')}" if not promo_models else "\n".join(
|
text = f"{_('admin_active_promos_list_header')}\n\n{_('admin_no_active_promos')}" if not promo_models else "\n".join(
|
||||||
[_("admin_active_promos_list_header"), ""] + [
|
[_("admin_active_promos_list_header"), ""] + [
|
||||||
f"🎟 <code>{p.code}</code> | 🎁 {p.bonus_days}д | 📊 {p.current_activations}/{p.max_activations} | ⏰ {p.valid_until.strftime('%d.%m.%Y') if p.valid_until else _('admin_promo_valid_indefinitely')}"
|
f"{get_promo_status_emoji_and_text(p, i18n, current_lang)[0]} <code>{p.code}</code> | 🎁 {p.bonus_days}д | 📊 {p.current_activations}/{p.max_activations} | ⏰ {p.valid_until.strftime('%d.%m.%Y') if p.valid_until else _('admin_promo_valid_indefinitely')}"
|
||||||
for p in promo_models
|
for p in promo_models
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
@@ -77,29 +87,66 @@ async def view_promo_codes_handler(callback: types.CallbackQuery, i18n_data: dic
|
|||||||
await callback.answer()
|
await callback.answer()
|
||||||
|
|
||||||
|
|
||||||
async def promo_management_handler(callback: types.CallbackQuery, i18n_data: dict, settings: Settings, session: AsyncSession):
|
async def promo_management_handler(callback: types.CallbackQuery, i18n_data: dict, settings: Settings, session: AsyncSession, page: int = 0):
|
||||||
current_lang = i18n_data.get("current_language", settings.DEFAULT_LANGUAGE)
|
current_lang = i18n_data.get("current_language", "ru")
|
||||||
i18n: Optional[JsonI18n] = i18n_data.get("i18n_instance")
|
i18n: Optional[JsonI18n] = i18n_data.get("i18n_instance")
|
||||||
if not i18n or not callback.message:
|
if not i18n or not callback.message:
|
||||||
await callback.answer("Error processing request.", show_alert=True)
|
await callback.answer("Error processing request.", show_alert=True)
|
||||||
return
|
return
|
||||||
_ = lambda key, **kwargs: i18n.gettext(current_lang, key, **kwargs)
|
_ = lambda key, **kwargs: i18n.gettext(current_lang, key, **kwargs)
|
||||||
|
|
||||||
promo_models = await promo_code_dal.get_all_promo_codes_with_details(session, limit=50, offset=0)
|
page_size = 10 # Количество промокодов на странице
|
||||||
if not promo_models:
|
offset = page * page_size
|
||||||
|
|
||||||
|
# Получаем общее количество промокодов
|
||||||
|
total_count = await promo_code_dal.get_promo_codes_count(session)
|
||||||
|
total_pages = (total_count + page_size - 1) // page_size if total_count > 0 else 1
|
||||||
|
|
||||||
|
promo_models = await promo_code_dal.get_all_promo_codes_with_details(session, limit=page_size, offset=offset)
|
||||||
|
if not promo_models and page == 0:
|
||||||
await callback.message.edit_text(_("admin_promo_management_empty"), reply_markup=get_back_to_admin_panel_keyboard(current_lang, i18n), parse_mode="HTML")
|
await callback.message.edit_text(_("admin_promo_management_empty"), reply_markup=get_back_to_admin_panel_keyboard(current_lang, i18n), parse_mode="HTML")
|
||||||
await callback.answer()
|
await callback.answer()
|
||||||
return
|
return
|
||||||
|
|
||||||
builder = InlineKeyboardBuilder()
|
builder = InlineKeyboardBuilder()
|
||||||
for promo in promo_models:
|
for promo in promo_models:
|
||||||
builder.row(InlineKeyboardButton(text=f"📝 {promo.code}", callback_data=f"promo_detail:{promo.promo_code_id}"))
|
status_emoji, _ = get_promo_status_emoji_and_text(promo, i18n, current_lang)
|
||||||
|
button_text = f"{status_emoji} {promo.code} ({promo.current_activations}/{promo.max_activations})"
|
||||||
|
builder.row(InlineKeyboardButton(text=button_text, callback_data=f"promo_detail:{promo.promo_code_id}"))
|
||||||
|
|
||||||
|
# Добавляем кнопки пагинации если есть больше одной страницы
|
||||||
|
if total_pages > 1:
|
||||||
|
pagination_buttons = []
|
||||||
|
if page > 0:
|
||||||
|
pagination_buttons.append(InlineKeyboardButton(text=_("prev_page_button"), callback_data=f"promo_management:{page-1}"))
|
||||||
|
if page < total_pages - 1:
|
||||||
|
pagination_buttons.append(InlineKeyboardButton(text=_("next_page_button"), callback_data=f"promo_management:{page+1}"))
|
||||||
|
|
||||||
|
if pagination_buttons:
|
||||||
|
builder.row(*pagination_buttons)
|
||||||
|
|
||||||
|
# Добавляем кнопки экспорта и возврата
|
||||||
|
builder.row(InlineKeyboardButton(text="📄 Экспорт CSV", callback_data="promo_export_all"))
|
||||||
builder.row(InlineKeyboardButton(text=_("back_to_admin_panel_button"), callback_data="admin_action:main"))
|
builder.row(InlineKeyboardButton(text=_("back_to_admin_panel_button"), callback_data="admin_action:main"))
|
||||||
|
|
||||||
await callback.message.edit_text(_("admin_promo_management_title"), reply_markup=builder.as_markup(), parse_mode="HTML")
|
# Формируем заголовок с информацией о страницах
|
||||||
|
title = _("admin_promo_management_title")
|
||||||
|
if total_pages > 1:
|
||||||
|
title += f"\n{_('admin_promo_list_page_info', current=page+1, total=total_pages, count=total_count)}"
|
||||||
|
|
||||||
|
await callback.message.edit_text(title, reply_markup=builder.as_markup(), parse_mode="HTML")
|
||||||
await callback.answer()
|
await callback.answer()
|
||||||
|
|
||||||
|
|
||||||
|
@router.callback_query(F.data.startswith("promo_management:"))
|
||||||
|
async def promo_management_pagination_handler(callback: types.CallbackQuery, i18n_data: dict, settings: Settings, session: AsyncSession):
|
||||||
|
try:
|
||||||
|
page = int(callback.data.split(":")[1])
|
||||||
|
await promo_management_handler(callback, i18n_data, settings, session, page)
|
||||||
|
except (ValueError, IndexError):
|
||||||
|
await callback.answer("Error processing pagination.", show_alert=True)
|
||||||
|
|
||||||
|
|
||||||
@router.callback_query(F.data.startswith("promo_detail:"))
|
@router.callback_query(F.data.startswith("promo_detail:"))
|
||||||
async def promo_detail_handler(callback: types.CallbackQuery, i18n_data: dict, session: AsyncSession):
|
async def promo_detail_handler(callback: types.CallbackQuery, i18n_data: dict, session: AsyncSession):
|
||||||
i18n: Optional[JsonI18n] = i18n_data.get("i18n_instance")
|
i18n: Optional[JsonI18n] = i18n_data.get("i18n_instance")
|
||||||
@@ -227,6 +274,63 @@ async def promo_export_activations_handler(callback: types.CallbackQuery, i18n_d
|
|||||||
await callback.answer()
|
await callback.answer()
|
||||||
|
|
||||||
|
|
||||||
|
@router.callback_query(F.data == "promo_export_all")
|
||||||
|
async def promo_export_all_handler(callback: types.CallbackQuery, i18n_data: dict, session: AsyncSession):
|
||||||
|
i18n: Optional[JsonI18n] = i18n_data.get("i18n_instance")
|
||||||
|
current_lang = i18n_data.get("current_language")
|
||||||
|
if not i18n or not callback.message or not current_lang:
|
||||||
|
return await callback.answer("Error processing request.", show_alert=True)
|
||||||
|
_ = lambda key, **kwargs: i18n.gettext(current_lang, key, **kwargs)
|
||||||
|
|
||||||
|
try:
|
||||||
|
await callback.answer("📄 Создаю CSV файл...", show_alert=True)
|
||||||
|
|
||||||
|
# Получаем все промокоды
|
||||||
|
all_promos = await promo_code_dal.get_all_promo_codes_with_details(session, limit=10000, offset=0)
|
||||||
|
|
||||||
|
output = io.StringIO()
|
||||||
|
writer = csv.writer(output)
|
||||||
|
|
||||||
|
# Заголовки CSV
|
||||||
|
writer.writerow([
|
||||||
|
"Код", "Бонусные дни", "Максимальные активации", "Текущие активации",
|
||||||
|
"Статус", "Активен", "Действителен до", "Создан", "Создал (Admin ID)"
|
||||||
|
])
|
||||||
|
|
||||||
|
for promo in all_promos:
|
||||||
|
# Определяем статус
|
||||||
|
status_emoji, status_text = get_promo_status_emoji_and_text(promo, i18n, current_lang)
|
||||||
|
|
||||||
|
# Формируем данные для CSV
|
||||||
|
row = [
|
||||||
|
promo.code,
|
||||||
|
promo.bonus_days,
|
||||||
|
promo.max_activations,
|
||||||
|
promo.current_activations,
|
||||||
|
status_text,
|
||||||
|
"Да" if promo.is_active else "Нет",
|
||||||
|
promo.valid_until.strftime("%Y-%m-%d %H:%M:%S") if promo.valid_until else "Без ограничений",
|
||||||
|
promo.created_at.strftime("%Y-%m-%d %H:%M:%S") if promo.created_at else "N/A",
|
||||||
|
promo.created_by_admin_id or "N/A"
|
||||||
|
]
|
||||||
|
writer.writerow(row)
|
||||||
|
|
||||||
|
output.seek(0)
|
||||||
|
|
||||||
|
# Создаем файл для отправки
|
||||||
|
filename = f"promo_codes_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv"
|
||||||
|
file = types.BufferedInputFile(
|
||||||
|
output.getvalue().encode('utf-8-sig'), # BOM для корректного отображения в Excel
|
||||||
|
filename=filename
|
||||||
|
)
|
||||||
|
|
||||||
|
caption = f"📄 Экспорт всех промокодов\n📊 Всего: {len(all_promos)} промокодов"
|
||||||
|
await callback.message.answer_document(file, caption=caption)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
await callback.answer(f"❌ Ошибка экспорта: {str(e)}", show_alert=True)
|
||||||
|
|
||||||
|
|
||||||
@router.callback_query(F.data.startswith("promo_delete:"))
|
@router.callback_query(F.data.startswith("promo_delete:"))
|
||||||
async def promo_delete_handler(callback: types.CallbackQuery, i18n_data: dict, session: AsyncSession):
|
async def promo_delete_handler(callback: types.CallbackQuery, i18n_data: dict, session: AsyncSession):
|
||||||
i18n: Optional[JsonI18n] = i18n_data.get("i18n_instance")
|
i18n: Optional[JsonI18n] = i18n_data.get("i18n_instance")
|
||||||
@@ -241,7 +345,7 @@ async def promo_delete_handler(callback: types.CallbackQuery, i18n_data: dict, s
|
|||||||
if promo:
|
if promo:
|
||||||
await session.commit()
|
await session.commit()
|
||||||
await callback.answer(_("admin_promo_deleted_success", code=promo.code), show_alert=True)
|
await callback.answer(_("admin_promo_deleted_success", code=promo.code), show_alert=True)
|
||||||
await promo_management_handler(callback, i18n_data, {}, session) # Settings not needed here
|
await promo_management_handler(callback, i18n_data, get_settings(), session, 0)
|
||||||
else:
|
else:
|
||||||
await callback.answer(_("admin_promo_not_found"), show_alert=True)
|
await callback.answer(_("admin_promo_not_found"), show_alert=True)
|
||||||
except (ValueError, IndexError):
|
except (ValueError, IndexError):
|
||||||
|
|||||||
@@ -216,7 +216,9 @@ async def start_command_handler(message: types.Message,
|
|||||||
f"Failed to update existing user {user_id} in session: {e_update}",
|
f"Failed to update existing user {user_id} in session: {e_update}",
|
||||||
exc_info=True)
|
exc_info=True)
|
||||||
|
|
||||||
await message.answer(_(key="welcome", user_name=hd.quote(user.full_name)))
|
# Send welcome message if not disabled
|
||||||
|
if not settings.DISABLE_WELCOME_MESSAGE:
|
||||||
|
await message.answer(_(key="welcome", user_name=hd.quote(user.full_name)))
|
||||||
|
|
||||||
# Auto-apply promo code if provided via start parameter
|
# Auto-apply promo code if provided via start parameter
|
||||||
if promo_code_to_apply:
|
if promo_code_to_apply:
|
||||||
|
|||||||
@@ -105,10 +105,12 @@ def get_system_functions_keyboard(i18n_instance, lang: str) -> InlineKeyboardMar
|
|||||||
callback_data="admin_action:broadcast")
|
callback_data="admin_action:broadcast")
|
||||||
builder.button(text=_(key="admin_sync_panel_button"),
|
builder.button(text=_(key="admin_sync_panel_button"),
|
||||||
callback_data="admin_action:sync_panel")
|
callback_data="admin_action:sync_panel")
|
||||||
|
builder.button(text=_(key="admin_queue_status_button"),
|
||||||
|
callback_data="admin_action:queue_status")
|
||||||
|
|
||||||
builder.button(text=_(key="back_to_admin_panel_button"),
|
builder.button(text=_(key="back_to_admin_panel_button"),
|
||||||
callback_data="admin_action:main")
|
callback_data="admin_action:main")
|
||||||
builder.adjust(2, 1)
|
builder.adjust(2, 1, 1)
|
||||||
return builder.as_markup()
|
return builder.as_markup()
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -42,6 +42,8 @@ from bot.services.tribute_service import TributeService, tribute_webhook_route
|
|||||||
from bot.services.crypto_pay_service import CryptoPayService, cryptopay_webhook_route
|
from bot.services.crypto_pay_service import CryptoPayService, cryptopay_webhook_route
|
||||||
|
|
||||||
from bot.handlers.user import payment as user_payment_webhook_module
|
from bot.handlers.user import payment as user_payment_webhook_module
|
||||||
|
from bot.handlers.admin.sync_admin import perform_sync
|
||||||
|
from bot.utils.message_queue import init_queue_manager
|
||||||
|
|
||||||
|
|
||||||
class DBSessionMiddleware(BaseMiddleware):
|
class DBSessionMiddleware(BaseMiddleware):
|
||||||
@@ -191,6 +193,34 @@ async def on_startup_configured(dispatcher: Dispatcher):
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.error(f"STARTUP: Failed to set bot commands: {e}", exc_info=True)
|
logging.error(f"STARTUP: Failed to set bot commands: {e}", exc_info=True)
|
||||||
|
|
||||||
|
# Initialize message queue manager
|
||||||
|
try:
|
||||||
|
queue_manager = init_queue_manager(bot)
|
||||||
|
dispatcher["queue_manager"] = queue_manager
|
||||||
|
logging.info("STARTUP: Message queue manager initialized")
|
||||||
|
except Exception as e:
|
||||||
|
logging.error(f"STARTUP: Failed to initialize message queue manager: {e}", exc_info=True)
|
||||||
|
|
||||||
|
# Automatic sync on startup
|
||||||
|
try:
|
||||||
|
logging.info("STARTUP: Running automatic panel sync...")
|
||||||
|
|
||||||
|
async with async_session_factory() as session:
|
||||||
|
sync_result = await perform_sync(
|
||||||
|
panel_service=panel_service,
|
||||||
|
session=session,
|
||||||
|
settings=settings,
|
||||||
|
i18n_instance=i18n_instance
|
||||||
|
)
|
||||||
|
|
||||||
|
if sync_result.get("status") == "completed":
|
||||||
|
logging.info(f"STARTUP: Automatic sync completed successfully. Details: {sync_result.get('details', 'N/A')}")
|
||||||
|
else:
|
||||||
|
logging.warning(f"STARTUP: Automatic sync completed with issues. Status: {sync_result.get('status', 'unknown')}")
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logging.error(f"STARTUP: Failed to run automatic sync: {e}", exc_info=True)
|
||||||
|
|
||||||
logging.info("STARTUP: Bot on_startup_configured completed.")
|
logging.info("STARTUP: Bot on_startup_configured completed.")
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -2,12 +2,14 @@ import logging
|
|||||||
import asyncio
|
import asyncio
|
||||||
from aiogram import Bot
|
from aiogram import Bot
|
||||||
from aiogram.utils.text_decorations import html_decoration as hd
|
from aiogram.utils.text_decorations import html_decoration as hd
|
||||||
|
from aiogram.exceptions import TelegramRetryAfter
|
||||||
from datetime import datetime, timezone
|
from datetime import datetime, timezone
|
||||||
from typing import Optional, Union, Dict, Any
|
from typing import Optional, Union, Dict, Any
|
||||||
|
|
||||||
from config.settings import Settings
|
from config.settings import Settings
|
||||||
from sqlalchemy.orm import sessionmaker
|
from sqlalchemy.orm import sessionmaker
|
||||||
from bot.middlewares.i18n import JsonI18n
|
from bot.middlewares.i18n import JsonI18n
|
||||||
|
from bot.utils.message_queue import get_queue_manager
|
||||||
|
|
||||||
|
|
||||||
class NotificationService:
|
class NotificationService:
|
||||||
@@ -19,16 +21,30 @@ class NotificationService:
|
|||||||
self.i18n = i18n
|
self.i18n = i18n
|
||||||
|
|
||||||
async def _send_to_log_channel(self, message: str, thread_id: Optional[int] = None):
|
async def _send_to_log_channel(self, message: str, thread_id: Optional[int] = None):
|
||||||
"""Send message to configured log channel/group"""
|
"""Send message to configured log channel/group using message queue"""
|
||||||
if not self.settings.LOG_CHAT_ID:
|
if not self.settings.LOG_CHAT_ID:
|
||||||
return
|
return
|
||||||
|
|
||||||
|
queue_manager = get_queue_manager()
|
||||||
|
if not queue_manager:
|
||||||
|
logging.warning("Message queue manager not available, falling back to direct send")
|
||||||
|
try:
|
||||||
|
await self.bot.send_message(
|
||||||
|
chat_id=self.settings.LOG_CHAT_ID,
|
||||||
|
text=message,
|
||||||
|
parse_mode="HTML",
|
||||||
|
disable_web_page_preview=True,
|
||||||
|
message_thread_id=thread_id or self.settings.LOG_THREAD_ID
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
logging.error(f"Failed to send notification to log channel {self.settings.LOG_CHAT_ID}: {e}")
|
||||||
|
return
|
||||||
|
|
||||||
try:
|
try:
|
||||||
# Use thread_id if provided, otherwise use from settings
|
# Use thread_id if provided, otherwise use from settings
|
||||||
final_thread_id = thread_id or self.settings.LOG_THREAD_ID
|
final_thread_id = thread_id or self.settings.LOG_THREAD_ID
|
||||||
|
|
||||||
kwargs = {
|
kwargs = {
|
||||||
"chat_id": self.settings.LOG_CHAT_ID,
|
|
||||||
"text": message,
|
"text": message,
|
||||||
"parse_mode": "HTML",
|
"parse_mode": "HTML",
|
||||||
"disable_web_page_preview": True
|
"disable_web_page_preview": True
|
||||||
@@ -38,26 +54,42 @@ class NotificationService:
|
|||||||
if final_thread_id:
|
if final_thread_id:
|
||||||
kwargs["message_thread_id"] = final_thread_id
|
kwargs["message_thread_id"] = final_thread_id
|
||||||
|
|
||||||
await self.bot.send_message(**kwargs)
|
# Queue message for sending (groups are rate limited to 15/minute)
|
||||||
|
await queue_manager.send_message(self.settings.LOG_CHAT_ID, **kwargs)
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.error(f"Failed to send notification to log channel {self.settings.LOG_CHAT_ID}: {e}")
|
logging.error(f"Failed to queue notification to log channel {self.settings.LOG_CHAT_ID}: {e}")
|
||||||
|
|
||||||
async def _send_to_admins(self, message: str):
|
async def _send_to_admins(self, message: str):
|
||||||
"""Send message to all admin users"""
|
"""Send message to all admin users using message queue"""
|
||||||
if not self.settings.ADMIN_IDS:
|
if not self.settings.ADMIN_IDS:
|
||||||
return
|
return
|
||||||
|
|
||||||
|
queue_manager = get_queue_manager()
|
||||||
|
if not queue_manager:
|
||||||
|
logging.warning("Message queue manager not available, falling back to direct send")
|
||||||
|
for admin_id in self.settings.ADMIN_IDS:
|
||||||
|
try:
|
||||||
|
await self.bot.send_message(
|
||||||
|
chat_id=admin_id,
|
||||||
|
text=message,
|
||||||
|
parse_mode="HTML",
|
||||||
|
disable_web_page_preview=True
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
logging.error(f"Failed to send notification to admin {admin_id}: {e}")
|
||||||
|
return
|
||||||
|
|
||||||
for admin_id in self.settings.ADMIN_IDS:
|
for admin_id in self.settings.ADMIN_IDS:
|
||||||
try:
|
try:
|
||||||
await self.bot.send_message(
|
await queue_manager.send_message(
|
||||||
chat_id=admin_id,
|
chat_id=admin_id,
|
||||||
text=message,
|
text=message,
|
||||||
parse_mode="HTML",
|
parse_mode="HTML",
|
||||||
disable_web_page_preview=True
|
disable_web_page_preview=True
|
||||||
)
|
)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.error(f"Failed to send notification to admin {admin_id}: {e}")
|
logging.error(f"Failed to queue notification to admin {admin_id}: {e}")
|
||||||
|
|
||||||
async def notify_new_user_registration(self, user_id: int, username: Optional[str] = None,
|
async def notify_new_user_registration(self, user_id: int, username: Optional[str] = None,
|
||||||
first_name: Optional[str] = None,
|
first_name: Optional[str] = None,
|
||||||
|
|||||||
@@ -563,6 +563,10 @@ class SubscriptionService:
|
|||||||
)
|
)
|
||||||
start_date = datetime.now(timezone.utc)
|
start_date = datetime.now(timezone.utc)
|
||||||
new_end_date_obj = start_date + timedelta(days=bonus_days)
|
new_end_date_obj = start_date + timedelta(days=bonus_days)
|
||||||
|
|
||||||
|
# For promo code activations, use the configured user traffic limit
|
||||||
|
traffic_limit = self.settings.user_traffic_limit_bytes if "promo code" in reason.lower() else self.settings.trial_traffic_limit_bytes
|
||||||
|
|
||||||
bonus_sub_payload = {
|
bonus_sub_payload = {
|
||||||
"user_id": user_id,
|
"user_id": user_id,
|
||||||
"panel_user_uuid": panel_uuid,
|
"panel_user_uuid": panel_uuid,
|
||||||
@@ -572,7 +576,7 @@ class SubscriptionService:
|
|||||||
"duration_months": 0,
|
"duration_months": 0,
|
||||||
"is_active": True,
|
"is_active": True,
|
||||||
"status_from_panel": "ACTIVE_BONUS",
|
"status_from_panel": "ACTIVE_BONUS",
|
||||||
"traffic_limit_bytes": self.settings.user_traffic_limit_bytes,
|
"traffic_limit_bytes": traffic_limit,
|
||||||
}
|
}
|
||||||
await subscription_dal.deactivate_other_active_subscriptions(
|
await subscription_dal.deactivate_other_active_subscriptions(
|
||||||
session, panel_uuid, panel_sub_uuid
|
session, panel_uuid, panel_sub_uuid
|
||||||
@@ -593,14 +597,23 @@ class SubscriptionService:
|
|||||||
)
|
)
|
||||||
|
|
||||||
if updated_sub_model:
|
if updated_sub_model:
|
||||||
|
# Prepare panel update payload
|
||||||
|
panel_update_payload = {
|
||||||
|
"expireAt": new_end_date_obj.isoformat(
|
||||||
|
timespec="milliseconds"
|
||||||
|
).replace("+00:00", "Z")
|
||||||
|
}
|
||||||
|
|
||||||
|
# For promo code activations, remove traffic limit
|
||||||
|
if "promo code" in reason.lower():
|
||||||
|
panel_update_payload["trafficLimitBytes"] = self.settings.user_traffic_limit_bytes
|
||||||
|
panel_update_payload["trafficLimitStrategy"] = self.settings.USER_TRAFFIC_STRATEGY
|
||||||
|
logging.info(f"Updating traffic limit for user {user_id} to {self.settings.user_traffic_limit_bytes} bytes due to promo code activation")
|
||||||
|
|
||||||
panel_update_success = (
|
panel_update_success = (
|
||||||
await self.panel_service.update_user_details_on_panel(
|
await self.panel_service.update_user_details_on_panel(
|
||||||
panel_uuid,
|
panel_uuid,
|
||||||
{
|
panel_update_payload,
|
||||||
"expireAt": new_end_date_obj.isoformat(
|
|
||||||
timespec="milliseconds"
|
|
||||||
).replace("+00:00", "Z")
|
|
||||||
},
|
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
if not panel_update_success:
|
if not panel_update_success:
|
||||||
|
|||||||
@@ -0,0 +1 @@
|
|||||||
|
# Bot utilities package
|
||||||
@@ -0,0 +1,183 @@
|
|||||||
|
import asyncio
|
||||||
|
import logging
|
||||||
|
from typing import Dict, Any, Callable, Awaitable, Optional
|
||||||
|
from dataclasses import dataclass
|
||||||
|
from datetime import datetime, timedelta
|
||||||
|
from collections import deque
|
||||||
|
from aiogram import Bot
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class QueuedMessage:
|
||||||
|
"""Represents a queued message with all necessary parameters"""
|
||||||
|
chat_id: int
|
||||||
|
method_name: str # 'send_message', 'edit_message_text', etc.
|
||||||
|
kwargs: Dict[str, Any]
|
||||||
|
callback: Optional[Callable[[Any], Awaitable[None]]] = None # Optional callback for result
|
||||||
|
|
||||||
|
|
||||||
|
class MessageQueue:
|
||||||
|
"""Message queue with rate limiting for Telegram API"""
|
||||||
|
|
||||||
|
def __init__(self, messages_per_second: float, burst_size: int = 5):
|
||||||
|
self.messages_per_second = messages_per_second
|
||||||
|
self.burst_size = burst_size
|
||||||
|
self.queue: deque[QueuedMessage] = deque()
|
||||||
|
self.last_send_times: deque[datetime] = deque()
|
||||||
|
self.is_processing = False
|
||||||
|
self.delay_between_messages = 1.0 / messages_per_second
|
||||||
|
|
||||||
|
async def add_message(self, message: QueuedMessage) -> None:
|
||||||
|
"""Add message to queue"""
|
||||||
|
self.queue.append(message)
|
||||||
|
if not self.is_processing:
|
||||||
|
asyncio.create_task(self._process_queue())
|
||||||
|
|
||||||
|
async def _process_queue(self) -> None:
|
||||||
|
"""Process messages from queue with rate limiting"""
|
||||||
|
if self.is_processing:
|
||||||
|
return
|
||||||
|
|
||||||
|
self.is_processing = True
|
||||||
|
|
||||||
|
try:
|
||||||
|
while self.queue:
|
||||||
|
# Check if we need to wait
|
||||||
|
await self._wait_if_needed()
|
||||||
|
|
||||||
|
# Get and process next message
|
||||||
|
message = self.queue.popleft()
|
||||||
|
try:
|
||||||
|
await self._send_message(message)
|
||||||
|
self.last_send_times.append(datetime.now())
|
||||||
|
|
||||||
|
# Keep only recent send times (last minute)
|
||||||
|
cutoff_time = datetime.now() - timedelta(seconds=60)
|
||||||
|
while self.last_send_times and self.last_send_times[0] < cutoff_time:
|
||||||
|
self.last_send_times.popleft()
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logging.error(f"Failed to send queued message to {message.chat_id}: {e}")
|
||||||
|
|
||||||
|
finally:
|
||||||
|
self.is_processing = False
|
||||||
|
|
||||||
|
async def _wait_if_needed(self) -> None:
|
||||||
|
"""Wait if we need to respect rate limits"""
|
||||||
|
if not self.last_send_times:
|
||||||
|
return
|
||||||
|
|
||||||
|
# Calculate time since last message
|
||||||
|
time_since_last = (datetime.now() - self.last_send_times[-1]).total_seconds()
|
||||||
|
|
||||||
|
if time_since_last < self.delay_between_messages:
|
||||||
|
wait_time = self.delay_between_messages - time_since_last
|
||||||
|
await asyncio.sleep(wait_time)
|
||||||
|
|
||||||
|
async def _send_message(self, message: QueuedMessage) -> Any:
|
||||||
|
"""Send a single message - to be implemented by subclass"""
|
||||||
|
raise NotImplementedError("Subclass must implement _send_message")
|
||||||
|
|
||||||
|
|
||||||
|
class TelegramMessageQueue(MessageQueue):
|
||||||
|
"""Telegram-specific message queue"""
|
||||||
|
|
||||||
|
def __init__(self, bot: Bot, messages_per_second: float, burst_size: int = 5):
|
||||||
|
super().__init__(messages_per_second, burst_size)
|
||||||
|
self.bot = bot
|
||||||
|
|
||||||
|
async def _send_message(self, message: QueuedMessage) -> Any:
|
||||||
|
"""Send message using bot method"""
|
||||||
|
method = getattr(self.bot, message.method_name)
|
||||||
|
result = await method(chat_id=message.chat_id, **message.kwargs)
|
||||||
|
|
||||||
|
# Call callback if provided
|
||||||
|
if message.callback:
|
||||||
|
await message.callback(result)
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
class MessageQueueManager:
|
||||||
|
"""Manager for different types of message queues"""
|
||||||
|
|
||||||
|
def __init__(self, bot: Bot):
|
||||||
|
self.bot = bot
|
||||||
|
|
||||||
|
# Different queues for different types of chats
|
||||||
|
self.group_queue = TelegramMessageQueue(
|
||||||
|
bot=bot,
|
||||||
|
messages_per_second=15/60, # 15 messages per minute for groups
|
||||||
|
burst_size=3
|
||||||
|
)
|
||||||
|
|
||||||
|
self.user_queue = TelegramMessageQueue(
|
||||||
|
bot=bot,
|
||||||
|
messages_per_second=25, # 25 messages per second for users
|
||||||
|
burst_size=10
|
||||||
|
)
|
||||||
|
|
||||||
|
def _is_group_chat(self, chat_id: int) -> bool:
|
||||||
|
"""Check if chat_id belongs to a group or channel"""
|
||||||
|
return str(chat_id).startswith('-100')
|
||||||
|
|
||||||
|
async def send_message(self, chat_id: int, **kwargs) -> None:
|
||||||
|
"""Queue a send_message call"""
|
||||||
|
queue = self.group_queue if self._is_group_chat(chat_id) else self.user_queue
|
||||||
|
message = QueuedMessage(
|
||||||
|
chat_id=chat_id,
|
||||||
|
method_name='send_message',
|
||||||
|
kwargs=kwargs
|
||||||
|
)
|
||||||
|
await queue.add_message(message)
|
||||||
|
|
||||||
|
async def edit_message_text(self, chat_id: int, **kwargs) -> None:
|
||||||
|
"""Queue an edit_message_text call"""
|
||||||
|
queue = self.group_queue if self._is_group_chat(chat_id) else self.user_queue
|
||||||
|
message = QueuedMessage(
|
||||||
|
chat_id=chat_id,
|
||||||
|
method_name='edit_message_text',
|
||||||
|
kwargs=kwargs
|
||||||
|
)
|
||||||
|
await queue.add_message(message)
|
||||||
|
|
||||||
|
async def send_document(self, chat_id: int, **kwargs) -> None:
|
||||||
|
"""Queue a send_document call"""
|
||||||
|
queue = self.group_queue if self._is_group_chat(chat_id) else self.user_queue
|
||||||
|
message = QueuedMessage(
|
||||||
|
chat_id=chat_id,
|
||||||
|
method_name='send_document',
|
||||||
|
kwargs=kwargs
|
||||||
|
)
|
||||||
|
await queue.add_message(message)
|
||||||
|
|
||||||
|
async def answer_callback_query(self, callback_query_id: str, **kwargs) -> None:
|
||||||
|
"""Send callback query answer immediately (not rate limited)"""
|
||||||
|
await self.bot.answer_callback_query(callback_query_id, **kwargs)
|
||||||
|
|
||||||
|
def get_queue_stats(self) -> Dict[str, Any]:
|
||||||
|
"""Get statistics about queues"""
|
||||||
|
return {
|
||||||
|
"group_queue_size": len(self.group_queue.queue),
|
||||||
|
"user_queue_size": len(self.user_queue.queue),
|
||||||
|
"group_queue_processing": self.group_queue.is_processing,
|
||||||
|
"user_queue_processing": self.user_queue.is_processing,
|
||||||
|
"group_recent_sends": len(self.group_queue.last_send_times),
|
||||||
|
"user_recent_sends": len(self.user_queue.last_send_times)
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# Global queue manager instance
|
||||||
|
_queue_manager: Optional[MessageQueueManager] = None
|
||||||
|
|
||||||
|
|
||||||
|
def init_queue_manager(bot: Bot) -> MessageQueueManager:
|
||||||
|
"""Initialize global queue manager"""
|
||||||
|
global _queue_manager
|
||||||
|
_queue_manager = MessageQueueManager(bot)
|
||||||
|
return _queue_manager
|
||||||
|
|
||||||
|
|
||||||
|
def get_queue_manager() -> Optional[MessageQueueManager]:
|
||||||
|
"""Get global queue manager instance"""
|
||||||
|
return _queue_manager
|
||||||
@@ -113,6 +113,7 @@ class Settings(BaseSettings):
|
|||||||
SUBSCRIPTION_MINI_APP_URL: Optional[str] = Field(default=None)
|
SUBSCRIPTION_MINI_APP_URL: Optional[str] = Field(default=None)
|
||||||
|
|
||||||
START_COMMAND_DESCRIPTION: Optional[str] = Field(default=None)
|
START_COMMAND_DESCRIPTION: Optional[str] = Field(default=None)
|
||||||
|
DISABLE_WELCOME_MESSAGE: bool = Field(default=False, description="Disable welcome message on /start command")
|
||||||
|
|
||||||
# Inline mode thumbnail URLs
|
# Inline mode thumbnail URLs
|
||||||
INLINE_REFERRAL_THUMBNAIL_URL: str = Field(default="https://cdn-icons-png.flaticon.com/512/1077/1077114.png")
|
INLINE_REFERRAL_THUMBNAIL_URL: str = Field(default="https://cdn-icons-png.flaticon.com/512/1077/1077114.png")
|
||||||
|
|||||||
@@ -64,6 +64,14 @@ async def get_all_promo_codes_with_details(session: AsyncSession, limit: int = 5
|
|||||||
return result.scalars().all()
|
return result.scalars().all()
|
||||||
|
|
||||||
|
|
||||||
|
async def get_promo_codes_count(session: AsyncSession) -> int:
|
||||||
|
"""Get total count of all promo codes"""
|
||||||
|
from sqlalchemy import func
|
||||||
|
stmt = select(func.count(PromoCode.promo_code_id))
|
||||||
|
result = await session.execute(stmt)
|
||||||
|
return result.scalar_one()
|
||||||
|
|
||||||
|
|
||||||
async def get_promo_activations_by_code_id(session: AsyncSession, promo_code_id: int, limit: Optional[int] = None, offset: int = 0) -> List[PromoCodeActivation]:
|
async def get_promo_activations_by_code_id(session: AsyncSession, promo_code_id: int, limit: Optional[int] = None, offset: int = 0) -> List[PromoCodeActivation]:
|
||||||
"""Get activation history for a specific promo code with optional pagination."""
|
"""Get activation history for a specific promo code with optional pagination."""
|
||||||
stmt = (select(PromoCodeActivation)
|
stmt = (select(PromoCodeActivation)
|
||||||
|
|||||||
@@ -150,6 +150,16 @@
|
|||||||
"admin_promo_invalid_values": "Неверные значения. {error}",
|
"admin_promo_invalid_values": "Неверные значения. {error}",
|
||||||
"admin_promo_invalid_format_general": "Ошибка парсинга деталей промокода. Проверьте формат.",
|
"admin_promo_invalid_format_general": "Ошибка парсинга деталей промокода. Проверьте формат.",
|
||||||
"admin_promo_created_success": "✅ Промокод <code>{code}</code> успешно создан!\nБонус: {bonus_days} дней\nМакс. активаций: {max_activations}\nДействителен: {valid_until_str}",
|
"admin_promo_created_success": "✅ Промокод <code>{code}</code> успешно создан!\nБонус: {bonus_days} дней\nМакс. активаций: {max_activations}\nДействителен: {valid_until_str}",
|
||||||
|
"admin_promo_set_validity_days": "⏰ Установить срок (дни)",
|
||||||
|
"admin_back_to_panel": "⬅️ В панель",
|
||||||
|
"admin_promo_unlimited": "♾️ Неограниченно",
|
||||||
|
"admin_bulk_promo_created_title": "📦 Массовое создание завершено",
|
||||||
|
"admin_bulk_promo_created_stats": "✅ Создано промокодов: {created_count}\n📅 Бонусные дни: {bonus_days}\n🔢 Максимальные активации: {max_activations}\n⏰ Действительны: {validity_info}",
|
||||||
|
"admin_bulk_promo_settings": "📝 Настройки промокодов",
|
||||||
|
"admin_promo_list_page_info": "Страница {current}/{total} ({count} промокодов)",
|
||||||
|
"admin_queue_status_button": "📊 Статус очередей",
|
||||||
|
"admin_queue_status_title": "📊 Статус очередей сообщений",
|
||||||
|
"admin_queue_status_info": "📤 <b>Очереди сообщений:</b>\n\n👥 <b>Пользователи (25 сообщ/сек):</b>\n 📋 В очереди: {user_queue_size}\n 🔄 Обрабатывается: {user_processing}\n 📈 Отправлено за минуту: {user_recent}\n\n📢 <b>Группы/каналы (15 сообщ/мин):</b>\n 📋 В очереди: {group_queue_size}\n 🔄 Обрабатывается: {group_processing}\n 📈 Отправлено за минуту: {group_recent}",
|
||||||
"admin_promo_creation_failed_duplicate": "❌ Ошибка: Промокод <code>{code}</code> уже существует.",
|
"admin_promo_creation_failed_duplicate": "❌ Ошибка: Промокод <code>{code}</code> уже существует.",
|
||||||
"admin_promo_creation_failed": "❌ Не удалось создать промокод. Пожалуйста, попробуйте позже.",
|
"admin_promo_creation_failed": "❌ Не удалось создать промокод. Пожалуйста, попробуйте позже.",
|
||||||
"admin_active_promos_list_header": "Активные промокоды:",
|
"admin_active_promos_list_header": "Активные промокоды:",
|
||||||
|
|||||||
Reference in New Issue
Block a user