from pathlib import Path

from django.core.management.base import BaseCommand

from core.importers import import_polos


class Command(BaseCommand):
    help = "Importa polos de cursos_locais.csv e vincula cursos EaD ao primeiro polo encontrado."

    def add_arguments(self, parser) -> None:
        parser.add_argument("--csv", default="docs/cursos_locais.csv")

    def handle(self, *args, **options) -> None:
        result = import_polos(csv_path=Path(options["csv"]))

        self.stdout.write(
            self.style.SUCCESS(
                "Polos importados: "
                f"{result.total} linhas processadas, "
                f"{result.created} criados, "
                f"{result.updated} atualizados, "
                f"{result.cursos_vinculados} cursos EaD vinculados, "
                f"{result.cursos_nao_encontrados} cursos nao encontrados."
            )
        )