from django.db import migrations

from ishtar_common.utils_migrations import update_import_key


def update_importkey(apps, __):
    updated = 0
    GeoVectorData = apps.get_model("ishtar_common", "geovectordata")
    for data in GeoVectorData.objects.all():
        updated += 1 if update_import_key(data) else 0
    if updated:
        print()
        print(f"* {updated} GeoVectorData import_key updated")


class Migration(migrations.Migration):

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

    operations = [
        migrations.RunPython(update_importkey)
    ]