Extend functionalities

This commit is contained in:
2022-03-29 21:15:53 +02:00
parent 9678ba3789
commit 91e6dd0cac
17 changed files with 237 additions and 24 deletions

View File

@@ -1,27 +1,27 @@
from rest_framework.generics import ListAPIView, RetrieveAPIView
from roster.models import Cab
from roster.serializers import CabSerializer
from roster.models import RollingStock
from roster.serializers import RollingStockSerializer
class RosterList(ListAPIView):
queryset = Cab.objects.all()
serializer_class = CabSerializer
queryset = RollingStock.objects.all()
serializer_class = RollingStockSerializer
class RosterGet(RetrieveAPIView):
queryset = Cab.objects.all()
serializer_class = CabSerializer
queryset = RollingStock.objects.all()
serializer_class = RollingStockSerializer
lookup_field = 'uuid'
class RosterAddress(RetrieveAPIView):
queryset = Cab.objects.all()
serializer_class = CabSerializer
queryset = RollingStock.objects.all()
serializer_class = RollingStockSerializer
lookup_field = 'address'
class RosterIdentifier(RetrieveAPIView):
queryset = Cab.objects.all()
serializer_class = CabSerializer
queryset = RollingStock.objects.all()
serializer_class = RollingStockSerializer
lookup_field = 'identifier'