code.grnet.gr Git - flowspy/blob - flowspec/viewsets.py

   1 from rest_framework import viewsets
   2 from flowspec.serializers import (
   3     MatchProtocolSerializer,
   4     FragmentTypeSerializer,
   5     ThenActionSerializer,
   6     RouteSerializer,
   7     PortSerializer
   8 )
   9
  10
  11 from flowspec.models import (
  12     MatchProtocol,
  13     FragmentType,
  14     ThenAction,
  15     Route,
  16     MatchPort
  17 )
  18
  19
  20 class MatchProtocolViewSet(viewsets.ModelViewSet):
  21     queryset = MatchProtocol.objects.all()
  22     serializer_class = MatchProtocolSerializer
  23
  24
  25 class FragmentTypeViewSet(viewsets.ModelViewSet):
  26     queryset = FragmentType.objects.all()
  27     serializer_class = FragmentTypeSerializer
  28
  29
  30 class ThenActionViewSet(viewsets.ModelViewSet):
  31     queryset = ThenAction.objects.all()
  32     serializer_class = ThenActionSerializer
  33
  34
  35 class RouteViewSet(viewsets.ModelViewSet):
  36     queryset = Route.objects.all()
  37     serializer_class = RouteSerializer
  38
  39
  40 class PortViewSet(viewsets.ModelViewSet):
  41     queryset = MatchPort.objects.all()
  42     serializer_class = PortSerializer