counterweight
280c1e5687
- Add get_available_slots() and _expand_availability_to_slots() to ExchangeService - Update routes/exchange.py to use ExchangeService.get_available_slots() - Remove all business logic from get_available_slots endpoint - Add AvailabilityRepository to ExchangeService dependencies - Add Availability and BookableSlot imports to ExchangeService - Fix import path for validate_date_in_range (use date_validation module) - Remove unused user_repo variable and import from routes/invites.py - Fix mypy error in ValidationError by adding proper type annotation
32 lines
1 KiB
Python
32 lines
1 KiB
Python
"""User repository for database queries."""
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from models import User
|
|
|
|
|
|
class UserRepository:
|
|
"""Repository for user-related database queries."""
|
|
|
|
def __init__(self, db: AsyncSession):
|
|
self.db = db
|
|
|
|
async def get_by_email(self, email: str) -> User | None:
|
|
"""Get a user by email."""
|
|
result = await self.db.execute(select(User).where(User.email == email))
|
|
return result.scalar_one_or_none()
|
|
|
|
async def get_by_id(self, user_id: int) -> User | None:
|
|
"""Get a user by ID."""
|
|
result = await self.db.execute(select(User).where(User.id == user_id))
|
|
return result.scalar_one_or_none()
|
|
|
|
async def get_godfather_email(self, godfather_id: int | None) -> str | None:
|
|
"""Get the email of a godfather user by ID."""
|
|
if not godfather_id:
|
|
return None
|
|
result = await self.db.execute(
|
|
select(User.email).where(User.id == godfather_id)
|
|
)
|
|
return result.scalar_one_or_none()
|