Replace expensive count() with cheap exists() (#478)

model-bakers · Jun 3, 2024 · 1899c19 · 1899c19
1 parent 01ebed3
commit 1899c19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model_bakery/recipe.py b/model_bakery/recipe.py
@@ -52,7 +52,7 @@ def _mapping(  # noqa: C901
                     m = finder.get_model(self._model)
                 else:
                     m = self._model
-                if k not in self._iterator_backups or m.objects.count() == 0:
+                if k not in self._iterator_backups or not m.objects.exists():
                     self._iterator_backups[k] = itertools.tee(
                         self._iterator_backups.get(k, [v])[0]
                     )