machka-pasla
3d58f60a4d
- Integrated new service building functions to streamline bot initialization, enhancing maintainability and clarity. - Updated the dispatcher setup to include core services dynamically, reducing redundancy in service registration. - Added support for multiple broadcast targets in the admin handler, allowing for more flexible message distribution. - Enhanced localization for new broadcast target options, improving user experience in both English and Russian. - Implemented new database access methods to retrieve user IDs based on subscription status, optimizing broadcast logic.
316 lines
13 KiB
Python
316 lines
13 KiB
Python
from aiogram.utils.keyboard import InlineKeyboardBuilder, InlineKeyboardButton
|
|
from aiogram.types import InlineKeyboardMarkup, WebAppInfo
|
|
from typing import Optional, List, Any
|
|
import math
|
|
|
|
from config.settings import Settings
|
|
from bot.middlewares.i18n import JsonI18n
|
|
from db.models import User
|
|
|
|
|
|
def get_admin_panel_keyboard(i18n_instance, lang: str,
|
|
settings: Settings) -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
|
|
# Статистика и мониторинг
|
|
builder.button(text=_(key="admin_stats_and_monitoring_section"),
|
|
callback_data="admin_section:stats_monitoring")
|
|
|
|
# Управление пользователями
|
|
builder.button(text=_(key="admin_user_management_section"),
|
|
callback_data="admin_section:user_management")
|
|
|
|
# Промокоды и маркетинг
|
|
builder.button(text=_(key="admin_promo_marketing_section"),
|
|
callback_data="admin_section:promo_marketing")
|
|
|
|
# Системные функции
|
|
builder.button(text=_(key="admin_system_functions_section"),
|
|
callback_data="admin_section:system_functions")
|
|
|
|
builder.adjust(1)
|
|
return builder.as_markup()
|
|
|
|
|
|
def get_stats_monitoring_keyboard(i18n_instance, lang: str) -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
|
|
builder.button(text=_(key="admin_stats_button"),
|
|
callback_data="admin_action:stats")
|
|
builder.button(text=_(key="admin_view_payments_button", default="💰 Платежи"),
|
|
callback_data="admin_action:view_payments")
|
|
builder.button(text=_(key="admin_view_logs_menu_button"),
|
|
callback_data="admin_action:view_logs_menu")
|
|
|
|
builder.button(text=_(key="back_to_admin_panel_button"),
|
|
callback_data="admin_action:main")
|
|
builder.adjust(2, 1, 1)
|
|
return builder.as_markup()
|
|
|
|
|
|
def get_user_management_keyboard(i18n_instance, lang: str) -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
|
|
builder.button(text=_(key="admin_users_management_button"),
|
|
callback_data="admin_action:users_management")
|
|
builder.button(text=_(key="admin_ban_management_section"),
|
|
callback_data="admin_section:ban_management")
|
|
|
|
builder.button(text=_(key="back_to_admin_panel_button"),
|
|
callback_data="admin_action:main")
|
|
builder.adjust(2, 1)
|
|
return builder.as_markup()
|
|
|
|
|
|
def get_ban_management_keyboard(i18n_instance, lang: str) -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
|
|
builder.button(text=_(key="admin_ban_user_button"),
|
|
callback_data="admin_action:ban_user_prompt")
|
|
builder.button(text=_(key="admin_unban_user_button"),
|
|
callback_data="admin_action:unban_user_prompt")
|
|
builder.button(text=_(key="admin_view_banned_users_button"),
|
|
callback_data="admin_action:view_banned:0")
|
|
|
|
builder.button(text=_(key="back_to_user_management_button"),
|
|
callback_data="admin_section:user_management")
|
|
builder.adjust(2, 1, 1)
|
|
return builder.as_markup()
|
|
|
|
|
|
def get_promo_marketing_keyboard(i18n_instance, lang: str) -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
|
|
builder.button(text=_(key="admin_create_promo_button"),
|
|
callback_data="admin_action:create_promo")
|
|
builder.button(text=_(key="admin_create_bulk_promo_button"),
|
|
callback_data="admin_action:create_bulk_promo")
|
|
builder.button(text=_(key="admin_promo_management_button"),
|
|
callback_data="admin_action:promo_management")
|
|
|
|
builder.button(text=_(key="back_to_admin_panel_button"),
|
|
callback_data="admin_action:main")
|
|
builder.adjust(2, 1, 1)
|
|
return builder.as_markup()
|
|
|
|
|
|
def get_system_functions_keyboard(i18n_instance, lang: str) -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
|
|
builder.button(text=_(key="admin_broadcast_button"),
|
|
callback_data="admin_action:broadcast")
|
|
builder.button(text=_(key="admin_sync_panel_button"),
|
|
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"),
|
|
callback_data="admin_action:main")
|
|
builder.adjust(2, 1, 1)
|
|
return builder.as_markup()
|
|
|
|
|
|
def get_logs_menu_keyboard(i18n_instance, lang: str) -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
builder.button(text=_(key="admin_view_all_logs_button"),
|
|
callback_data="admin_logs:view_all:0")
|
|
builder.button(text=_(key="admin_view_user_logs_prompt_button"),
|
|
callback_data="admin_logs:prompt_user")
|
|
builder.button(text=_(key="admin_export_logs_csv_button"),
|
|
callback_data="admin_logs:export_csv")
|
|
builder.row(
|
|
InlineKeyboardButton(text=_(key="back_to_admin_panel_button"),
|
|
callback_data="admin_action:main"))
|
|
builder.adjust(2, 1, 1)
|
|
return builder.as_markup()
|
|
|
|
|
|
def get_logs_pagination_keyboard(
|
|
current_page: int,
|
|
total_pages: int,
|
|
base_callback_data: str,
|
|
i18n_instance,
|
|
lang: str,
|
|
back_to_logs_menu: bool = False) -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
row_buttons = []
|
|
if current_page > 0:
|
|
row_buttons.append(
|
|
InlineKeyboardButton(
|
|
text="⬅️ " + _("prev_page_button", default="Prev"),
|
|
callback_data=f"{base_callback_data}:{current_page - 1}"))
|
|
if current_page < total_pages - 1:
|
|
row_buttons.append(
|
|
InlineKeyboardButton(
|
|
text=_("next_page_button", default="Next") + " ➡️",
|
|
callback_data=f"{base_callback_data}:{current_page + 1}"))
|
|
|
|
if row_buttons: builder.row(*row_buttons)
|
|
|
|
if back_to_logs_menu:
|
|
builder.row(
|
|
InlineKeyboardButton(text=_(key="admin_logs_menu_title"),
|
|
callback_data="admin_action:view_logs_menu"))
|
|
else:
|
|
builder.row(
|
|
InlineKeyboardButton(text=_(key="back_to_admin_panel_button"),
|
|
callback_data="admin_action:main"))
|
|
return builder.as_markup()
|
|
|
|
|
|
def get_banned_users_keyboard(banned_users: List[User], current_page: int,
|
|
total_banned: int, i18n_instance: JsonI18n,
|
|
lang: str,
|
|
settings: Settings) -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
page_size = settings.LOGS_PAGE_SIZE
|
|
|
|
if not banned_users and total_banned == 0:
|
|
pass
|
|
|
|
for user_row in banned_users:
|
|
|
|
user_display_parts = []
|
|
if user_row.first_name:
|
|
user_display_parts.append(user_row.first_name)
|
|
if user_row.username:
|
|
user_display_parts.append(f"(@{user_row.username})")
|
|
if not user_display_parts:
|
|
user_display_parts.append(f"ID: {user_row.user_id}")
|
|
|
|
user_display = " ".join(user_display_parts).strip()
|
|
|
|
button_text = _("admin_banned_user_button_text",
|
|
user_display=user_display,
|
|
user_id=user_row.user_id)
|
|
builder.row(
|
|
InlineKeyboardButton(
|
|
text=button_text,
|
|
callback_data=
|
|
f"admin_user_card:{user_row.user_id}:{current_page}"))
|
|
|
|
if total_banned > page_size:
|
|
total_pages = math.ceil(total_banned / page_size)
|
|
pagination_buttons = []
|
|
if current_page > 0:
|
|
pagination_buttons.append(
|
|
InlineKeyboardButton(
|
|
text=_("prev_page_button"),
|
|
callback_data=f"admin_action:view_banned:{current_page - 1}"
|
|
))
|
|
pagination_buttons.append(
|
|
InlineKeyboardButton(text=f"{current_page + 1}/{total_pages}",
|
|
callback_data="stub_page_display"))
|
|
if current_page < total_pages - 1:
|
|
pagination_buttons.append(
|
|
InlineKeyboardButton(
|
|
text=_("next_page_button"),
|
|
callback_data=f"admin_action:view_banned:{current_page + 1}"
|
|
))
|
|
if pagination_buttons:
|
|
builder.row(*pagination_buttons)
|
|
|
|
builder.row(
|
|
InlineKeyboardButton(text=_("back_to_admin_panel_button"),
|
|
callback_data="admin_action:main"))
|
|
return builder.as_markup()
|
|
|
|
|
|
def get_user_card_keyboard(user_id: int,
|
|
is_banned: bool,
|
|
i18n_instance,
|
|
lang: str,
|
|
banned_list_page: int = 0) -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
if is_banned:
|
|
builder.button(
|
|
text=_(key="user_card_unban_button"),
|
|
callback_data=f"admin_unban_confirm:{user_id}:{banned_list_page}")
|
|
else:
|
|
builder.button(
|
|
text=_(key="user_card_ban_button"),
|
|
callback_data=f"admin_ban_confirm:{user_id}:{banned_list_page}")
|
|
builder.button(
|
|
text=_(key="user_card_back_to_banned_list_button"),
|
|
callback_data=f"admin_action:view_banned:{banned_list_page}")
|
|
builder.button(text=_(key="back_to_admin_panel_button"),
|
|
callback_data="admin_action:main")
|
|
builder.adjust(1)
|
|
return builder.as_markup()
|
|
|
|
|
|
def get_confirmation_keyboard(yes_callback_data: str, no_callback_data: str,
|
|
i18n_instance,
|
|
lang: str) -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
builder.button(text=_(key="yes_button"), callback_data=yes_callback_data)
|
|
builder.button(text=_(key="no_button"), callback_data=no_callback_data)
|
|
return builder.as_markup()
|
|
|
|
|
|
def get_broadcast_confirmation_keyboard(lang: str,
|
|
i18n_instance,
|
|
target: str = "all") -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
|
|
# Row: target selection (all / active / inactive)
|
|
target_all_label = _(
|
|
key="broadcast_target_all_button",
|
|
default="👥 Все"
|
|
)
|
|
target_active_label = _(
|
|
key="broadcast_target_active_button",
|
|
default="✅ Активные"
|
|
)
|
|
target_inactive_label = _(
|
|
key="broadcast_target_inactive_button",
|
|
default="⌛ Неактивные"
|
|
)
|
|
|
|
# Highlight current selection with a prefix
|
|
def mark_selected(label: str, is_selected: bool) -> str:
|
|
return ("• " + label) if is_selected else label
|
|
|
|
builder.button(
|
|
text=mark_selected(target_all_label, target == "all"),
|
|
callback_data="broadcast_target:all",
|
|
)
|
|
builder.button(
|
|
text=mark_selected(target_active_label, target == "active"),
|
|
callback_data="broadcast_target:active",
|
|
)
|
|
builder.button(
|
|
text=mark_selected(target_inactive_label, target == "inactive"),
|
|
callback_data="broadcast_target:inactive",
|
|
)
|
|
builder.adjust(3)
|
|
|
|
# Row: confirmation
|
|
builder.button(text=_(key="confirm_broadcast_send_button", default="🚀 Отправить"),
|
|
callback_data="broadcast_final_action:send")
|
|
builder.button(text=_(key="cancel_broadcast_button", default="❌ Отмена"),
|
|
callback_data="broadcast_final_action:cancel")
|
|
builder.adjust(2)
|
|
return builder.as_markup()
|
|
|
|
|
|
def get_back_to_admin_panel_keyboard(lang: str,
|
|
i18n_instance) -> InlineKeyboardMarkup:
|
|
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
|
builder = InlineKeyboardBuilder()
|
|
builder.button(text=_(key="back_to_admin_panel_button"),
|
|
callback_data="admin_action:main")
|
|
return builder.as_markup()
|