Add support for using package_url as a field lookup

Signed-off-by: tdruez <tdruez@nexb.com>

Author: tdruez

scanpipe/models.py

@@ -2943,6 +2943,13 @@ def only_package_url_fields(self):
         """
         return self.only("uuid", *PACKAGE_URL_FIELDS)
 
+    def filter(self, *args, **kwargs):
+        """Add support for using ``package_url`` as a field lookup."""
+        if purl_str := kwargs.pop("package_url", None):
+            return super().filter(*args, **kwargs).for_package_url(purl_str)
+
+        return super().filter(*args, **kwargs)
+
 
 class AbstractPackage(models.Model):
     """These fields should be kept in line with `packagedcode.models.PackageData`."""

scanpipe/tests/test_models.py

@@ -1806,6 +1806,8 @@ def test_scanpipe_discovered_package_queryset_for_package_url(self):
         for purl, expected_count in inputs:
             qs = DiscoveredPackage.objects.for_package_url(purl)
             self.assertEqual(expected_count, qs.count(), msg=purl)
+            qs2 = DiscoveredPackage.objects.filter(package_url=purl)
+            self.assertEqual(expected_count, qs2.count(), msg=purl)
 
     def test_scanpipe_discovered_package_queryset_vulnerable(self):
         p1 = DiscoveredPackage.create_from_data(self.project1, package_data1)