counterweight
c4594a3f73
- Add create(), update(), and reload_with_relationships() methods to InviteRepository - Update InviteService to use repository methods instead of direct db operations
128 lines
4.2 KiB
Python
128 lines
4.2 KiB
Python
"""Invite repository for database queries."""
|
|
|
|
from sqlalchemy import desc, func, select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy.orm import joinedload
|
|
|
|
from models import Invite, InviteStatus
|
|
|
|
|
|
class InviteRepository:
|
|
"""Repository for invite-related database queries."""
|
|
|
|
def __init__(self, db: AsyncSession):
|
|
self.db = db
|
|
|
|
async def get_by_identifier(self, identifier: str) -> Invite | None:
|
|
"""Get an invite by identifier, eagerly loading relationships."""
|
|
result = await self.db.execute(
|
|
select(Invite)
|
|
.options(joinedload(Invite.godfather), joinedload(Invite.used_by))
|
|
.where(Invite.identifier == identifier)
|
|
)
|
|
return result.scalar_one_or_none()
|
|
|
|
async def get_by_id(self, invite_id: int) -> Invite | None:
|
|
"""Get an invite by ID, eagerly loading relationships."""
|
|
result = await self.db.execute(
|
|
select(Invite)
|
|
.options(joinedload(Invite.godfather), joinedload(Invite.used_by))
|
|
.where(Invite.id == invite_id)
|
|
)
|
|
return result.scalar_one_or_none()
|
|
|
|
async def get_by_godfather_id(
|
|
self, godfather_id: int, order_by_desc: bool = True
|
|
) -> list[Invite]:
|
|
"""Get all invites for a godfather user, eagerly loading relationships."""
|
|
query = (
|
|
select(Invite)
|
|
.options(joinedload(Invite.used_by))
|
|
.where(Invite.godfather_id == godfather_id)
|
|
)
|
|
if order_by_desc:
|
|
query = query.order_by(desc(Invite.created_at))
|
|
else:
|
|
query = query.order_by(Invite.created_at)
|
|
result = await self.db.execute(query)
|
|
return list(result.scalars().all())
|
|
|
|
async def count(
|
|
self,
|
|
status: InviteStatus | None = None,
|
|
godfather_id: int | None = None,
|
|
) -> int:
|
|
"""Count invites matching filters."""
|
|
query = select(func.count(Invite.id))
|
|
if status:
|
|
query = query.where(Invite.status == status)
|
|
if godfather_id:
|
|
query = query.where(Invite.godfather_id == godfather_id)
|
|
result = await self.db.execute(query)
|
|
return result.scalar() or 0
|
|
|
|
async def list_paginated(
|
|
self,
|
|
page: int,
|
|
per_page: int,
|
|
status: InviteStatus | None = None,
|
|
godfather_id: int | None = None,
|
|
) -> list[Invite]:
|
|
"""Get paginated list of invites, eagerly loading relationships."""
|
|
offset = (page - 1) * per_page
|
|
query = select(Invite).options(
|
|
joinedload(Invite.godfather), joinedload(Invite.used_by)
|
|
)
|
|
if status:
|
|
query = query.where(Invite.status == status)
|
|
if godfather_id:
|
|
query = query.where(Invite.godfather_id == godfather_id)
|
|
query = query.order_by(desc(Invite.created_at)).offset(offset).limit(per_page)
|
|
result = await self.db.execute(query)
|
|
return list(result.scalars().all())
|
|
|
|
async def create(self, invite: Invite) -> Invite:
|
|
"""
|
|
Create a new invite record.
|
|
|
|
Args:
|
|
invite: Invite instance to persist
|
|
|
|
Returns:
|
|
Created Invite record (committed and refreshed)
|
|
"""
|
|
self.db.add(invite)
|
|
await self.db.commit()
|
|
await self.db.refresh(invite)
|
|
return invite
|
|
|
|
async def update(self, invite: Invite) -> Invite:
|
|
"""
|
|
Update an existing invite record.
|
|
|
|
Args:
|
|
invite: Invite instance to update
|
|
|
|
Returns:
|
|
Updated Invite record (committed and refreshed)
|
|
"""
|
|
await self.db.commit()
|
|
await self.db.refresh(invite)
|
|
return invite
|
|
|
|
async def reload_with_relationships(self, invite_id: int) -> Invite:
|
|
"""
|
|
Reload an invite with all relationships eagerly loaded.
|
|
|
|
Args:
|
|
invite_id: ID of the invite to reload
|
|
|
|
Returns:
|
|
Invite record with relationships loaded
|
|
"""
|
|
result = await self.db.execute(
|
|
select(Invite)
|
|
.options(joinedload(Invite.godfather), joinedload(Invite.used_by))
|
|
.where(Invite.id == invite_id)
|
|
)
|
|
return result.scalar_one()
|