- Updated the pagination row to only display when multiple pages are available, improving the user interface for ad navigation. - Enhanced the inline keyboard structure for better clarity and usability in the admin panel's ads management section.
393 lines
16 KiB
Python
393 lines
16 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_ads_section", default="📈 Реклама"),
|
||
callback_data="admin_action:ads")
|
||
|
||
# Системные функции
|
||
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_ads_menu_keyboard(i18n_instance, lang: str) -> InlineKeyboardMarkup:
|
||
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
||
builder = InlineKeyboardBuilder()
|
||
builder.button(text=_(key="admin_ads_create_button", default="➕ Создать кампанию"),
|
||
callback_data="admin_action:ads_create")
|
||
builder.button(text=_(key="back_to_admin_panel_button"),
|
||
callback_data="admin_action:main")
|
||
builder.adjust(1, 1)
|
||
return builder.as_markup()
|
||
|
||
|
||
def get_ads_list_keyboard(
|
||
i18n_instance,
|
||
lang: str,
|
||
campaigns: list,
|
||
current_page: int,
|
||
total_pages: int,
|
||
) -> InlineKeyboardMarkup:
|
||
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
||
builder = InlineKeyboardBuilder()
|
||
|
||
for c in campaigns:
|
||
title = f"{c.source}"
|
||
builder.button(
|
||
text=title,
|
||
callback_data=f"admin_ads:card:{c.ad_campaign_id}:{current_page}",
|
||
)
|
||
|
||
# Pagination row (only when needed)
|
||
if total_pages > 1:
|
||
row = []
|
||
if current_page > 0:
|
||
row.append(
|
||
InlineKeyboardButton(
|
||
text="⬅️ " + _("prev_page_button", default="Prev"),
|
||
callback_data=f"admin_ads:page:{current_page - 1}",
|
||
)
|
||
)
|
||
row.append(
|
||
InlineKeyboardButton(
|
||
text=f"{current_page + 1}/{total_pages}",
|
||
callback_data="ads_page_display",
|
||
)
|
||
)
|
||
if current_page < total_pages - 1:
|
||
row.append(
|
||
InlineKeyboardButton(
|
||
text=_("next_page_button", default="Next") + " ➡️",
|
||
callback_data=f"admin_ads:page:{current_page + 1}",
|
||
)
|
||
)
|
||
if row:
|
||
builder.row(*row)
|
||
|
||
builder.button(text=_(key="admin_ads_create_button", default="➕ Создать кампанию"),
|
||
callback_data="admin_action:ads_create")
|
||
builder.button(text=_(key="back_to_admin_panel_button"),
|
||
callback_data="admin_action:main")
|
||
builder.adjust(1)
|
||
return builder.as_markup()
|
||
|
||
|
||
def get_ad_card_keyboard(i18n_instance, lang: str, campaign_id: int, back_page: int) -> InlineKeyboardMarkup:
|
||
_ = lambda key, **kwargs: i18n_instance.gettext(lang, key, **kwargs)
|
||
builder = InlineKeyboardBuilder()
|
||
builder.button(text=_(key="back_to_ads_list_button", default="⬅️ К списку"),
|
||
callback_data=f"admin_ads:page:{back_page}")
|
||
builder.button(text=_(key="back_to_admin_panel_button"),
|
||
callback_data="admin_action:main")
|
||
builder.adjust(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()
|