Раис Юсупалиев
87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
"""PostgreSQL order repository."""
|
|
|
|
from contextlib import AbstractAsyncContextManager
|
|
from dataclasses import dataclass
|
|
from typing import Any
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker
|
|
|
|
from app.repositories.order.models import Order
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class OrderData:
|
|
order_uuid: str
|
|
payment_url: str
|
|
price: int
|
|
tariff_code: int
|
|
account_email: str
|
|
payload: dict[str, Any]
|
|
|
|
|
|
class OrderRepository:
|
|
def __init__(self, session_factory: async_sessionmaker[AsyncSession]) -> None:
|
|
self._session_factory = session_factory
|
|
|
|
def session(self) -> AbstractAsyncContextManager[AsyncSession]:
|
|
return self._session_factory.begin()
|
|
|
|
async def create_order(self, session: AsyncSession, order_data: OrderData) -> Order:
|
|
order = Order(
|
|
order_uuid=order_data.order_uuid,
|
|
payment_url=order_data.payment_url,
|
|
price=order_data.price,
|
|
tariff_code=order_data.tariff_code,
|
|
account_email=order_data.account_email,
|
|
payload=order_data.payload,
|
|
)
|
|
session.add(order)
|
|
await session.flush()
|
|
await session.refresh(order)
|
|
return order
|
|
|
|
async def get_order_by_order_uuid(
|
|
self,
|
|
session: AsyncSession,
|
|
order_uuid: str,
|
|
) -> Order | None:
|
|
result = await session.execute(
|
|
select(Order).where(Order.order_uuid == order_uuid)
|
|
)
|
|
return result.scalar_one_or_none()
|
|
|
|
async def mark_payment_status(
|
|
self,
|
|
session: AsyncSession,
|
|
order_uuid: str,
|
|
status: str,
|
|
payment_id: int,
|
|
) -> Order | None:
|
|
order = await self.get_order_by_order_uuid(session, order_uuid)
|
|
if order is None:
|
|
return None
|
|
|
|
order.payment_status = status
|
|
order.tbank_payment_id = payment_id
|
|
await session.flush()
|
|
return order
|
|
|
|
async def mark_cdek_order_registered(
|
|
self,
|
|
session: AsyncSession,
|
|
order_uuid: str,
|
|
cdek_order_uuid: str,
|
|
cdek_waybill_uuid: str | None = None,
|
|
cdek_waybill_url: str | None = None,
|
|
) -> Order | None:
|
|
order = await self.get_order_by_order_uuid(session, order_uuid)
|
|
if order is None:
|
|
return None
|
|
|
|
order.cdek_order_uuid = cdek_order_uuid
|
|
order.cdek_waybill_uuid = cdek_waybill_uuid
|
|
order.cdek_waybill_url = cdek_waybill_url
|
|
await session.flush()
|
|
return order
|