Fill letterboxd exported data using OMDb API

letterboxd.py

import csv
import json
import requests

DIR = '/path/to/files'
API_KEY = 'apikey'
URL = 'http://www.omdbapi.com'

_, TITLE, YEAR, _ = range(4)

def read_files(dir):
    with open(DIR + 'watched.csv') as f:
        reader = csv.reader(f)
        watched = list(reader)

    with open(DIR + 'watchlist.csv') as f:
        reader = csv.reader(f)
        watchlist = list(reader)

    for film in watched:
        film.pop()
        film.append('Watched')

    for film in watchlist:
        film.pop()
        film.append('Watchlist')

    return watched + watchlist

def get_director(film):
    query_params = {'t': film[TITLE], 'y': film[YEAR], 'apikey': API_KEY}
    r = requests.get(URL, params=query_params)
    try:
        return json.loads(r.text)['Director']
    except:
        return ''

def write_csv(films):
    with open('output.csv', 'w') as csvfile:
        writer = csv.writer(csvfile)
        writer.writerow(['Date', 'Title', 'Year', 'Status', 'Director'])
        writer.writerows(films)

films = read_files(DIR)

for film in films:
    film.append(get_director(film))

print(films)
write_csv(films)