Раис Юсупалиев
95 lines
2.7 KiB
Python
95 lines
2.7 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
|
|
delivery_type: int
|
|
tariff_code: int
|
|
sender: dict[str, Any]
|
|
recipient: dict[str, Any]
|
|
from_location: dict[str, Any]
|
|
to_location: dict[str, Any]
|
|
packages: list[dict[str, Any]]
|
|
services: list[dict[str, Any]] | None
|
|
comment: str | None
|
|
|
|
|
|
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,
|
|
delivery_type=order_data.delivery_type,
|
|
tariff_code=order_data.tariff_code,
|
|
sender=order_data.sender,
|
|
recipient=order_data.recipient,
|
|
from_location=order_data.from_location,
|
|
to_location=order_data.to_location,
|
|
packages=order_data.packages,
|
|
services=order_data.services,
|
|
comment=order_data.comment,
|
|
)
|
|
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,
|
|
) -> 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
|
|
await session.flush()
|
|
return order
|