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

startScan/migrations/0007_auto_20201010_1546.py

@@ -0,0 +1,17 @@
+# Generated by Django 3.0.7 on 2020-10-10 15:46
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('startScan', '0006_vulnerabilityscanner'),
+    ]
+
+    operations = [
+        migrations.RenameModel(
+            old_name='VulnerabilityScanner',
+            new_name='VulnerabilityScan',
+        ),
+    ]

startScan/models.py

@@ -59,7 +59,7 @@ class ScanActivity(models.Model):
         return str(self.title)
 
 
-class VulnerabilityScanner(models.Model):
+class VulnerabilityScan(models.Model):
     vulnerability_of = models.ForeignKey(ScanHistory, on_delete=models.CASCADE)
     url = models.CharField(max_length=1000)
     title = models.CharField(max_length=400)