Egor
118 lines
3.6 KiB
Python
118 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 (
|
|
ReportPeriod,
|
|
ReportingServiceError,
|
|
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: # noqa: BLE001
|
|
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")
|
|
|