code formatting

Author: VKFisher

bulk_sync/__init__.py

@@ -1,5 +1,6 @@
 from collections import OrderedDict
 import logging
+
 from django.db import transaction
 from django.core.exceptions import FieldDoesNotExist
 
@@ -16,7 +17,8 @@ def bulk_sync(
     `new_models`: Django ORM objects that are the desired state.  They may or may not have `id` set.
     `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.  Use `None` or `[]` if you want to sync against the entire table.
+    `filters`: Q() filters specifying the subset of the database to work in. Use `None` or `[]` if you want to sync
+            against the entire table.
     `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`: (optional) list of fields to update. If not set, will sync all fields that are editable and not
@@ -71,7 +73,7 @@ def get_key(obj):
 
         if not skip_creates:
             db_class.objects.bulk_create(new_objs, batch_size=batch_size)
-            
+
         if not skip_updates:
             db_class.objects.bulk_update(existing_objs, fields=fields, batch_size=batch_size)
 
@@ -82,8 +84,8 @@ def get_key(obj):
         assert len(existing_objs) == len(new_models) - len(new_objs)
 
         stats = {
-            "created": 0 if skip_creates else len(new_objs), 
-            "updated": 0 if skip_updates else (len(new_models) - len(new_objs)), 
+            "created": 0 if skip_creates else len(new_objs),
+            "updated": 0 if skip_updates else (len(new_models) - len(new_objs)),
             "deleted": 0 if skip_deletes else len(obj_dict)
         }
 

runtest.py

@@ -15,4 +15,3 @@
     test_runner = TestRunner()
     failures = test_runner.run_tests(["tests"])
     sys.exit(bool(failures))
-    

tests/tests.py

@@ -84,13 +84,12 @@ def test_provided_pk_is_retained_but_raises_if_mismatch_with_keyfield(self):
         ret = bulk_sync(new_models=new_objs, filters=Q(company_id=c1.id), key_fields=("name",))
 
         self.assertEqual(0, ret["stats"]["updated"])
-        self.assertEqual(1, ret["stats"]["created"]) # Added 'Notscott'
-        self.assertEqual(1, ret["stats"]["deleted"]) # Deleted 'Scott'
+        self.assertEqual(1, ret["stats"]["created"])  # Added 'Notscott'
+        self.assertEqual(1, ret["stats"]["deleted"])  # Deleted 'Scott'
 
         # Make sure we retained the PK
         self.assertEqual(Employee.objects.filter(id=unique_pk).count(), 1)
 
-
     def test_fields_parameter(self):
         c1 = Company.objects.create(name="Foo Products, Ltd.")
         c2 = Company.objects.create(name="Bar Microcontrollers, Inc.")
@@ -220,13 +219,11 @@ def test_skip_updates(self):
         self.assertEqual(2, Employee.objects.count())
         self.assertEqual(["Scott", "Alice"], [x.name for x in Employee.objects.all().order_by('id')])
 
-
         self.assertEqual(0, ret["stats"]["updated"])
         self.assertEqual(1, ret["stats"]["created"])
         self.assertEqual(1, ret["stats"]["deleted"])
 
 
-
 class BulkCompareTests(TestCase):
     """ Test `bulk_compare` method """