def writeModel(self, root, title, year, genre, source, cover, desc):
obj = Movie()
obj.title = title
obj.year = year
obj.genre = genre
obj.source = source
obj.cover = cover
obj.description = desc
obj.save()
def handle(self, *args, **options):
if Movie.objects.exists():
print('Movie data already loaded...exiting.')
print(ALREADY_LOADED_ERROR_MESSAGE)
return
print("Creating movie data")
for row in DictReader(open('./movies.csv')):
movie = Movie()
movie.title = row['Title']
#movie.movie_poster_image = models.ImageField(blank=True, null=True,
#upload_to="posters/")
movie.release_date = row['Release_date']
movie.director = row['Director']
movie.rating = row['Rating']
movie.genre = row['Genre']
movie.description = row['Description']
movie.save()
def make_model():
movie = Movie()
try:
movie.start_date = movie_dict['기간시작']
except:
pass
try:
movie.end_date = movie_dict['기간종료']
except:
pass
try:
movie.rank = movie_dict['순위']
except:
pass
try:
movie.poster_url = movie_dict['썸네일_이미지의_URL']
except:
pass
try:
movie.title = movie_dict['영화명(국문)']
except:
pass
try:
movie.subtitle = movie_dict['영화명(영문)']
except:
pass
try:
movie.actor = movie_dict['배우']
except:
pass
try:
movie.movie_code = movie_dict['영화코드']
except:
pass
try:
movie.naver_movie_url = movie_dict['하이퍼텍스트_링크']
except:
pass
try:
movie.description = movie_dict['줄거리']
except:
pass
movie.save()
def add_data(request):
conn = sqlite3.connect('movies.db')
cursor = conn.execute("SELECT * from mytable")
for row in cursor:
m = Movie(name=row[7], rating=row[19])
if row[21] != None:
m.cast = row[21]
else:
m.cast = 'None'
id = None
if row[23] != None:
if Director.objects.filter(name=row[23]).exists():
id = Director.objects.filter(name=row[23]).first().id
else:
d = Director(name=row[23])
d.save()
id = d.id
m.director = d
else:
if not Director.objects.filter(name='None').exists():
Director(name='None').save()
d = Director.objects.filter(name='None').first()
m.director = d
genre = None
try:
genre = row[2].split()
for g in genre:
if Genre.objects.filter(genre=g).exists():
gen = Genre.objects.filter(genre=g).first()
m.genre.add(gen)
else:
gen = Genre(genre=g)
gen.save()
m.genre.add(gen)
except:
pass
m.description = 'Lorem ipsum dolor sit amet, consectetur adipiscing elit. Aliquam sed ligula libero. Maecenas ut semper mi. Duis ut tempus urna, sit amet condimentum dolor. Donec sollicitudin lacus ut placerat vestibulum. Nam eu neque ut tellus feugiat cursus non a eros. Fusce gravida nisl ac diam iaculis dignissim. Quisque eleifend tristique enim, semper facilisis nisi rhoncus in. Ut at velit dolor. Etiam placerat est sed sollicitudin pulvinar. Phasellus consectetur arcu et massa porttitor convallis. In posuere a ex sed mattis. Donec vitae dui cursus, tristique nibh a, fringilla dui. Nam ornare in ante nec ornare.'
m.save()
print(m.name)
return JsonResponse({})
def store_movies(year):
print(f'storing-----------{year}')
with open(f'./data/movie_{year}.json', 'r') as fp:
movies = json.load(fp)
# delete stored movies
movies_year = Movie.objects.filter(release_date__year=year)
movies_year.delete()
# using bulk
to_db_movies = []
for mv in tqdm(movies):
movie = Movie()
movie.pk = mv['id']
movie.title_en = mv['original_title']
movie.title_ko = mv['title']
movie.rate = mv['vote_average']
movie.rate_people_count = mv['vote_count']
movie.poster_path = mv['poster_path']
movie.description = mv['overview']
movie.release_date = datetime.datetime.strptime(
mv['release_date'], '%Y-%m-%d')
to_db_movies.append(movie)
div = 100
to_db_movies_batch = []
for i in range(len(to_db_movies) // div + 1):
to_db_movies_batch.append(to_db_movies[i * div:(i + 1) * div])
for idx, batch in enumerate(to_db_movies_batch):
print(round(idx / len(to_db_movies_batch), 2), end='|')
temp = Movie.objects.bulk_create(batch)
# add genre
mgs = []
for mv in movies:
for gid in mv['genre_ids']:
mg = MG()
mg.movie_id = mv['id']
mg.genre_id = gid
mgs.append(mg)
temp = MG.objects.bulk_create(mgs)
