from django.core.management.base import BaseCommand, CommandError
from lyrics.models import Album, Artista, Song
import csv
import argparse
from datetime import datetime
import pandas as pd


class Command(BaseCommand):
    help = "Importa la lista de letras"

    def add_arguments(self, parser):
        parser.add_argument("fichero_csv", type=str, help='Ruta al fichero csv')

    def handle(self, *args, **options):
        fichero = options["fichero_csv"]
        contenido = pd.read_csv(fichero)

        for fila in contenido.iterrows():
            print(fila[1].title, ", ", fila[1].artist, ", ", fila[1].album)

            artista = Artista.objects.get(nombre=fila[1].artist)
            album = Album.objects.get(pk=fila[1].album)

            cancion = Song(title=fila[1].title, artist=artista, album=album, pista=fila[1].pista, lyrics=fila[1].lyrics)
            cancion.save()