Switch to Django 2.2. bulk update.

Author: keyz182

README.md

@@ -49,6 +49,7 @@ key_fields = ('name', )
 ret = bulk_sync(
         new_models=new_models,
         filters=filters,
+        fields=['name', 'phone_number', ...],
         key_fields=key_fields)
 
 print("Results of bulk_sync: "
@@ -66,6 +67,7 @@ Combine bulk create, update, and delete.  Make the DB match a set of in-memory o
 - `key_fields`: Identifying attribute name(s) to match up `new_models` items with database rows.  If a foreign key is being used as a key field, be sure to pass the `fieldname_id` rather than the `fieldname`.
 - `filters`: Q() filters specifying the subset of the database to work in.
 - `batch_size`: passes through to Django `bulk_create.batch_size` and `bulk_update.batch_size`, and controls how many objects are created/updated per SQL query.
+- `fields`: a list of fields to update - passed through to Django's built in `buld_update` 
 
 `def bulk_compare(old_models, new_models, key_fields, ignore_fields=None):`
 Compare two sets of models by `key_fields`.
@@ -85,4 +87,4 @@ Compare two sets of models by `key_fields`.
 
 ## Frameworks Supported
 
-This library is tested using Python 3 against Django 1.11 and Django 2.2.
+This library is tested using Python 3 against Django 2.2.

bulk_sync/__init__.py

@@ -2,12 +2,11 @@
 import logging
 
 from django.db import transaction
-from django_bulk_update.helper import bulk_update
 
 logger = logging.getLogger(__name__)
 
 
-def bulk_sync(new_models, key_fields, filters, batch_size=None):
+def bulk_sync(new_models, key_fields, filters, batch_size=None, fields=None):
     """ Combine bulk create, update, and delete.  Make the DB match a set of in-memory objects.
 
     `new_models`: Django ORM objects that are the desired state.  They may or may not have `id` set.
@@ -16,10 +15,17 @@ def bulk_sync(new_models, key_fields, filters, batch_size=None):
     `filters`: Q() filters specifying the subset of the database to work in.
     `batch_size`: passes through to Django `bulk_create.batch_size` and `bulk_update.batch_size`, and controls
             how many objects are created/updated per SQL query.
+    `fields`: a list of fields to update - passed to django's bulk_update
 
     """
     db_class = new_models[0].__class__
 
+    if not fields:
+        # Get a list of fields that aren't PKs and aren't editable (e.g. auto_add_now) for bulk_update
+        fields = [field.name
+                  for field in db_class._meta.fields
+                  if not field.primary_key and not field.auto_created and field.editable]
+
     with transaction.atomic():
         objs = db_class.objects.all()
         if filters:
@@ -45,8 +51,7 @@ def get_key(obj):
                 existing_objs.append(new_obj)
 
         db_class.objects.bulk_create(new_objs, batch_size=batch_size)
-
-        bulk_update(existing_objs, batch_size=batch_size)
+        db_class.objects.bulk_update(existing_objs, fields=fields, batch_size=batch_size)
 
         # delete stale ones...
         objs.filter(pk__in=[_.pk for _ in list(obj_dict.values())]).delete()

requirements.txt

@@ -1,2 +1 @@
-Django>=1.8
-django-bulk-update>2
+Django>=2.2