import sys, os
import pandas as pd


os.environ.setdefault("DJANGO_SETTINGS_MODULE", "movierec.settings")


import django
django.setup()


from movies.models import Movie


def save_movie_from_row(movie_row):
    movie = Movie()
    movie.title = movie_row[1]
    movie.genres = movie_row[2]
    movie.save()


if __name__ == "__main__":

    if len(sys.argv) == 2:
        print("Reading from file {}".format(str(sys.argv[1])))
        movies_df = pd.read_csv(sys.argv[1])
        print(movies_df)

        movies_df.apply(
            save_movie_from_row,
            axis=1
        )

        print("There are {} reviews in DB".format(Movie.objects.count()))
    else:
        print("Please, provide Movie file path")