c0mrade
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')
|