from datetime import datetime

from pytz import timezone
from sqlalchemy import insert, select
from sqlalchemy.exc import SQLAlchemyError
from sqlalchemy.ext.asyncio import AsyncSession

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 []