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

from django.db import migrations, models
from funkwhale_api.music.utils import guess_mimetype


def populate_mimetype(apps, schema_editor):
    TrackFile = apps.get_model("music", "TrackFile")

    for tf in TrackFile.objects.filter(audio_file__isnull=False, mimetype__isnull=True).only('audio_file'):
        try:
            tf.mimetype = guess_mimetype(tf.audio_file)
        except Exception as e:
            print('Error on track file {}: {}'.format(tf.pk, e))
            continue
        print('Track file {}: {}'.format(tf.pk, tf.mimetype))
        tf.save(update_fields=['mimetype'])


def rewind(apps, schema_editor):
    pass


class Migration(migrations.Migration):

    dependencies = [
        ('music', '0018_auto_20180218_1554'),
    ]

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