File size: 2,231 Bytes
fcf8749
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
"""
StopIssue database model.
Represents an issue at a specific stop.
"""

import enum
import uuid
from datetime import datetime
from typing import TYPE_CHECKING

from sqlalchemy import Integer, Text, DateTime, ForeignKey, Enum
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.database import Base, GUID

if TYPE_CHECKING:
    from app.models.assignment import Assignment
    from app.models.route import Route
    from app.models.driver import Driver


class StopIssueType(str, enum.Enum):
    """Type of issue at a stop."""
    NAVIGATION = "NAVIGATION"
    SAFETY = "SAFETY"
    TIME_WINDOW = "TIME_WINDOW"
    CUSTOMER_UNAVAILABLE = "CUSTOMER_UNAVAILABLE"
    OTHER = "OTHER"


class StopIssue(Base):
    """
    StopIssue model representing an issue at a specific stop.
    Used to track and report problems during deliveries.
    """
    __tablename__ = "stop_issues"

    id: Mapped[uuid.UUID] = mapped_column(
        GUID(),
        primary_key=True,
        default=uuid.uuid4,
    )
    assignment_id: Mapped[uuid.UUID] = mapped_column(
        GUID(),
        ForeignKey("assignments.id", ondelete="CASCADE"),
        nullable=False,
        index=True,
    )
    route_id: Mapped[uuid.UUID] = mapped_column(
        GUID(),
        ForeignKey("routes.id", ondelete="CASCADE"),
        nullable=False,
        index=True,
    )
    driver_id: Mapped[uuid.UUID] = mapped_column(
        GUID(),
        ForeignKey("drivers.id", ondelete="CASCADE"),
        nullable=False,
        index=True,
    )
    stop_order: Mapped[int] = mapped_column(Integer, nullable=False)
    issue_type: Mapped[StopIssueType] = mapped_column(
        Enum(StopIssueType),
        nullable=False,
    )
    notes: Mapped[str] = mapped_column(Text, nullable=False)
    created_at: Mapped[datetime] = mapped_column(
        DateTime,
        default=datetime.utcnow,
        nullable=False,
    )

    # Relationships
    assignment: Mapped["Assignment"] = relationship("Assignment")
    route: Mapped["Route"] = relationship("Route")
    driver: Mapped["Driver"] = relationship("Driver")

    def __repr__(self) -> str:
        return f"<StopIssue(id={self.id}, stop_order={self.stop_order}, type={self.issue_type})>"