Last active
August 8, 2017 14:33
-
-
Save ebsaral/d419e7f056aa7a93a9ce5320a1da1d09 to your computer and use it in GitHub Desktop.
Replace duplicate objects with the original
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from django.db import transaction | |
from django.db.models.deletion import Collector | |
from django.db.models.fields.related import ForeignKey, ManyToManyField, OneToOneField | |
duplicates = [] | |
items = {} | |
client = Client.objects.get(name='Demo') | |
for c in Company.objects.filter(active=True, client=client): | |
company_name = c.name | |
if c.id in duplicates: | |
continue | |
else: | |
other_companies = Company.objects.filter(client=client, active=True, name=company_name).exclude(id=c.id) | |
if other_companies.exists(): | |
ids = [x.id for x in other_companies] | |
items[c.id] = ids | |
duplicates += ids | |
black_list = ['Company_company_owners'] | |
with transaction.atomic(): | |
for k, v in items.iteritems(): | |
org = Company.objects.get(id=k) | |
for s_v in v: | |
dup = Company.objects.get(id=s_v) | |
collector = Collector(using='default') | |
collector.collect([dup]) | |
for model, instance in collector.instances_with_model(): | |
model_name = model.__name__ | |
if model_name not in black_list: | |
print dup.id | |
fields_with_foreign = model._meta.fields | |
for f in fields_with_foreign: | |
if isinstance(f, ForeignKey) or isinstance(f, OneToOneField): | |
f_name = f.rel.field.attname | |
if f.rel.to is org.__class__: | |
print 'Foreign Key is setting %s to %s of instance %s, %s' % (f_name, org.id, instance, model) | |
setattr(instance, f.rel.field.attname, org.id) | |
many_to_many_fields = model._meta.many_to_many | |
for f in many_to_many_fields: | |
f_name = f.rel.field.attname | |
if f.rel.to is org.__class__: | |
print 'ManytoMany Key is setting %s to %s from %s of instance %s, %s' % (f_name, org.id, dup.id, instance, model) | |
instance.__getattribute__(f_name).remove(dup) | |
instance.__getattribute__(f_name).add(org) | |
instance.save() | |
dup.active = False | |
dup.save() | |
print "Duplicate %s set to deactivate" % dup.id |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment