Fix all lookups handling for related fields (#128)

Author: Ryan P Kilby

rest_framework_filters/filterset.py

@@ -9,6 +9,7 @@
 from django.utils import six
 
 from django_filters import filterset, rest_framework
+from django_filters.utils import get_model_field
 
 from . import filters
 from . import utils
@@ -139,7 +140,7 @@ def filters_for_model(cls, model, opts):
                 lookups = '__all__'
 
             if lookups == '__all__':
-                field = model._meta.get_field(name)
+                field = get_model_field(model, name)
                 fields[name] = utils.lookups_for_field(field)
 
         return filterset.filters_for_model(

tests/test_filterset.py

@@ -63,6 +63,17 @@ class Meta:
         self.assertIsInstance(F.base_filters['author'], filters.ModelChoiceFilter)
         self.assertIsInstance(F.base_filters['author__in'], BaseInFilter)
 
+    def test_alllookupsfilter_for_related_field(self):
+        # See: https://github.com/philipn/django-rest-framework-filters/issues/127
+        class F(FilterSet):
+            author = filters.AllLookupsFilter(name='author__last_name')
+
+            class Meta:
+                model = Note
+
+        self.assertIsInstance(F.base_filters['author'], filters.CharFilter)
+        self.assertEqual(F.base_filters['author'].name, 'author__last_name')
+
     def test_relatedfilter_combined_with__all__(self):
         # ensure that related filter is compatible with __all__ lookups.
         class F(FilterSet):