Merge pull request #197 from kwongtn/pass-info-in-generic-filter

Pass info for generic filter type

Author: bellini666

strawberry_django/filters.py

@@ -146,7 +146,11 @@ def apply(filters, queryset: QuerySet, info=UNSET, pk=UNSET) -> QuerySet:
 
     filter_method = getattr(filters, "filter", None)
     if filter_method:
-        return filter_method(queryset)
+        if function_allow_passing_info(filter_method):
+            return filter_method(queryset=queryset, info=info)
+
+        else:
+            return filter_method(queryset=queryset)
 
     filter_kwargs, filter_methods = build_filter_kwargs(filters)
     queryset = queryset.filter(**filter_kwargs)

tests/filters/test_filters.py

@@ -201,6 +201,33 @@ def fruits(filters: FruitFilterWithInfo, info: Info) -> List[Fruit]:
     ]
 
 
+def test_resolver_filter_override_with_info(fruits):
+    from strawberry.types.info import Info
+
+    @strawberry_django.filters.filter(models.Fruit, lookups=True)
+    class FruitFilterWithInfo:
+        custom_field: bool
+
+        def filter(self, queryset, info: Info):
+            # Test here is to prove that info can be passed properly
+            assert isinstance(info, Info)
+            return queryset.filter(name="banana")
+
+    @strawberry.type
+    class Query:
+        @strawberry.field
+        def fruits(filters: FruitFilterWithInfo, info: Info) -> List[Fruit]:
+            queryset = models.Fruit.objects.all()
+            return strawberry_django.filters.apply(filters, queryset, info=info)
+
+    query = utils.generate_query(Query)
+    result = query("{ fruits(filters: { customField: true }) { id name } }")
+    assert not result.errors
+    assert result.data["fruits"] == [
+        {"id": "3", "name": "banana"},
+    ]
+
+
 def test_resolver_nonfilter(fruits):
     @strawberry.type
     class Query: