diff options
Diffstat (limited to 'ishtar_common/model_merging.py')
| -rw-r--r-- | ishtar_common/model_merging.py | 68 |
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 |