def create_movie(db: SQLAlchemy, data: MovieData):
movie = Movie.query.filter_by(imdb_id=data.imdb_id).first()
if movie is None:
data.look_up()
movie = Movie(imdb_id=data.imdb_id,
imdb_url=data.imdb_url,
title=data.title,
avatar=data.avatar,
description=data.description,
banner=data.banner,
trailer=data.trailer,
year=data.year,
age=data.age,
budget=data.budget,
rating=data.rating,
runtime=data.runtime,
original_lang=data.original_lang,
production_co=data.production_co)
else:
movie.imdb_url = data.imdb_url
movie.title = data.title
movie.avatar = data.avatar
for genre in create_genre(db, data.genres):
movie.genres.append(genre)
for director in create_people(db, data.directors):
movie.directors.append(director)
for writer in create_people(db, data.writers):
movie.writers.append(writer)
for artist in create_people(db, data.artists):
movie.artists.append(artist)
for country in create_country(db, data.countries):
movie.countries.append(country)
for photo in create_photo(db, data.photos):
movie.photos.append(photo)
print(movie.to_dict())
db.session.add(movie)
db.session.commit()
