Egor
87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
import logging
|
|
|
|
from aiogram import Dispatcher, F, types
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.database.crud.subscription import (
|
|
get_trial_statistics,
|
|
reset_trials_for_users_without_paid_subscription,
|
|
)
|
|
from app.database.models import User
|
|
from app.keyboards.admin import get_admin_trials_keyboard
|
|
from app.localization.texts import get_texts
|
|
from app.utils.decorators import admin_required, error_handler
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@admin_required
|
|
@error_handler
|
|
async def show_trials_panel(
|
|
callback: types.CallbackQuery,
|
|
db_user: User,
|
|
db: AsyncSession,
|
|
):
|
|
texts = get_texts(db_user.language)
|
|
|
|
stats = await get_trial_statistics(db)
|
|
message = texts.t("ADMIN_TRIALS_TITLE", "🧪 Управление триалами") + "\n\n" + texts.t(
|
|
"ADMIN_TRIALS_STATS",
|
|
"• Использовано всего: {used}\n"
|
|
"• Активно сейчас: {active}\n"
|
|
"• Доступно к сбросу: {resettable}",
|
|
).format(
|
|
used=stats.get("used_trials", 0),
|
|
active=stats.get("active_trials", 0),
|
|
resettable=stats.get("resettable_trials", 0),
|
|
)
|
|
|
|
await callback.message.edit_text(
|
|
message,
|
|
reply_markup=get_admin_trials_keyboard(db_user.language),
|
|
)
|
|
await callback.answer()
|
|
|
|
|
|
@admin_required
|
|
@error_handler
|
|
async def reset_trials(
|
|
callback: types.CallbackQuery,
|
|
db_user: User,
|
|
db: AsyncSession,
|
|
):
|
|
texts = get_texts(db_user.language)
|
|
|
|
reset_count = await reset_trials_for_users_without_paid_subscription(db)
|
|
stats = await get_trial_statistics(db)
|
|
|
|
message = texts.t(
|
|
"ADMIN_TRIALS_RESET_RESULT",
|
|
"♻️ Сбросили {reset_count} триалов.\n\n"
|
|
"• Использовано всего: {used}\n"
|
|
"• Активно сейчас: {active}\n"
|
|
"• Доступно к сбросу: {resettable}",
|
|
).format(
|
|
reset_count=reset_count,
|
|
used=stats.get("used_trials", 0),
|
|
active=stats.get("active_trials", 0),
|
|
resettable=stats.get("resettable_trials", 0),
|
|
)
|
|
|
|
await callback.message.edit_text(
|
|
message,
|
|
reply_markup=get_admin_trials_keyboard(db_user.language),
|
|
)
|
|
await callback.answer(texts.t("ADMIN_TRIALS_RESET_TOAST", "✅ Сброс завершен"))
|
|
|
|
|
|
def register_handlers(dp: Dispatcher) -> None:
|
|
dp.callback_query.register(
|
|
show_trials_panel,
|
|
F.data == "admin_trials",
|
|
)
|
|
dp.callback_query.register(
|
|
reset_trials,
|
|
F.data == "admin_trials_reset",
|
|
)
|