import sys
from django.db import migrations

from ishtar_common.utils_migrations import update_import_key, print_progress


def update_importkey(apps, __):
    updated = 0
    GeoVectorData = apps.get_model("ishtar_common", "geovectordata")
    total = GeoVectorData.objects.count()
    sys.stdout.write("\n")
    for idx, data in enumerate(GeoVectorData.objects.all()):
        print_progress(idx, total)
        updated += 1 if update_import_key(data) else 0


class Migration(migrations.Migration):

    dependencies = [
        ('ishtar_common', '0269_gis_importchunk'),
    ]

    operations = [
        migrations.RunPython(update_importkey)
    ]