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")
aW1wb3J0IHN5cywgb3MKaW1wb3J0IHBhbmRhcyBhcyBwZAoKCm9zLmVudmlyb24uc2V0ZGVmYXVsdCgiREpBTkdPX1NFVFRJTkdTX01PRFVMRSIsICJtb3ZpZXJlYy5zZXR0aW5ncyIpCgoKaW1wb3J0IGRqYW5nbwpkamFuZ28uc2V0dXAoKQoKCmZyb20gbW92aWVzLm1vZGVscyBpbXBvcnQgTW92aWUKCgpkZWYgc2F2ZV9tb3ZpZV9mcm9tX3Jvdyhtb3ZpZV9yb3cpOgogICAgbW92aWUgPSBNb3ZpZSgpCiAgICBtb3ZpZS50aXRsZSA9IG1vdmllX3Jvd1sxXQogICAgbW92aWUuZ2VucmVzID0gbW92aWVfcm93WzJdCiAgICBtb3ZpZS5zYXZlKCkKCgppZiBfX25hbWVfXyA9PSAiX19tYWluX18iOgoKICAgIGlmIGxlbihzeXMuYXJndikgPT0gMjoKICAgICAgICBwcmludCgiUmVhZGluZyBmcm9tIGZpbGUge30iLmZvcm1hdChzdHIoc3lzLmFyZ3ZbMV0pKSkKICAgICAgICBtb3ZpZXNfZGYgPSBwZC5yZWFkX2NzdihzeXMuYXJndlsxXSkKICAgICAgICBwcmludChtb3ZpZXNfZGYpCgogICAgICAgIG1vdmllc19kZi5hcHBseSgKICAgICAgICAgICAgc2F2ZV9tb3ZpZV9mcm9tX3JvdywKICAgICAgICAgICAgYXhpcz0xCiAgICAgICAgKQoKICAgICAgICBwcmludCgiVGhlcmUgYXJlIHt9IHJldmlld3MgaW4gREIiLmZvcm1hdChNb3ZpZS5vYmplY3RzLmNvdW50KCkpKQogICAgZWxzZToKICAgICAgICBwcmludCgiUGxlYXNlLCBwcm92aWRlIE1vdmllIGZpbGUgcGF0aCIpCg==