Fix bulk_update

tortoise · Aug 1, 2024 · 1bf6b04 · 1bf6b04
1 parent 3b620e4
commit 1bf6b04
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tortoise/queryset.py b/tortoise/queryset.py
@@ -1800,7 +1800,7 @@ def _make_query(self) -> None:
         )
         executor = self._db.executor_class(model=self.model, db=self._db)
         pk_attr = self.model._meta.pk_attr
-        source_pk_attr = self.model._meta.fields_map["id"].source_field or pk_attr
+        source_pk_attr = self.model._meta.fields_map[pk_attr].source_field or pk_attr
         pk = Field(source_pk_attr)
         for objects_item in chunk(self.objects, self.batch_size):
             query = copy(self.query)