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>
90 lines
3.3 KiB
Python
90 lines
3.3 KiB
Python
"""Dispatch engine aggregates — Phase 10.5."""
|
|
from __future__ import annotations
|
|
|
|
import uuid
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import DateTime, Index, String, Text, UniqueConstraint
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
from sqlalchemy.types import JSON
|
|
|
|
from app.core.database import Base
|
|
from app.models.base import (
|
|
ActorAuditMixin,
|
|
OptimisticLockMixin,
|
|
SoftDeleteMixin,
|
|
TenantMixin,
|
|
TimestampMixin,
|
|
UUIDPrimaryKeyMixin,
|
|
)
|
|
from app.models.types import DispatchJobStatus, GUID, JobAssignmentStatus
|
|
|
|
|
|
class DispatchJob(
|
|
Base,
|
|
UUIDPrimaryKeyMixin,
|
|
TenantMixin,
|
|
TimestampMixin,
|
|
SoftDeleteMixin,
|
|
ActorAuditMixin,
|
|
OptimisticLockMixin,
|
|
):
|
|
"""Vertical job refs only — no deep order ownership."""
|
|
|
|
__tablename__ = "dispatch_jobs"
|
|
|
|
organization_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
|
|
hub_id: Mapped[uuid.UUID | None] = mapped_column(GUID(), nullable=True)
|
|
code: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
status: Mapped[DispatchJobStatus] = mapped_column(
|
|
default=DispatchJobStatus.PENDING, nullable=False
|
|
)
|
|
external_order_ref: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
external_source: Mapped[str | None] = mapped_column(String(100), nullable=True)
|
|
priority: Mapped[int] = mapped_column(default=0, nullable=False)
|
|
pickup_address_ref: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
dropoff_address_ref: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
scheduled_at: Mapped[datetime | None] = mapped_column(
|
|
DateTime(timezone=True), nullable=True
|
|
)
|
|
status_reason: Mapped[str | None] = mapped_column(String(500), nullable=True)
|
|
metadata_json: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
|
|
|
__table_args__ = (
|
|
UniqueConstraint("tenant_id", "code", name="uq_dispatch_jobs_tenant_code"),
|
|
Index("ix_dispatch_jobs_tenant_status", "tenant_id", "status"),
|
|
Index("ix_dispatch_jobs_external_ref", "tenant_id", "external_order_ref"),
|
|
Index("ix_dispatch_jobs_tenant_deleted", "tenant_id", "is_deleted"),
|
|
)
|
|
|
|
|
|
class JobAssignment(
|
|
Base,
|
|
UUIDPrimaryKeyMixin,
|
|
TenantMixin,
|
|
TimestampMixin,
|
|
SoftDeleteMixin,
|
|
ActorAuditMixin,
|
|
OptimisticLockMixin,
|
|
):
|
|
__tablename__ = "job_assignments"
|
|
|
|
dispatch_job_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
|
|
driver_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
|
|
vehicle_id: Mapped[uuid.UUID | None] = mapped_column(GUID(), nullable=True)
|
|
status: Mapped[JobAssignmentStatus] = mapped_column(
|
|
default=JobAssignmentStatus.PENDING, nullable=False
|
|
)
|
|
assigned_at: Mapped[datetime | None] = mapped_column(
|
|
DateTime(timezone=True), nullable=True
|
|
)
|
|
notes: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
metadata_json: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
|
|
|
__table_args__ = (
|
|
Index("ix_job_assignments_job", "tenant_id", "dispatch_job_id"),
|
|
Index("ix_job_assignments_driver", "tenant_id", "driver_id"),
|
|
Index("ix_job_assignments_tenant_status", "tenant_id", "status"),
|
|
Index("ix_job_assignments_tenant_deleted", "tenant_id", "is_deleted"),
|
|
)
|