mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-03-04 21:04:00 +00:00
9a2aea038a
- Add pyproject.toml with uv and ruff configuration - Pin Python version to 3.13 via .python-version - Add Makefile commands: lint, format, fix - Apply ruff formatting to entire codebase - Remove unused imports (base64 in yookassa/simple_subscription) - Update .gitignore for new config files
90 lines
2.6 KiB
Python
90 lines
2.6 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',
|
||
)
|