import os
import django

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'config.settings')
django.setup()

from interview.models import Organization, Group, Applicant

orgs_data = [
    {"type": "Ingeniería", "career": "Ingeniería en Desarrollo de Software", "groups": 4, "gradientIndex": 0},
    {"type": "Técnico", "career": "Hardware Computacional", "groups": 2, "gradientIndex": 1},
    {"type": "Ingeniería", "career": "Ingeniería de Infraestructura de Redes Informáticas", "groups": 0, "gradientIndex": 4},
    {"type": "Ingeniería", "career": "Ingeniería en Informática Inteligente", "groups": 1, "gradientIndex": 3}
]

created_orgs = []
for data in orgs_data:
    org, _ = Organization.objects.get_or_create(
        career=data["career"],
        defaults={"type": data["type"], "gradient_index": data["gradientIndex"]}
    )
    created_orgs.append(org)
    if org.groups.count() == 0:
        for i in range(data["groups"]):
            Group.objects.create(name=f"Grupo {i+1}", organization=org)

# Assign random organizations to existing applicants
import random
apps = Applicant.objects.all()
for a in apps:
    a.organization = random.choice(created_orgs)
    a.save()

print("Orgs seeded and applicants updated.")