mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-02-27 23:00:53 +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
117 lines
3.6 KiB
Python
117 lines
3.6 KiB
Python
import logging
|
||
|
||
from aiogram import Dispatcher, F, types
|
||
from aiogram.exceptions import TelegramBadRequest, TelegramForbiddenError
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from app.database.models import User
|
||
from app.keyboards.admin import (
|
||
get_admin_report_result_keyboard,
|
||
get_admin_reports_keyboard,
|
||
)
|
||
from app.localization.texts import get_texts
|
||
from app.services.reporting_service import (
|
||
ReportingServiceError,
|
||
ReportPeriod,
|
||
reporting_service,
|
||
)
|
||
from app.utils.decorators import admin_required, error_handler
|
||
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
@admin_required
|
||
@error_handler
|
||
async def show_reports_menu(
|
||
callback: types.CallbackQuery,
|
||
db_user: User,
|
||
db: AsyncSession,
|
||
) -> None:
|
||
await callback.message.edit_text(
|
||
'📊 <b>Отчеты</b>\n\nВыберите период, чтобы отправить отчет в админский топик.',
|
||
reply_markup=get_admin_reports_keyboard(db_user.language),
|
||
parse_mode='HTML',
|
||
)
|
||
await callback.answer()
|
||
|
||
|
||
@admin_required
|
||
@error_handler
|
||
async def send_daily_report(
|
||
callback: types.CallbackQuery,
|
||
db_user: User,
|
||
db: AsyncSession,
|
||
) -> None:
|
||
await _send_report(callback, ReportPeriod.DAILY, db_user.language)
|
||
|
||
|
||
@admin_required
|
||
@error_handler
|
||
async def send_weekly_report(
|
||
callback: types.CallbackQuery,
|
||
db_user: User,
|
||
db: AsyncSession,
|
||
) -> None:
|
||
await _send_report(callback, ReportPeriod.WEEKLY, db_user.language)
|
||
|
||
|
||
@admin_required
|
||
@error_handler
|
||
async def send_monthly_report(
|
||
callback: types.CallbackQuery,
|
||
db_user: User,
|
||
db: AsyncSession,
|
||
) -> None:
|
||
await _send_report(callback, ReportPeriod.MONTHLY, db_user.language)
|
||
|
||
|
||
async def _send_report(
|
||
callback: types.CallbackQuery,
|
||
period: ReportPeriod,
|
||
language: str,
|
||
) -> None:
|
||
try:
|
||
report_text = await reporting_service.send_report(period, send_to_topic=True)
|
||
except ReportingServiceError as exc:
|
||
logger.warning('Не удалось отправить отчет: %s', exc)
|
||
await callback.answer(str(exc), show_alert=True)
|
||
return
|
||
except Exception as exc:
|
||
logger.error('Непредвиденная ошибка при отправке отчета: %s', exc)
|
||
await callback.answer('Не удалось отправить отчет. Попробуйте позже.', show_alert=True)
|
||
return
|
||
|
||
await callback.message.answer(
|
||
report_text,
|
||
reply_markup=get_admin_report_result_keyboard(language),
|
||
)
|
||
await callback.answer('Отчет отправлен в топик')
|
||
|
||
|
||
@admin_required
|
||
@error_handler
|
||
async def close_report_message(
|
||
callback: types.CallbackQuery,
|
||
db_user: User,
|
||
db: AsyncSession,
|
||
) -> None:
|
||
texts = get_texts(db_user.language)
|
||
|
||
try:
|
||
await callback.message.delete()
|
||
except (TelegramBadRequest, TelegramForbiddenError) as exc:
|
||
logger.warning('Не удалось закрыть сообщение отчета: %s', exc)
|
||
await callback.answer(texts.t('REPORT_CLOSE_ERROR', 'Не удалось закрыть отчет.'), show_alert=True)
|
||
return
|
||
|
||
await callback.answer(texts.t('REPORT_CLOSED', 'Отчет закрыт.'))
|
||
|
||
|
||
def register_handlers(dp: Dispatcher) -> None:
|
||
dp.callback_query.register(show_reports_menu, F.data == 'admin_reports')
|
||
dp.callback_query.register(send_daily_report, F.data == 'admin_reports_daily')
|
||
dp.callback_query.register(send_weekly_report, F.data == 'admin_reports_weekly')
|
||
dp.callback_query.register(send_monthly_report, F.data == 'admin_reports_monthly')
|
||
dp.callback_query.register(close_report_message, F.data == 'admin_close_report')
|