funkwhale/api/funkwhale_api/music/migrations/0024_populate_uuid.py

78 wiersze
2.3 KiB
Python

# -*- coding: utf-8 -*-
from __future__ import unicode_literals
import uuid
from django.db import migrations, models
def populate_uuids(apps, schema_editor):
models = [
"Album",
"Artist",
"Importbatch",
"Importjob",
"Lyrics",
"Track",
"Trackfile",
"Work",
]
for m in models:
kls = apps.get_model("music", m)
qs = kls.objects.filter(uuid__isnull=True).only("id")
print("Setting uuids for {} ({} objects)".format(m, len(qs)))
for o in qs:
o.uuid = uuid.uuid4()
o.save(update_fields=["uuid"])
def rewind(apps, schema_editor):
pass
class Migration(migrations.Migration):
dependencies = [("music", "0023_auto_20180407_1010")]
operations = [
migrations.RunPython(populate_uuids, rewind),
migrations.AlterField(
model_name="album",
name="uuid",
field=models.UUIDField(db_index=True, default=uuid.uuid4, unique=True),
),
migrations.AlterField(
model_name="artist",
name="uuid",
field=models.UUIDField(db_index=True, default=uuid.uuid4, unique=True),
),
migrations.AlterField(
model_name="importbatch",
name="uuid",
field=models.UUIDField(db_index=True, default=uuid.uuid4, unique=True),
),
migrations.AlterField(
model_name="importjob",
name="uuid",
field=models.UUIDField(db_index=True, default=uuid.uuid4, unique=True),
),
migrations.AlterField(
model_name="lyrics",
name="uuid",
field=models.UUIDField(db_index=True, default=uuid.uuid4, unique=True),
),
migrations.AlterField(
model_name="track",
name="uuid",
field=models.UUIDField(db_index=True, default=uuid.uuid4, unique=True),
),
migrations.AlterField(
model_name="trackfile",
name="uuid",
field=models.UUIDField(db_index=True, default=uuid.uuid4, unique=True),
),
migrations.AlterField(
model_name="work",
name="uuid",
field=models.UUIDField(db_index=True, default=uuid.uuid4, unique=True),
),
]