Vladless
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
from datetime import datetime
|
|
|
|
from sqlalchemy import insert, select
|
|
from sqlalchemy.exc import SQLAlchemyError
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from pytz import timezone
|
|
|
|
from database.models import Payment
|
|
from logger import logger
|
|
|
|
|
|
MOSCOW_TZ = timezone("Europe/Moscow")
|
|
|
|
|
|
async def add_payment(
|
|
session: AsyncSession, tg_id: int, amount: float, payment_system: str
|
|
):
|
|
try:
|
|
now_moscow = datetime.now(MOSCOW_TZ).replace(tzinfo=None)
|
|
stmt = insert(Payment).values(
|
|
tg_id=tg_id,
|
|
amount=amount,
|
|
payment_system=payment_system,
|
|
status="success",
|
|
created_at=now_moscow,
|
|
)
|
|
await session.execute(stmt)
|
|
await session.commit()
|
|
logger.info(
|
|
f"✅ Успешно добавлен платёж: {tg_id}, {amount}₽ через {payment_system}"
|
|
)
|
|
except SQLAlchemyError as e:
|
|
logger.error(f"❌ Ошибка при добавлении платежа: {e}")
|
|
await session.rollback()
|
|
raise
|
|
|
|
|
|
async def get_last_payments(session: AsyncSession, tg_id: int, limit: int = 3):
|
|
try:
|
|
result = await session.execute(
|
|
select(Payment)
|
|
.where(Payment.tg_id == tg_id)
|
|
.order_by(Payment.created_at.desc())
|
|
.limit(limit)
|
|
)
|
|
payments = result.scalars().all()
|
|
logger.info(
|
|
f"✅ Получены последние платежи пользователя {tg_id}, всего: {len(payments)}"
|
|
)
|
|
return [dict(p.__dict__) for p in payments]
|
|
except SQLAlchemyError as e:
|
|
logger.error(f"❌ Ошибка при получении платежей пользователя {tg_id}: {e}")
|
|
return []
|