Skip to content

Commit

Permalink
Merge pull request #972 from stianjensen/restore-bulk-create
Browse files Browse the repository at this point in the history
Re-introduce bulk_create on supported databases
  • Loading branch information
etianen authored Aug 9, 2024
2 parents a8bedda + cefcd2a commit 4ac5f35
Showing 1 changed file with 11 additions and 2 deletions.
13 changes: 11 additions & 2 deletions reversion/revisions.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
from django.apps import apps
from django.core import serializers
from django.core.exceptions import ObjectDoesNotExist
from django.db import models, transaction, router
from django.db import models, transaction, router, connections
from django.db.models.query import QuerySet
from django.db.models.signals import post_save, m2m_changed
from django.utils.encoding import force_str
Expand Down Expand Up @@ -211,6 +211,7 @@ def add_to_revision(obj, model_db=None):

def _save_revision(versions, user=None, comment="", meta=(), date_created=None, using=None):
from reversion.models import Revision
from reversion.models import Version
# Only save versions that exist in the database.
# Use _base_manager so we don't have problems when _default_manager is overriden
model_db_pks = defaultdict(lambda: defaultdict(set))
Expand Down Expand Up @@ -248,9 +249,17 @@ def _save_revision(versions, user=None, comment="", meta=(), date_created=None,
# Save the revision.
revision.save(using=using)
# Save version models.

can_use_bulk_create = connections[using].features.can_return_rows_from_bulk_insert

for version in versions:
version.revision = revision
version.save(using=using)
if not can_use_bulk_create:
version.save(using=using)

if can_use_bulk_create:
Version.objects.using(using).bulk_create(versions)

# Save the meta information.
for meta_model, meta_fields in meta:
meta_model._base_manager.db_manager(using=using).create(
Expand Down

0 comments on commit 4ac5f35

Please sign in to comment.