Много переделал
This commit is contained in:
Disledg
170
handlers/profile.py
Normal file
170
handlers/profile.py
Normal file
@@ -0,0 +1,170 @@
|
||||
from aiogram import Router, types
|
||||
from aiogram.types import CallbackQuery
|
||||
import logging
|
||||
from datetime import datetime
|
||||
from aiogram.enums.parse_mode import ParseMode
|
||||
import locale
|
||||
from instences.config import BASE_URL_FASTAPI
|
||||
import aiohttp
|
||||
from keyboard.keyboards import account_keyboard, popup_keyboard, tranhist_keyboard, confirm_popup_keyboard, guide_keyboard, balance_keyboard
|
||||
|
||||
locale.setlocale(locale.LC_TIME, "ru_RU.UTF-8")
|
||||
|
||||
router = Router()
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
async def call_api(method, endpoint, data=None):
|
||||
"""
|
||||
Выполняет HTTP-запрос к FastAPI.
|
||||
"""
|
||||
url = f"{BASE_URL_FASTAPI}{endpoint}"
|
||||
logger.info(f"Инициализация запроса: {method} {url} с данными {data}")
|
||||
|
||||
try:
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.request(method, url, json=data) as response:
|
||||
logger.info(f"Получен ответ от {url}: статус {response.status}")
|
||||
|
||||
if response.status in {200, 201}:
|
||||
result = await response.json()
|
||||
logger.debug(f"Ответ JSON: {result}")
|
||||
return result
|
||||
if response.status == 404:
|
||||
logger.debug(f"Код {response.status}, возвращаю ничего")
|
||||
return None
|
||||
logger.error(f"Ошибка в запросе: статус {response.status}, причина {response.reason}")
|
||||
return "ERROR"
|
||||
except Exception as e:
|
||||
logger.exception(f"Исключение при выполнении запроса к {url}: {e}")
|
||||
return "ERROR"
|
||||
|
||||
@router.callback_query(lambda callback: callback.data == "profile")
|
||||
async def profile_callback_handler(callback: CallbackQuery):
|
||||
"""
|
||||
Обработчик callback_query для профиля.
|
||||
"""
|
||||
try:
|
||||
user_data = await call_api("GET", f"/user/{callback.from_user.id}")
|
||||
if not user_data:
|
||||
await callback.message.answer("Произошла ошибка, попробуйте позже или свяжитесь с администрацией.")
|
||||
await callback.answer()
|
||||
return
|
||||
|
||||
sub_data = await call_api("GET", f"/subscription/{user_data['id']}/last")
|
||||
if sub_data == "ERROR" or not isinstance(sub_data, dict):
|
||||
sub_data = None
|
||||
|
||||
balance_text = f"Баланс: {user_data['balance']} ₽"
|
||||
|
||||
if not sub_data:
|
||||
text = f"Профиль {callback.from_user.username}\n{balance_text}\nПополните баланс и приобретите подписку, чтобы получить активный статус (🐣,🦅)"
|
||||
else:
|
||||
expiry_date = sub_data.get("expiry_date")
|
||||
formatted_date = datetime.fromisoformat(expiry_date).strftime("%d %B %Y г.") if expiry_date else None
|
||||
is_expired = datetime.fromisoformat(expiry_date) < datetime.now() if expiry_date else True
|
||||
status_icon = "✖️" if is_expired else "☑️"
|
||||
profile_status = "🦅" if "Pro" in sub_data.get("plan", "") else "🐣"
|
||||
sub_text = (
|
||||
f"Ваша подписка действует до {formatted_date} {status_icon}"
|
||||
if not is_expired else f"Статус подписки: {status_icon}"
|
||||
)
|
||||
text = f"Профиль {callback.from_user.username} {profile_status}:\n{sub_text}\n{balance_text}"
|
||||
|
||||
await callback.message.edit_text(text, reply_markup=account_keyboard())
|
||||
except Exception as e:
|
||||
logger.exception(f"Ошибка в обработчике профиля: {e}")
|
||||
await callback.message.answer("Произошла ошибка. Попробуйте позже.")
|
||||
finally:
|
||||
await callback.answer()
|
||||
|
||||
@router.callback_query(lambda callback: callback.data == "balance")
|
||||
async def balance_callback_handler(callback: CallbackQuery):
|
||||
"""
|
||||
Обработчик callback_query для баланса.
|
||||
"""
|
||||
user_data = await call_api("GET", f"/user/{callback.from_user.id}")
|
||||
if not user_data:
|
||||
await callback.message.answer("Произошла ошибка, попробуйте позже или свяжитесь с администрацией.")
|
||||
await callback.answer()
|
||||
return
|
||||
|
||||
await callback.message.edit_text(
|
||||
f"Ваш баланс: {user_data['balance']} ₽. Выберите сумму для пополнения 🐥",
|
||||
reply_markup=balance_keyboard()
|
||||
)
|
||||
await callback.answer()
|
||||
|
||||
@router.callback_query(lambda callback: callback.data == "popup")
|
||||
async def popup_callback_handler(callback: CallbackQuery):
|
||||
"""
|
||||
Обработчик callback_query для пополнения.
|
||||
"""
|
||||
user = await call_api("GET", f"/user/{callback.from_user.id}")
|
||||
if not user:
|
||||
await callback.message.answer("Произошла ошибка, попробуйте позже или свяжитесь с администрацией.")
|
||||
await callback.answer()
|
||||
return
|
||||
|
||||
await callback.message.edit_text("Работает в режиме теста!!!", reply_markup=popup_keyboard())
|
||||
await callback.answer()
|
||||
|
||||
@router.callback_query(lambda callback: callback.data == "tranhist")
|
||||
async def tranhist_callback_handler(callback: CallbackQuery):
|
||||
"""
|
||||
Обработчик callback_query для истории транзакций.
|
||||
"""
|
||||
user_data = await call_api("GET", f"/user/{callback.from_user.id}")
|
||||
if not user_data:
|
||||
await callback.message.edit_text("Вы еще не зарегистрированы.")
|
||||
await callback.answer()
|
||||
return
|
||||
|
||||
try:
|
||||
transactions = await call_api("GET", f"/user/{user_data['id']}/transactions")
|
||||
if not transactions:
|
||||
await callback.message.edit_text("У вас нет транзакций.", reply_markup=tranhist_keyboard())
|
||||
await callback.answer()
|
||||
return
|
||||
|
||||
result = "Ваши транзакции:```\n"
|
||||
for count, tran in enumerate(transactions, start=1):
|
||||
dt = datetime.fromisoformat(tran['created_at']).strftime("%d.%m.%Y %H:%M:%S")
|
||||
result += f"{count}. Сумма: {tran['amount']}, Дата: {dt}\n"
|
||||
if len(result) > 4000:
|
||||
result += "...\nСлишком много транзакций для отображения."
|
||||
break
|
||||
result += "```"
|
||||
await callback.message.edit_text(result,parse_mode=ParseMode.MARKDOWN_V2, reply_markup=tranhist_keyboard())
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка обработки транзакций: {e}")
|
||||
await callback.message.edit_text("Произошла ошибка. Попробуйте позже.")
|
||||
finally:
|
||||
await callback.answer()
|
||||
|
||||
@router.callback_query(lambda callback: callback.data.startswith("popup:"))
|
||||
async def popup_confirm_callback_handler(callback: CallbackQuery):
|
||||
"""
|
||||
Обработчик подтверждения пополнения баланса.
|
||||
"""
|
||||
data = callback.data.split(":")
|
||||
popup_info = data[1]
|
||||
result = await call_api("POST", f"/user/{callback.from_user.id}/balance/{float(popup_info)}")
|
||||
if result == "ERROR":
|
||||
await callback.message.answer("Произошла ошибка, попробуйте позже или свяжитесь с администрацией.")
|
||||
await callback.answer()
|
||||
return
|
||||
|
||||
text = f"Вы пополнили свой баланс на {popup_info} ₽. Спасибо!"
|
||||
await callback.message.edit_text(text=text, reply_markup=confirm_popup_keyboard())
|
||||
await callback.answer()
|
||||
|
||||
@router.callback_query(lambda callback: callback.data == "guide")
|
||||
async def guide_callback_handler(callback: CallbackQuery):
|
||||
"""
|
||||
Обработчик callback_query для руководства.
|
||||
"""
|
||||
await callback.message.edit_text(
|
||||
"Руководство по использованию продукта:",
|
||||
reply_markup=guide_keyboard()
|
||||
)
|
||||
await callback.answer()
|
||||
Reference in New Issue
Block a user