76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from db.models.orders import Order, OrderAddon, OrderStatus
|
|
|
|
|
|
class OrderRepository:
|
|
def __init__(self, session: AsyncSession) -> None:
|
|
self.session = session
|
|
|
|
async def create(
|
|
self,
|
|
*,
|
|
user_id: int,
|
|
devices: int,
|
|
duration_days: int,
|
|
total_amount: float,
|
|
balance_amount: float,
|
|
addons: list[str] | None = None,
|
|
status: OrderStatus = OrderStatus.PENDING,
|
|
) -> Order:
|
|
if not addons:
|
|
addons = []
|
|
|
|
order = Order(
|
|
user_id=user_id,
|
|
devices=devices,
|
|
duration_days=duration_days,
|
|
total_amount=total_amount,
|
|
balance_amount=balance_amount,
|
|
status=status,
|
|
)
|
|
self.session.add(order)
|
|
await self.session.flush()
|
|
|
|
for addon_id in addons:
|
|
addon = OrderAddon(order_id=order.id, addon_id=addon_id)
|
|
self.session.add(addon)
|
|
|
|
await self.session.commit()
|
|
await self.session.refresh(order, attribute_names=["addons"])
|
|
|
|
return order
|
|
|
|
async def get_by_id(self, order_id: int) -> Order | None:
|
|
stmt = select(Order).where(Order.id == order_id)
|
|
r = await self.session.execute(stmt)
|
|
|
|
return r.scalar_one_or_none()
|
|
|
|
async def get_order_by_user_id(self, user_id: int) -> list[Order]:
|
|
stmt = select(Order).where(Order.user_id == user_id)
|
|
r = await self.session.execute(stmt)
|
|
|
|
return list(r.scalars().all())
|
|
|
|
async def get_active_by_user_id(self, user_id: int) -> list[Order]:
|
|
stmt = (
|
|
select(Order).where(Order.user_id == user_id).where(Order.status == OrderStatus.PENDING)
|
|
)
|
|
r = await self.session.execute(stmt)
|
|
|
|
return list(r.scalars().all())
|
|
|
|
async def get_paid_unapplied_by_user_id(self, user_id: int) -> list[Order]:
|
|
stmt = (
|
|
select(Order)
|
|
.where(Order.user_id == user_id)
|
|
.where(Order.status == OrderStatus.PAID)
|
|
.where(Order.applied_at.is_(None))
|
|
.order_by(Order.applies_at.asc(), Order.id.asc())
|
|
)
|
|
r = await self.session.execute(stmt)
|
|
|
|
return list(r.scalars().all())
|