62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
from aiogram import F, Router
|
||
from aiogram.filters import Command, CommandObject
|
||
from aiogram.fsm.context import FSMContext
|
||
from aiogram.types import CallbackQuery, Message
|
||
from aiogram.utils.deep_linking import create_start_link
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from config import settings
|
||
from keyboards.client import referals_kb
|
||
from misc.utils import format_share_deep_link
|
||
from schemas.di import DependenciesDTO
|
||
from texts import GENERAL_PROCESSING, REFERALS_MENU
|
||
|
||
router = Router()
|
||
|
||
|
||
@router.callback_query(F.data == "referal")
|
||
async def referal_menu(
|
||
cb: CallbackQuery, state: FSMContext, session: AsyncSession, deps: DependenciesDTO
|
||
):
|
||
await state.clear()
|
||
|
||
user = await deps.user_repository.get_user_by_id(session, cb.from_user.id)
|
||
referal_link = await create_start_link(cb.bot, f"ref_{user.id}", encode=True)
|
||
share_link = format_share_deep_link(referal_link)
|
||
|
||
await cb.message.edit_text(
|
||
REFERALS_MENU.format(balance=user.balance, link=referal_link),
|
||
reply_markup=referals_kb(share_link),
|
||
)
|
||
|
||
|
||
@router.message(F.chat.id == settings.admin_group_id, Command("refpay"))
|
||
async def refpay(
|
||
msg: Message,
|
||
command: CommandObject,
|
||
state: FSMContext,
|
||
session: AsyncSession,
|
||
deps: DependenciesDTO,
|
||
):
|
||
await state.clear()
|
||
|
||
await msg.delete()
|
||
status_msg = await msg.answer(GENERAL_PROCESSING)
|
||
user = await deps.user_service.fetch_user_by_thread(session, msg.message_thread_id)
|
||
referal = await deps.user_repository.get_user_by_id(session, user.referal_id)
|
||
if not referal:
|
||
await status_msg.edit_text("❌ У пользователя нет реферала.")
|
||
return
|
||
|
||
amount = command.args
|
||
|
||
if not amount.isdigit():
|
||
await status_msg.edit_text(
|
||
"❌ Укажите в аргументах количество средств, которые необходимо перевести рефералу пользователя."
|
||
)
|
||
return
|
||
|
||
amount = int(amount)
|
||
await deps.user_repository.increase_user_balance(session, referal.id, amount)
|
||
await status_msg.edit_text(f"<b>💸 Баланс реферала ({referal.id}) пополнен на {amount}₽</b>")
|