Migrating to Postgres from sqlite3

2025-05-20 18:41:32 +00:00
parent ac2f83c061
commit a56d804b58
38 changed files with 5017 additions and 4389 deletions
@@ -0,0 +1,115 @@
+import logging
+from typing import Optional, List, Dict, Any
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.future import select
+from sqlalchemy import update, func
+from sqlalchemy.orm import selectinload
+
+from db.models import Payment, User
+
+
+async def create_payment_record(session: AsyncSession,
+                                payment_data: Dict[str, Any]) -> Payment:
+
+    from .user_dal import get_user_by_id
+    user = await get_user_by_id(session, payment_data["user_id"])
+    if not user:
+
+        raise ValueError(
+            f"User with id {payment_data['user_id']} not found for creating payment."
+        )
+
+    if payment_data.get("promo_code_id"):
+        from .promo_code_dal import get_promo_code_by_id
+        promo = await get_promo_code_by_id(session,
+                                           payment_data["promo_code_id"])
+        if not promo:
+            raise ValueError(
+                f"Promo code with id {payment_data['promo_code_id']} not found."
+            )
+
+    new_payment = Payment(**payment_data)
+    session.add(new_payment)
+    await session.flush()
+    await session.refresh(new_payment)
+    logging.info(
+        f"Payment record {new_payment.payment_id} created for user {new_payment.user_id}"
+    )
+    return new_payment
+
+
+async def get_payment_by_yookassa_id(
+        session: AsyncSession, yookassa_payment_id: str) -> Optional[Payment]:
+    stmt = select(Payment).where(
+        Payment.yookassa_payment_id == yookassa_payment_id)
+    result = await session.execute(stmt)
+    return result.scalar_one_or_none()
+
+
+async def get_payment_by_db_id(session: AsyncSession,
+                               payment_db_id: int) -> Optional[Payment]:
+
+    stmt = select(Payment).where(Payment.payment_id == payment_db_id).options(
+        selectinload(Payment.user), selectinload(Payment.promo_code_used))
+    result = await session.execute(stmt)
+    return result.scalar_one_or_none()
+
+
+async def get_payment_by_db_id_with_promo(
+        session: AsyncSession, payment_db_id: int) -> Optional[Payment]:
+
+    stmt = select(Payment).where(Payment.payment_id == payment_db_id).options(
+        selectinload(Payment.promo_code_used))
+    result = await session.execute(stmt)
+    return result.scalar_one_or_none()
+
+
+async def update_payment_status_by_db_id(
+        session: AsyncSession,
+        payment_db_id: int,
+        new_status: str,
+        yk_payment_id: Optional[str] = None) -> Optional[Payment]:
+    payment = await get_payment_by_db_id(session, payment_db_id)
+    if payment:
+        payment.status = new_status
+        payment.updated_at = func.now()
+        if yk_payment_id and payment.yookassa_payment_id is None:
+            payment.yookassa_payment_id = yk_payment_id
+        await session.flush()
+        await session.refresh(payment)
+        logging.info(
+            f"Payment record {payment.payment_id} status updated to {new_status}."
+        )
+    else:
+        logging.warning(
+            f"Payment record with DB ID {payment_db_id} not found for status update."
+        )
+    return payment
+
+
+async def update_payment_status_by_yk_id(session: AsyncSession,
+                                         yookassa_payment_id: str,
+                                         new_status: str) -> Optional[Payment]:
+    payment = await get_payment_by_yookassa_id(session, yookassa_payment_id)
+    if payment:
+        payment.status = new_status
+        payment.updated_at = func.now()
+        await session.flush()
+        await session.refresh(payment)
+        logging.info(
+            f"Payment record with YK ID {yookassa_payment_id} status updated to {new_status}."
+        )
+    else:
+        logging.warning(
+            f"Payment record with YK ID {yookassa_payment_id} not found for status update."
+        )
+    return payment
+
+
+async def get_recent_payment_logs_with_user(session: AsyncSession,
+                                            limit: int = 20,
+                                            offset: int = 0) -> List[Payment]:
+    stmt = (select(Payment).options(selectinload(Payment.user)).order_by(
+        Payment.created_at.desc()).limit(limit).offset(offset))
+    result = await session.execute(stmt)
+    return result.scalars().all()