machka-pasla
c6292a1c8e
- Introduced ad campaign management functionality, including the ability to create campaigns and attribute users based on ad parameters. - Updated user command handlers to process new ad-related parameters and log user interactions with ad campaigns. - Enhanced admin interfaces with new keyboard options for managing ads and displaying campaign information. - Added database models for ad campaigns and attributions, improving data management for advertising features. - Updated localization files to support new ad-related strings in both English and Russian.
331 lines
13 KiB
Python
331 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_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_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()
|