From c93f2aff4e8fa74e3b8cbef5682fb9e65ed40aa2 Mon Sep 17 00:00:00 2001 From: James Kassemi Date: Wed, 8 Jan 2025 13:53:36 -0500 Subject: [PATCH] fix: batch size applied properly to new qs --- mongoengine/queryset/queryset.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/mongoengine/queryset/queryset.py b/mongoengine/queryset/queryset.py index 0b065853f..caa6bc43d 100644 --- a/mongoengine/queryset/queryset.py +++ b/mongoengine/queryset/queryset.py @@ -806,6 +806,7 @@ def hint(self, index=None): queryset = self.clone() queryset._hint = index + # If a cursor object has already been created, apply the hint to it. if queryset._cursor_obj: queryset._cursor_obj.hint(queryset._hint) @@ -813,8 +814,12 @@ def hint(self, index=None): def batch_size(self, size): queryset = self.clone() - queryset._cursor.batch_size(size) queryset._batch_size = size + + # If a cursor object has already been created, apply the batch size to it. + if queryset._cursor_obj: + queryset._cursor_obj.batch_size(queryset._batch_size) + return queryset def distinct(self, field, dereference=True):