load_movie.py

import os
import environ
import requests
import psycopg2
from  datetime import datetime, date, timezone 
import sys


def add_movie(movie_id):
    env = environ.Env()b
    environ.Env.read_env('.env')
    print('API_KEY: ', env('API_KEY'))
    print('API_TOKEN: ', env('API_TOKEN'))

    '''
    url --request GET \
         --url 'https://api.themoviedb.org/3/movie/76341?language=en-US' \
         --header 'Authorization: Aasdfqwer' \
         --header 'accept: application/json'
    '''
    headers = {
        "accept": "application/json",
        "Authorization": f"Bearer {env('API_TOKEN')}"}



    r = requests.get(f'https://api.themoviedb.org/3/movie/{movie_id}?language=en-US', headers=headers) 
    print(r.json())
    m = r.json()

    conn = psycopg2.connect("dbname=django_test user=ubuntu password=password")
    cur = conn.cursor()
    
    '''Verificar si la película ya existe en la base de datos'''
    sql = 'SELECT * FROM movies_movie WHERE title = %s'
    cur.execute(sql, (m['title'],))
    movie_exists = cur.fetchall()

    print(movie_exists)

    r = requests.get(f'https://api.themoviedb.org/3/movie/{movie_id}/credits?language=en-US', headers=headers) 
    credits = r.json()


    actors = [( actor['name'], actor['known_for_department']) for actor in credits['cast'][:10]] 
    crew =   [(   job['name'], job['job']) for job in credits['crew'][:15]]

    credits_list = actors + crew




    jobs = [job for person, job in credits_list]
    jobs = set(jobs)
    print(jobs)

    sql = 'SELECT * FROM movies_job WHERE name IN %s'
    cur.execute(sql, (tuple(jobs),))
    jobs_in_db = cur.fetchall()

    jobs_to_create = [(name,) for name in  jobs if name not in [name for id, name in jobs_in_db]]
    sql = 'INSERT INTO movies_job (name) values  (%s)'
    cur.executemany(sql, jobs_to_create) 


    persons = [person for person, job in credits_list]
    persons = set(persons)
    print(persons)
    sql = 'SELECT * FROM movies_person WHERE name IN %s'
    cur.execute(sql, (tuple(persons),))
    persons_in_db = cur.fetchall()

    persons_to_create = [(name,) for name in  persons if name not in [name for id, name in persons_in_db]]
    sql = 'INSERT INTO movies_person (name) values  (%s)'
    cur.executemany(sql, persons_to_create) 


    genres = [  d['name']  for  d in m['genres']] 
    print(genres)

    sql = 'SELECT * FROM movies_genre WHERE name IN %s'
    cur.execute(sql, (tuple(genres),))
    genres_in_db = cur.fetchall()

    genres_to_create = [(name,) for name in  genres if name not in [name for id, name in genres_in_db]]
    sql = 'INSERT INTO movies_genre (name) values  (%s)'
    cur.executemany(sql, genres_to_create) 



    date_obj = date.fromisoformat(m['release_date']) 
    date_time = datetime.combine(date_obj, datetime.min.time())

    sql = '''INSERT INTO movies_movie 
             (title,
              overview,
              release_date,
              running_time,
              budget,
              tmdb_id,
              revenue,
              poster_path) values  (%s, %s, %s, %s, %s, %s, %s, %s);'''

    movie_tuple = (m['title'], m['overview'], date_time.astimezone(timezone.utc), m['runtime'], 
                   m['budget'] , movie_id, m['revenue'], m['poster_path'] )
    print(movie_tuple)


    sql = '''INSERT INTO movies_movie 
             (title,
              overview,
              release_date,
              running_time,
              budget,
              tmdb_id,
              revenue,
              poster_path) values  (%s, %s, %s, %s, %s, %s, %s, %s);'''

    movie_tuple = (m['title'], m['overview'], date_time.astimezone(timezone.utc), m['runtime'], 
                   m['budget'] , movie_id, m['revenue'], m['poster_path'] )
    print(movie_tuple)
    cur.execute(sql, movie_tuple)

     
    sql = '''INSERT INTO movies_movie_genres (movie_id, genre_id)
             SELECT (SELECT id FROM movies_movie WHERE title = %s) as movie_id, id as genre_id 
             FROM movies_genre 
             WHERE name IN %s'''
    cur.execute(sql, (m['title'], tuple(genres),))


    print(credits_list)
    for credit in credits_list:
        sql = '''INSERT INTO movies_moviecredit (movie_id, person_id, job_id)
                 SELECT id,
                 (SELECT id FROM movies_person WHERE name = %s)  as person_id,
                 (SELECT id FROM movies_job WHERE name = %s)  as job_id
                 FROM movies_movie 
                 WHERE title = %s'''
        cur.execute(sql, (credit[0],credit[1], m['title'],))

    conn.commit()

if __name__ == "__main__":
    add_movie(int(sys.argv[1]))