Address review comments

Signed-off-by: Tushar Goel <[email protected]>

Author: TG1999

vulnerabilities/api.py

@@ -22,6 +22,7 @@
 from vulnerabilities.models import Vulnerability
 from vulnerabilities.models import VulnerabilityReference
 from vulnerabilities.models import VulnerabilitySeverity
+from vulnerabilities.models import get_purl_query_lookups
 
 
 class VulnerabilitySeveritySerializer(serializers.ModelSerializer):
@@ -214,15 +215,6 @@ def filter_purl(self, queryset, name, value):
         return self.queryset.filter(**lookups)
 
 
-def get_purl_query_lookups(purl):
-    lookup_fields = ["type", "namespace", "name", "version"]
-    return {
-        field_name: value
-        for field_name, value in purl.to_dict().items()
-        if value and field_name in lookup_fields
-    }
-
-
 class PackageViewSet(viewsets.ReadOnlyModelViewSet):
     queryset = Package.objects.all()
     serializer_class = PackageSerializer

vulnerabilities/models.py

@@ -218,6 +218,15 @@ def with_vulnerability_counts(self):
         )
 
 
+def get_purl_query_lookups(purl):
+    lookup_fields = ["type", "namespace", "name", "version"]
+    return {
+        field_name: value
+        for field_name, value in purl.to_dict().items()
+        if value and field_name in lookup_fields
+    }
+
+
 class Package(PackageURLMixin):
     """
     A software package with related vulnerabilities.