from rest_framework.routers import APIRootView

from extras.api.views import CustomFieldModelViewSet
from wireless import filtersets
from wireless.models import *
from . import serializers


class WirelessRootView(APIRootView):
    """
    Wireless API root view
    """
    def get_view_name(self):
        return 'Wireless'


#
# Providers
#

class SSIDViewSet(CustomFieldModelViewSet):
    queryset = SSID.objects.prefetch_related('tags')
    serializer_class = serializers.SSIDSerializer
    filterset_class = filtersets.SSIDFilterSet