Format - black: ishtar_common

1 files changed, 37 insertions, 31 deletions

diff --git a/ishtar_common/model_merging.py b/ishtar_common/model_merging.py
index 6b839a143..0e1d34c58 100644
--- a/ishtar_common/model_merging.py
+++ b/ishtar_common/model_merging.py
@@ -6,22 +6,26 @@ from django.db.models import Model
 from django.contrib.contenttypes.fields import GenericForeignKey
 from django.core.exceptions import ObjectDoesNotExist
 
-from ishtar_common.utils import get_all_related_many_to_many_objects, \
-    get_all_related_objects
+from ishtar_common.utils import (
+    get_all_related_many_to_many_objects,
+    get_all_related_objects,
+)
 
 
 def get_models():
     _apps = apps.app_configs.items()
     models = []
     for app_name, app_config in _apps:
-        models += [apps.get_model(app_name, m)
-                   for m in apps.get_app_config(app_name).models]
+        models += [
+            apps.get_model(app_name, m) for m in apps.get_app_config(app_name).models
+        ]
     return models
 
 
 @transaction.atomic
-def merge_model_objects(primary_object, alias_objects=None, keep_old=False,
-                        exclude_fields=None):
+def merge_model_objects(
+    primary_object, alias_objects=None, keep_old=False, exclude_fields=None
+):
     """
     Use this function to merge model objects (i.e. Users, Organizations,
     etc.) and migrate all of the related fields from the alias objects to the
@@ -38,7 +42,7 @@ def merge_model_objects(primary_object, alias_objects=None, keep_old=False,
     if not exclude_fields:
         exclude_fields = []
 
-    MERGE_FIELDS = ('merge_candidate', 'merge_exclusion')
+    MERGE_FIELDS = ("merge_candidate", "merge_exclusion")
 
     MERGE_STRING_FIELDS = []
     if getattr(primary_object, "MERGE_STRING_FIELDS", None):
@@ -52,11 +56,11 @@ def merge_model_objects(primary_object, alias_objects=None, keep_old=False,
     primary_class = primary_object.__class__
 
     if not issubclass(primary_class, Model):
-        raise TypeError('Only django.db.models.Model subclasses can be merged')
+        raise TypeError("Only django.db.models.Model subclasses can be merged")
 
     for alias_object in alias_objects:
         if not isinstance(alias_object, primary_class):
-            raise TypeError('Only models of same class can be merged')
+            raise TypeError("Only models of same class can be merged")
 
     # Get a list of all GenericForeignKeys in all models
     # TODO: this is a bit of a hack, since the generics framework should
@@ -64,8 +68,9 @@ def merge_model_objects(primary_object, alias_objects=None, keep_old=False,
     # method to the ForeignKey field for accessing the generic related fields.
     generic_fields = []
     for model in get_models():
-        for field_name, field in filter(lambda x: isinstance(
-                x[1], GenericForeignKey), model.__dict__.items()):
+        for field_name, field in filter(
+            lambda x: isinstance(x[1], GenericForeignKey), model.__dict__.items()
+        ):
             generic_fields.append(field)
 
     blank_local_fields = set()
@@ -74,7 +79,7 @@ def merge_model_objects(primary_object, alias_objects=None, keep_old=False,
         # string fields with only spaces are empty fields
         if isinstance(value, str):
             value = value.strip()
-        if value in [None, '']:
+        if value in [None, ""]:
             blank_local_fields.add(field.attname)
 
     # Loop through all alias objects and migrate their data to the primary
@@ -89,14 +94,16 @@ def merge_model_objects(primary_object, alias_objects=None, keep_old=False,
             obj_varname = related_object.field.name
             if obj_varname in exclude_fields:
                 continue
-            if getattr(related_object.field, "related_model", None) and \
-                    not related_object.related_model._meta.managed:
+            if (
+                getattr(related_object.field, "related_model", None)
+                and not related_object.related_model._meta.managed
+            ):
                 continue
             try:
                 related_objects = getattr(alias_object, alias_varname)
             except ObjectDoesNotExist:
                 continue
-            if not hasattr(related_objects, 'all'):
+            if not hasattr(related_objects, "all"):
                 # one to one field
                 setattr(related_objects, obj_varname, primary_object)
                 related_objects.save()
@@ -107,8 +114,7 @@ def merge_model_objects(primary_object, alias_objects=None, keep_old=False,
 
         # Migrate all many to many references from alias object to primary
         # object.
-        related_many_objects = \
-            get_all_related_many_to_many_objects(alias_object)
+        related_many_objects = get_all_related_many_to_many_objects(alias_object)
         related_many_object_names = set()
         for related_many_object in related_many_objects:
             alias_varname = related_many_object.get_accessor_name()
@@ -118,13 +124,11 @@ def merge_model_objects(primary_object, alias_objects=None, keep_old=False,
 
             if alias_varname is not None:
                 # standard case
-                q_related_many_objects = getattr(
-                    alias_object, alias_varname).all()
+                q_related_many_objects = getattr(alias_object, alias_varname).all()
                 related_many_object_names.add(alias_varname)
             else:
                 # special case, symmetrical relation, no reverse accessor
-                q_related_many_objects = getattr(
-                    alias_object, obj_varname).all()
+                q_related_many_objects = getattr(alias_object, obj_varname).all()
                 related_many_object_names.add(obj_varname)
             for obj in q_related_many_objects.all():
                 getattr(obj, obj_varname).remove(alias_object)
@@ -134,8 +138,10 @@ def merge_model_objects(primary_object, alias_objects=None, keep_old=False,
         # object.
         for many_to_many_object in alias_object._meta.many_to_many:
             alias_varname = many_to_many_object.get_attname()
-            if alias_varname in related_many_object_names or \
-               alias_varname in MERGE_FIELDS:
+            if (
+                alias_varname in related_many_object_names
+                or alias_varname in MERGE_FIELDS
+            ):
                 continue
 
             many_to_many_objects = getattr(alias_object, alias_varname).all()
@@ -150,21 +156,21 @@ def merge_model_objects(primary_object, alias_objects=None, keep_old=False,
         for field in generic_fields:
             filter_kwargs = {}
             filter_kwargs[field.fk_field] = alias_object._get_pk_val()
-            filter_kwargs[field.ct_field] = field.get_content_type(
-                alias_object)
-            for generic_related_object in field.model.objects.filter(
-                    **filter_kwargs):
+            filter_kwargs[field.ct_field] = field.get_content_type(alias_object)
+            for generic_related_object in field.model.objects.filter(**filter_kwargs):
                 if field.name in exclude_fields:
                     continue
                 setattr(generic_related_object, field.name, primary_object)
                 generic_related_object.save()
 
         for field_name in MERGE_STRING_FIELDS:
-            if getattr(primary_object, field_name) and \
-                    getattr(alias_object, field_name):
+            if getattr(primary_object, field_name) and getattr(
+                alias_object, field_name
+            ):
                 val = "{} ; {}".format(
                     getattr(primary_object, field_name),
-                    getattr(alias_object, field_name))
+                    getattr(alias_object, field_name),
+                )
                 if field_name in exclude_fields:
                     continue
                 setattr(primary_object, field_name, val)
@@ -174,7 +180,7 @@ def merge_model_objects(primary_object, alias_objects=None, keep_old=False,
         filled_up = set()
         for field_name in blank_local_fields:
             val = getattr(alias_object, field_name)
-            if val not in [None, '']:
+            if val not in [None, ""]:
                 setattr(primary_object, field_name, val)
                 filled_up.add(field_name)
         blank_local_fields -= filled_up