django.setup() from main.models import DVD, MovieCas from cassandra.cqlengine import connection from cassandra.cqlengine.management import sync_table from cassandra.cluster import Cluster dir_name = os.path.dirname(os.path.abspath(__file__)) file_name = "dvd.csv" dvd_csv = os.path.join(dir_name, file_name) csv_file = open(dvd_csv, "r") reader = csv.DictReader(csv_file) for row in reader: dvd_name = row["DVD_Title"] print dvd_name cluster = Cluster(["127.0.0.1"]) session = cluster.connect() session.set_keyspace("movie_database") movie = MovieCas(title=dvd_name) movie.save() cluster.shutdown()
#SQL model new_genre, created = Genre.objects.get_or_create(genre=row['Genre']) new_studio, created = Studio.objects.get_or_create(studio=row['Studio']) new_movie, created = Movie.objects.get_or_create(dvd_title=row['DVD_Title']) new_movie.status = row['Status'] new_movie.price = row['Price'] new_movie.rating = row['Rating'] new_movie.release = row['DVD_ReleaseDate'] new_movie.genre = new_genre new_movie.studio = new_studio try: new_movie.save() except Exception, e: print e #Cassandra model cluster = Cluster(['127.0.0.1']) session = cluster.connect() session.set_keyspace('movie_db2') movie = MovieCas(dvd_title=unidecode(row['DVD_Title'])) movie.id = new_movie.id movie.save() cluster.shutdown() # csv_file.close()