Blacked the code

This commit is contained in:
Eliot Berriot 2018-06-09 15:36:16 +02:00
parent b6fc0051fa
commit 62ca3bd736
No known key found for this signature in database
GPG key ID: DD6965E2476E5C27
279 changed files with 8861 additions and 9527 deletions

View file

@ -5,7 +5,14 @@ from django.db import migrations, models
def get_duplicates(model):
return [i['mbid'] for i in model.objects.values('mbid').annotate(idcount=models.Count('mbid')).order_by('-idcount') if i['idcount'] > 1]
return [
i["mbid"]
for i in model.objects.values("mbid")
.annotate(idcount=models.Count("mbid"))
.order_by("-idcount")
if i["idcount"] > 1
]
def deduplicate(apps, schema_editor):
Artist = apps.get_model("music", "Artist")
@ -13,28 +20,25 @@ def deduplicate(apps, schema_editor):
Track = apps.get_model("music", "Track")
for mbid in get_duplicates(Artist):
ref = Artist.objects.filter(mbid=mbid).order_by('pk').first()
ref = Artist.objects.filter(mbid=mbid).order_by("pk").first()
duplicates = Artist.objects.filter(mbid=mbid).exclude(pk=ref.pk)
Album.objects.filter(artist__in=duplicates).update(artist=ref)
Track.objects.filter(artist__in=duplicates).update(artist=ref)
duplicates.delete()
for mbid in get_duplicates(Album):
ref = Album.objects.filter(mbid=mbid).order_by('pk').first()
ref = Album.objects.filter(mbid=mbid).order_by("pk").first()
duplicates = Album.objects.filter(mbid=mbid).exclude(pk=ref.pk)
Track.objects.filter(album__in=duplicates).update(album=ref)
duplicates.delete()
def rewind(*args, **kwargs):
pass
class Migration(migrations.Migration):
dependencies = [
('music', '0004_track_tags'),
]
dependencies = [("music", "0004_track_tags")]
operations = [
migrations.RunPython(deduplicate, rewind),
]
operations = [migrations.RunPython(deduplicate, rewind)]