vulnerability serializer addeds

startScan/api/serializers.py

@@ -1,6 +1,6 @@
 from rest_framework import serializers
 
-from startScan.models import ScannedHost, ScanHistory, WayBackEndPoint
+from startScan.models import ScannedHost, ScanHistory, WayBackEndPoint, VulnerabilityScan
 
 
 class ScanHistorySerializer(serializers.ModelSerializer):
@@ -16,3 +16,10 @@ class EndpointSerializer(serializers.ModelSerializer):
     class Meta:
         model = WayBackEndPoint
         fields = '__all__'
+
+
+class VulnerabilitySerializer(serializers.ModelSerializer):
+
+    class Meta:
+        model = VulnerabilityScan
+        fields = '__all__'

startScan/api/urls.py

@@ -1,7 +1,7 @@
 from django.conf.urls import url, include
 from django.urls import path
 from rest_framework import routers
-from .views import ScanHistoryViewSet, EndPointViewSet
+from .views import ScanHistoryViewSet, EndPointViewSet, VulnerabilityViewSet
 
 app_name = 'startScan'
 router = routers.DefaultRouter()
@@ -10,6 +10,8 @@ router.register(r'scanHistory', ScanHistoryViewSet)
 
 router.register(r'listEndpoints', EndPointViewSet)
 
+router.register(r'vulnerability', VulnerabilityViewSet)
+
 urlpatterns = [
     url('^', include(router.urls)),
 

startScan/api/views.py

@@ -1,6 +1,6 @@
-from startScan.api.serializers import ScanHistorySerializer, EndpointSerializer
+from startScan.api.serializers import ScanHistorySerializer, EndpointSerializer, VulnerabilitySerializer
 from rest_framework import viewsets
-from startScan.models import ScannedHost, ScanHistory, WayBackEndPoint
+from startScan.models import ScannedHost, ScanHistory, WayBackEndPoint, VulnerabilityScan
 from rest_framework.response import Response
 from rest_framework import status
 from rest_framework.decorators import api_view, action
@@ -33,3 +33,18 @@ class EndPointViewSet(viewsets.ModelViewSet):
             return self.queryset
         else:
             return self.queryset
+
+
+class VulnerabilityViewSet(viewsets.ModelViewSet):
+    queryset = VulnerabilityScan.objects.all()
+    serializer_class = VulnerabilitySerializer
+
+    def get_queryset(self):
+        req = self.request
+        vulnerability_of = req.query_params.get('vulnerability_of')
+        if vulnerability_of:
+            self.queryset = WayBackEndPoint.objects.filter(
+                vulnerability_of__id=vulnerability_of)
+            return self.queryset
+        else:
+            return self.queryset