fix: allow passing a filterset class to the filterset factory

carltongibson · Jan 19, 2024 · b590408 · b590408
1 parent af9f13b
commit b590408
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/django_filters/filterset.py b/django_filters/filterset.py
@@ -458,9 +458,10 @@ class FilterSet(BaseFilterSet, metaclass=FilterSetMetaclass):
     pass
 
 
-def filterset_factory(model, fields=ALL_FIELDS):
-    meta = type(str("Meta"), (object,), {"model": model, "fields": fields})
-    filterset = type(
-        str("%sFilterSet" % model._meta.object_name), (FilterSet,), {"Meta": meta}
+def filterset_factory(model, filterset=FilterSet, fields=ALL_FIELDS):
+    attrs = {"model": model, "fields": fields}
+    bases = (filterset.Meta,) if hasattr(filterset, "Meta") else ()
+    Meta = type("Meta", bases, attrs)
+    return type(filterset)(
+        str("%sFilterSet" % model._meta.object_name), (filterset,), {"Meta": Meta}
     )
-    return filterset