src/data_classes.py

from ast import Dict, List
from dataclasses import dataclass
import datetime

future_exit_time = datetime.datetime(9999, 12, 31, 23, 59, 59)
@dataclass
class Event:
    id: int
    trailer_id: int
    facility_id: int
    entrance_time: datetime
    exit_time: datetime
    @classmethod
    def from_dict(cls, data: Dict):
        return cls(
            id=data["id"],
            trailer_id=data["trailer_id"],
            facility_id =data["facility_id"],
            entrance_time = datetime.fromisoformat(data["entrance_time"]),
            exit_time = (datetime.fromisoformat(data["exit_time"]) if data["exit_time"] != None else future_exit_time),
        )


from dataclasses import dataclass, field
from typing import List
from datetime import datetime

@dataclass
class TimeWindow:
    start_time: datetime
    exit_time: datetime
    events: List[Event] = field(default_factory=list)

    @classmethod
    def from_event(cls, event: Event):
        return cls(
            start_time=event.entrance_time,
            exit_time=event.exit_time,
            events=[event]
        )

    def get_events_count(self):
        return len(self.events)
    
    def has_overlap_with_event(self, event: Event):
        # when the event overlaps with the window from the left
        if event.exit_time >=self.start_time and event.entrance_time <=self.start_time:
            return True
        # when the event overlaps with the window from the right
        elif event.entrance_time <=self.exit_time and event.exit_time >=self.exit_time:
            return True
        # when the event is completely inside the window
        elif event.entrance_time >=self.start_time and event.exit_time <=self.exit_time:
            return True
        else:
            return False
        
    def is_exact_match(self, event: Event):
        return event.entrance_time == self.start_time and event.exit_time == self.exit_time