Display delays

sncf/api/serializers.py

@@ -1,7 +1,7 @@
 from rest_framework import serializers
 
 from sncfgtfs.models import Agency, Stop, Route, Trip, StopTime, Calendar, CalendarDate, \
-    Transfer, FeedInfo
+    Transfer, FeedInfo, TripUpdate, StopTimeUpdate
 
 
 class AgencySerializer(serializers.ModelSerializer):
@@ -61,3 +61,15 @@ class FeedInfoSerializer(serializers.ModelSerializer):
     class Meta:
         model = FeedInfo
         fields = '__all__'
+
+
+class TripUpdateSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = TripUpdate
+        fields = '__all__'
+
+
+class StopTimeUpdateSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = StopTimeUpdate
+        fields = '__all__'

sncf/api/views.py

@@ -10,10 +10,9 @@ from rest_framework.filters import OrderingFilter, SearchFilter
 
 from sncf.api.serializers import AgencySerializer, StopSerializer, RouteSerializer, TripSerializer, \
     StopTimeSerializer, CalendarSerializer, CalendarDateSerializer, TransferSerializer, \
-    FeedInfoSerializer
+    FeedInfoSerializer, TripUpdateSerializer, StopTimeUpdateSerializer
 from sncfgtfs.models import Agency, Stop, Route, Trip, StopTime, Calendar, CalendarDate, \
-    Transfer, FeedInfo
-
+    Transfer, FeedInfo, TripUpdate, StopTimeUpdate
 
 CACHE_CONTROL = cache_control(max_age=7200)
 LAST_MODIFIED = last_modified(lambda *args, **kwargs: datetime.fromisoformat(FeedInfo.objects.get().version))
@@ -102,6 +101,20 @@ class FeedInfoViewSet(viewsets.ReadOnlyModelViewSet):
     filterset_fields = '__all__'
 
 
+class TripUpdateViewSet(viewsets.ReadOnlyModelViewSet):
+    queryset = TripUpdate.objects.all()
+    serializer_class = TripUpdateSerializer
+    filter_backends = [DjangoFilterBackend]
+    filterset_fields = '__all__'
+
+
+class StopTimeUpdateViewSet(viewsets.ReadOnlyModelViewSet):
+    queryset = StopTimeUpdate.objects.all()
+    serializer_class = StopTimeUpdateSerializer
+    filter_backends = [DjangoFilterBackend]
+    filterset_fields = '__all__'
+
+
 class NextDeparturesViewSet(viewsets.ReadOnlyModelViewSet):
     queryset = StopTime.objects.none()
     serializer_class = StopTimeSerializer