0021_populate_batch_status.py

# -*- coding: utf-8 -*-
from __future__ import unicode_literals
import os

from django.db import migrations, models


def populate_status(apps, schema_editor):
    from funkwhale_api.music.utils import compute_status
    ImportBatch = apps.get_model("music", "ImportBatch")

    for ib in ImportBatch.objects.prefetch_related('jobs'):
        ib.status = compute_status(ib.jobs.all())
        ib.save(update_fields=['status'])


def rewind(apps, schema_editor):
    pass


class Migration(migrations.Migration):

    dependencies = [
        ('music', '0020_importbatch_status'),
    ]

    operations = [
        migrations.RunPython(populate_status, rewind),
    ]