# Generated by Django 4.1.1 on 2022-10-05 13:49

from django.db import migrations


from OSM_outils.OSMnx_recuperation_graph_information import OSMnx_recuperation_graph_information
import numpy as np

def ajoute_donnees_nouvelle_colonne(apps, schema_editor):
    """
    Cette méthode permet de faire une migration de donnée pour la nouvelle colonne ajoutée

    """
    graph_viaire_ville_information = apps.get_model('recuperateur_reseau_viaire', 'Graph_viaire_ville_information')
    fichier_retour_methode_fonction =  np.load("./retour_method.npy", allow_pickle=True)
    for graph in graph_viaire_ville_information.objects.all():
        graph.gvvi_degre_moyen = fichier_retour_methode_fonction.item().get(graph.gvvi_nom)
        graph.save()


class Migration(migrations.Migration):

    dependencies = [
        ('recuperateur_reseau_viaire', '0004_graph_viaire_ville_information_gvvi_degre_moyen'),
    ]

    operations = [
        migrations.RunPython(ajoute_donnees_nouvelle_colonne)
    ]
