from django.db import models from django.utils.translation import gettext_lazy as _ class TransportType(models.TextChoices): TGV = "TGV", _("TGV") TER = "TER", _("TER") INTERCITES = "IC", _("Intercités") TRANSILIEN = "TN", _("Transilien") class LocationType(models.IntegerChoices): STOP_PLATFORM = 0, _("Stop/platform") STATION = 1, _("Station") ENTRANCE_EXIT = 2, _("Entrance/exit") GENERIC_NODE = 3, _("Generic node") BOARDING_AREA = 4, _("Boarding area") class AccessInformation(models.IntegerChoices): NO_INFORMATION = 0, _("No information") POSSIBLE = 1, _("Possible") NOT_POSSIBLE = 2, _("Not possible") class PickupType(models.IntegerChoices): REGULAR = 0, _("Regular") NONE = 1, _("None") MUST_PHONE_AGENCY = 2, _("Must phone agency") MUST_COORDINATE_WITH_DRIVER = 3, _("Must coordinate with driver") class RouteType(models.IntegerChoices): TRAM = 0, _("Tram") METRO = 1, _("Metro") RAIL = 2, _("Rail") BUS = 3, _("Bus") FERRY = 4, _("Ferry") CABLE_CAR = 5, _("Cable car") GONDOLA = 6, _("Gondola") FUNICULAR = 7, _("Funicular") class Direction(models.IntegerChoices): OUTBOUND = 0, _("Outbound") INBOUND = 1, _("Inbound") class TransferType(models.IntegerChoices): RECOMMENDED = 0, _("Recommended") TIMED = 1, _("Timed") MINIMUM_TIME = 2, _("Minimum time") NOT_POSSIBLE = 3, _("Not possible") class ExceptionType(models.IntegerChoices): ADDED = 1, _("Added") REMOVED = 2, _("Removed") class Agency(models.Model): id = models.CharField( max_length=255, primary_key=True, verbose_name=_("Agency ID"), ) name = models.CharField( max_length=255, unique=True, verbose_name=_("Agency name"), ) url = models.URLField( verbose_name=_("Agency URL"), ) timezone = models.CharField( max_length=255, verbose_name=_("Agency timezone"), ) lang = models.CharField( max_length=255, verbose_name=_("Agency language"), blank=True, ) phone = models.CharField( max_length=255, verbose_name=_("Agency phone"), blank=True, ) email = models.EmailField( verbose_name=_("Agency email"), blank=True, ) def __str__(self): return self.name class Meta: verbose_name = _("Agency") verbose_name_plural = _("Agencies") ordering = ("name",) class Stop(models.Model): id = models.CharField( max_length=255, primary_key=True, verbose_name=_("Stop ID"), ) code = models.CharField( max_length=255, verbose_name=_("Stop code"), blank=True, ) name = models.CharField( max_length=255, verbose_name=_("Stop name"), ) desc = models.CharField( max_length=255, verbose_name=_("Stop description"), blank=True, ) lon = models.FloatField( verbose_name=_("Stop longitude"), ) lat = models.FloatField( verbose_name=_("Stop latitude"), ) zone_id = models.CharField( max_length=255, verbose_name=_("Zone ID"), ) url = models.URLField( verbose_name=_("Stop URL"), blank=True, ) location_type = models.IntegerField( verbose_name=_("Location type"), blank=True, choices=LocationType, default=LocationType.STOP_PLATFORM, ) parent_station = models.ForeignKey( to="Stop", on_delete=models.PROTECT, verbose_name=_("Parent station"), related_name="children", blank=True, null=True, ) timezone = models.CharField( max_length=255, verbose_name=_("Stop timezone"), blank=True, ) level_id = models.CharField( max_length=255, verbose_name=_("Level ID"), blank=True, ) wheelchair_boarding = models.IntegerField( verbose_name=_("Wheelchair boarding"), blank=True, choices=AccessInformation, default=AccessInformation.NO_INFORMATION, ) platform_code = models.CharField( max_length=255, verbose_name=_("Platform code"), blank=True, ) def __str__(self): return f"{self.name} ({self.id})" class Meta: verbose_name = _("Stop") verbose_name_plural = _("Stops") ordering = ("id",) class Route(models.Model): id = models.CharField( max_length=255, primary_key=True, verbose_name=_("ID"), ) agency = models.ForeignKey( to="Agency", on_delete=models.CASCADE, verbose_name=_("Agency ID"), related_name="routes", ) short_name = models.CharField( max_length=255, verbose_name=_("Route short name"), ) long_name = models.CharField( max_length=255, verbose_name=_("Route long name"), ) desc = models.CharField( max_length=255, verbose_name=_("Route description"), blank=True, ) type = models.IntegerField( verbose_name=_("Route type"), choices=RouteType, ) url = models.URLField( verbose_name=_("Route URL"), blank=True, ) color = models.CharField( max_length=255, verbose_name=_("Route color"), blank=True, ) text_color = models.CharField( max_length=255, verbose_name=_("Route text color"), blank=True, ) def __str__(self): return f"{self.long_name}" class Meta: verbose_name = _("Route") verbose_name_plural = _("Routes") ordering = ("id",) class Trip(models.Model): id = models.CharField( max_length=255, primary_key=True, verbose_name=_("Trip ID"), ) route = models.ForeignKey( to="Route", on_delete=models.CASCADE, verbose_name=_("Route"), related_name="trips", ) service = models.ForeignKey( to="Calendar", on_delete=models.CASCADE, verbose_name=_("Service"), related_name="trips", ) headsign = models.CharField( max_length=255, verbose_name=_("Trip headsign"), blank=True, ) short_name = models.CharField( max_length=255, verbose_name=_("Trip short name"), blank=True, ) direction_id = models.IntegerField( verbose_name=_("Direction"), choices=Direction, null=True, ) block_id = models.CharField( max_length=255, verbose_name=_("Block ID"), blank=True, ) shape_id = models.CharField( max_length=255, verbose_name=_("Shape ID"), blank=True, ) wheelchair_accessible = models.IntegerField( verbose_name=_("Wheelchair accessible"), choices=AccessInformation, default=AccessInformation.NO_INFORMATION, null=True, ) bikes_allowed = models.IntegerField( verbose_name=_("Bikes allowed"), choices=AccessInformation, default=AccessInformation.NO_INFORMATION, null=True, ) @property def origin(self): return self.stop_times.order_by('stop_sequence').first().stop @property def destination(self): return self.stop_times.order_by('-stop_sequence').first().stop @property def train_type(self): if self.service.transport_type == TransportType.TRANSILIEN: return self.route.short_name else: train_type = self.origin.id.split('StopPoint:OCE')[1].split('-')[0] if train_type == "Train TER": train_type = "TER" elif train_type == "INTERCITES": train_type = "INTER-CITÉS" elif train_type == "INTERCITES de nuit": train_type = "INTER-CITÉS de nuit" return train_type @property def train_number(self): if self.service.transport_type == TransportType.TRANSILIEN: return self.short_name else: return self.headsign @property def color(self): if self.route.color: return self.route.color elif self.train_type == "OUIGO": return "E60075" return "FFFFFF" @property def text_color(self): if self.route.text_color: return self.route.text_color elif self.train_type == "OUIGO": return "FFFFFF" elif self.train_type == "TGV INOUI": return "9B2743" elif self.train_type == "INTER-CITÉS" or self.train_type == "INTER-CITÉS de nuit": return "404042" return "000000" def __str__(self): return f"{self.route.long_name} - {self.id}" class Meta: verbose_name = _("Trip") verbose_name_plural = _("Trips") ordering = ("id",) class StopTime(models.Model): id = models.CharField( max_length=255, primary_key=True, verbose_name=_("ID"), ) trip = models.ForeignKey( to="Trip", on_delete=models.CASCADE, verbose_name=_("Trip"), related_name="stop_times", ) arrival_time = models.DurationField( verbose_name=_("Arrival time"), ) departure_time = models.DurationField( verbose_name=_("Departure time"), ) stop = models.ForeignKey( to="Stop", on_delete=models.CASCADE, verbose_name=_("Stop ID"), related_name="stop_times", ) stop_sequence = models.IntegerField( verbose_name=_("Stop sequence"), ) stop_headsign = models.CharField( max_length=255, verbose_name=_("Stop headsign"), blank=True, ) pickup_type = models.IntegerField( verbose_name=_("Pickup type"), choices=PickupType, default=PickupType.REGULAR, null=True, ) drop_off_type = models.IntegerField( verbose_name=_("Drop off type"), choices=PickupType, default=PickupType.REGULAR, null=True, ) timepoint = models.BooleanField( verbose_name=_("Timepoint"), default=True, null=True, ) @property def pretty_arrival_time(self): seconds = self.arrival_time.total_seconds() hours = int(seconds // 3600) % 24 minutes = int((seconds % 3600) // 60) return f"{hours:02}:{minutes:02}" @property def pretty_departure_time(self): seconds = self.departure_time.total_seconds() hours = int(seconds // 3600) % 24 minutes = int((seconds % 3600) // 60) return f"{hours:02}:{minutes:02}" def __str__(self): return f"{self.trip.route.long_name} - {self.trip_id} - {self.stop.name}" class Meta: verbose_name = _("Stop time") verbose_name_plural = _("Stop times") class Calendar(models.Model): id = models.CharField( max_length=255, primary_key=True, verbose_name=_("Service ID"), ) monday = models.BooleanField( verbose_name=_("Monday"), ) tuesday = models.BooleanField( verbose_name=_("Tuesday"), ) wednesday = models.BooleanField( verbose_name=_("Wednesday"), ) thursday = models.BooleanField( verbose_name=_("Thursday"), ) friday = models.BooleanField( verbose_name=_("Friday"), ) saturday = models.BooleanField( verbose_name=_("Saturday"), ) sunday = models.BooleanField( verbose_name=_("Sunday"), ) start_date = models.DateField( verbose_name=_("Start date"), ) end_date = models.DateField( verbose_name=_("End date"), ) transport_type = models.CharField( max_length=255, verbose_name=_("Transport type"), choices=TransportType, ) def __str__(self): return self.id class Meta: verbose_name = _("Calendar") verbose_name_plural = _("Calendars") ordering = ("id",) class CalendarDate(models.Model): id = models.CharField( max_length=255, primary_key=True, verbose_name=_("ID"), ) service = models.ForeignKey( to="Calendar", on_delete=models.CASCADE, verbose_name=_("Service"), related_name="dates", ) date = models.DateField( verbose_name=_("Date"), ) exception_type = models.IntegerField( verbose_name=_("Exception type"), choices=ExceptionType, ) transport_type = models.CharField( max_length=255, verbose_name=_("Transport type"), choices=TransportType, ) def __str__(self): return f"{self.service.id} - {self.date} - {self.exception_type}" class Meta: verbose_name = _("Calendar date") verbose_name_plural = _("Calendar dates") ordering = ("id",) class Transfer(models.Model): id = models.CharField( max_length=255, primary_key=True, verbose_name=_("ID"), ) from_stop = models.ForeignKey( to="Stop", on_delete=models.CASCADE, verbose_name=_("From stop"), related_name="transfers_from", ) to_stop = models.ForeignKey( to="Stop", on_delete=models.CASCADE, verbose_name=_("To stop"), related_name="transfers_to", ) transfer_type = models.IntegerField( verbose_name=_("Transfer type"), choices=TransferType, default=TransferType.RECOMMENDED, ) min_transfer_time = models.IntegerField( verbose_name=_("Minimum transfer time"), blank=True, ) class Meta: verbose_name = _("Transfer") verbose_name_plural = _("Transfers") ordering = ("id",) class FeedInfo(models.Model): publisher_name = models.CharField( max_length=255, verbose_name=_("Feed publisher name"), ) publisher_url = models.URLField( verbose_name=_("Feed publisher URL"), ) lang = models.CharField( max_length=255, verbose_name=_("Feed language"), ) start_date = models.DateField( verbose_name=_("Feed start date"), ) end_date = models.DateField( verbose_name=_("Feed end date"), ) version = models.CharField( max_length=255, verbose_name=_("Feed version"), ) class Meta: verbose_name = _("Feed info") verbose_name_plural = _("Feed infos") ordering = ("publisher_name",)