mapped = data_name_mapping.get(column.get('name'))
column_pos[mapped] = pos
pos += 1
for movie_data in data.get('data'):
movie_api_id = movie_data[column_pos.get('api_id')]
title = movie_data[column_pos.get('title')]
year = movie_data[column_pos.get('year')]
director = movie_data[column_pos.get('director')]
movie = Movie.query.filter(Movie.title == title,
Movie.year == year,
Movie.director == director).first()
if not movie:
movie = Movie()
movie.api_id = movie_data[column_pos.get('api_id')]
sys.stdout.write('\rNEW Movie %s' % movie.api_id)
movie.title = movie_data[column_pos.get('title')]
movie.year = movie_data[column_pos.get('year')]
movie.director = movie_data[column_pos.get('director')]
# Set actors
actors = [
movie_data[column_pos.get('actor_1')],
movie_data[column_pos.get('actor_2')],
movie_data[column_pos.get('actor_3')]
]
for actor_name in [a for a in actors if a]:
actor = Actor.query.filter(Actor.name == actor_name).first()
if not actor:
actor = Actor(actor_name)
movie.actors.append(actor)
name = column.get('name')
if name:
mapped = data_name_mapping.get(column.get('name'))
column_pos[mapped] = pos
pos += 1
for movie_data in data.get('data'):
movie_api_id = movie_data[column_pos.get('api_id')]
title = movie_data[column_pos.get('title')]
year = movie_data[column_pos.get('year')]
director = movie_data[column_pos.get('director')]
movie = Movie.query.filter(Movie.title == title, Movie.year == year,
Movie.director == director).first()
if not movie:
movie = Movie()
movie.api_id = movie_data[column_pos.get('api_id')]
sys.stdout.write('\rNEW Movie %s' % movie.api_id)
movie.title = movie_data[column_pos.get('title')]
movie.year = movie_data[column_pos.get('year')]
movie.director = movie_data[column_pos.get('director')]
# Set actors
actors = [
movie_data[column_pos.get('actor_1')],
movie_data[column_pos.get('actor_2')],
movie_data[column_pos.get('actor_3')]
]
for actor_name in [a for a in actors if a]:
actor = Actor.query.filter(Actor.name == actor_name).first()
if not actor:
actor = Actor(actor_name)
movie.actors.append(actor)
