from formulario.models import Municipio, Localidad

def run():
    fn = open("scripts/localidades.txt","r",encoding="utf8")

    for line in fn:
        aux = line.split(",")

        mun = aux[0]
        loc = aux[1]

        try:
            mun_obj = Municipio.objects.get(nombre=mun)
        except:
            mun_obj = Municipio(nombre=mun)
            mun_obj.save()

        loc_obj = Localidad(nombre=loc,municipio=mun_obj.id)
        loc_obj.save()
        print("Guardada localidad " + loc + ", " + mun)