autocensup/core/management/commands/import_polos.py

27 lines
891 B
Python

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."
)
)