72077908f1
Ship fleet, availability, pricing, dispatch, routing, tracking, and settlement on delivery-service 0.10.8.0 with migrations and phase tests green. Co-authored-by: Cursor <cursoragent@cursor.com>
39 lines
1.5 KiB
Python
39 lines
1.5 KiB
Python
"""Availability query handlers — Phase 10.3."""
|
|
from __future__ import annotations
|
|
|
|
from uuid import UUID
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.services.availability import AvailabilityService, ShiftService, WorkingZoneService
|
|
from app.specifications.availability import ShiftListSpec, WorkingZoneListSpec
|
|
|
|
|
|
class AvailabilityQueries:
|
|
def __init__(self, session: AsyncSession) -> None:
|
|
self.availability = AvailabilityService(session)
|
|
self.shifts = ShiftService(session)
|
|
self.zones = WorkingZoneService(session)
|
|
|
|
async def get_availability(self, tenant_id: UUID, driver_id: UUID):
|
|
return await self.availability.get_for_driver(tenant_id, driver_id)
|
|
|
|
async def get_shift(self, tenant_id: UUID, shift_id: UUID):
|
|
return await self.shifts.get(tenant_id, shift_id)
|
|
|
|
async def list_shifts(
|
|
self, tenant_id: UUID, *, offset: int, limit: int, spec: ShiftListSpec | None
|
|
):
|
|
return await self.shifts.list(tenant_id, offset=offset, limit=limit, spec=spec)
|
|
|
|
async def list_shift_assignments(self, tenant_id: UUID, shift_id: UUID):
|
|
return await self.shifts.list_assignments(tenant_id, shift_id)
|
|
|
|
async def get_zone(self, tenant_id: UUID, zone_id: UUID):
|
|
return await self.zones.get(tenant_id, zone_id)
|
|
|
|
async def list_zones(
|
|
self, tenant_id: UUID, *, offset: int, limit: int, spec: WorkingZoneListSpec | None
|
|
):
|
|
return await self.zones.list(tenant_id, offset=offset, limit=limit, spec=spec)
|