from rest_framework import viewsets

from apiv2.serializers import CrawlRunSerializer, ASLastSeenSerializer
from backend import crawler
from bgpdata.models import CrawlRun, ASLastSeen


class CrawlRunViewSet(viewsets.ReadOnlyModelViewSet):
    """Represents a CrawlRun.

    Graph is shown if with_graph is passed as query arg.
    With /live/ the current network is shown (internally, a crawl is triggered for each request)"""
    queryset = CrawlRun.objects.all()
    serializer_class = CrawlRunSerializer

    def get_object(self):
        lookup_url_kwarg = self.lookup_url_kwarg or self.lookup_field
        if self.kwargs.get(lookup_url_kwarg) == "live":
            net = crawler.get_current_network()
            obj = crawler.make_crawl_from_net(net)
            self.check_object_permissions(self.request, obj)
        else:
            obj = super().get_object()

        return obj


class ASLastSeenViewSet(viewsets.ReadOnlyModelViewSet):
    queryset = ASLastSeen.objects.all()
    serializer_class = ASLastSeenSerializer