Fix issue with class MRO

AlanCoding · May 8, 2024 · cdf4303 · cdf4303
1 parent f6d23e2
commit cdf4303
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/ansible_base/lib/routers/association_resource_router.py b/ansible_base/lib/routers/association_resource_router.py
@@ -141,7 +141,10 @@ def get_association_queryset(self) -> QuerySet:
             qs = cls.objects.all()
 
         if self.action == 'associate':
-            return self.filter_queryset(qs)
+            if hasattr(self, 'filter_associate_queryset'):
+                return self.filter_associate_queryset(qs)
+            else:
+                return self.filter_queryset(qs)
         return qs
 
     def get_serializer_class(self):

diff --git a/test_app/router.py b/test_app/router.py
@@ -17,9 +17,8 @@ def filter_queryset(self, qs):
         "Do not filter users for the related list view, /organizations/42/users/"
         return self.apply_optimizations(qs)
 
-    def get_association_queryset(self):
+    def filter_associate_queryset(self, qs):
         "Use RBAC filter when associating new users, /organizations/42/users/associate/"
-        qs = super.get_assocation_queryset()
         return super().filter_queryset(qs)