Skip to content

Commit d412f61

Browse files
committed
Correctly handle plural vs singular queryset
1 parent d8710cb commit d412f61

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

ansible_base/resource_registry/fields.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,11 @@ class CustomForwardOneToOneDescriptor(ForwardOneToOneDescriptor):
1313
def get_queryset(self, **hints):
1414
return self.field.remote_field.model._base_manager.db_manager(hints=hints).filter(content_type=ContentType.objects.get_for_model(self.field.model))
1515

16-
def get_prefetch_querysets(self, instances, queryset=None):
17-
if not queryset:
16+
def get_prefetch_querysets(self, instances, querysets=None):
17+
if querysets is None:
1818
queryset = self.get_queryset()
19+
else:
20+
queryset = querysets[0]
1921
queryset._add_hints(instance=instances[0])
2022

2123
query = models.Q.create(
@@ -49,7 +51,9 @@ def get_prefetch_querysets(self, instances, queryset=None):
4951

5052
def get_prefetch_queryset(self, instances, queryset=None):
5153
# Django 4 compatibility: renamed to get_prefetch_querysets in Django 5
52-
return self.get_prefetch_querysets(instances, queryset)
54+
if queryset is None:
55+
return self.get_prefetch_querysets(instances)
56+
return self.get_prefetch_querysets(instances, [queryset])
5357

5458

5559
class AnsibleResourceField(models.ForeignObject):

0 commit comments

Comments
 (0)