IMDB Tools

A suite of Python scripts for retrieving data from the IMDB homepage.

• get_directors.py contains functions for retrieving director names, either for an item identified by a given IMDB ID or a list of all directors that appear on the IMDB Top 250. Calling the script from the command line will print the names of all directors that appear on the IMDB Top 250.
• get_imdb_id.py contains functions for returning a list of titles and their associated IMDB IDs given a search query. Calling the script from the command line and passing a search string as parameter will print the first 10 results matching the search string.
• get_top_250.py contains a function for retrieving a list of titles on the IMDB Top 250. Calling the script from the command line will print a CSV-formatted list of entries on the IMDB Top 250.
• ratings_regression.py contains functions for plotting all ratings for a series identified by an IMDB ID together with a linear regression analysis for each season. Calling the script from the command line and passing an IMDB ID as parameter will display a scatter plot with regression lines for each season of the series associated to the IMDB ID.
• get_series_rating.py contains functions for retrieving all episodes accompanied by ratings and vote counts of a series identified by a given IMDB ID. Calling the script from the command line will print a CSV-formatted list of all episodes including ratings associated to the series identified by the given IMDB ID.

The script show_ratings.sh acts as a convenience script to carry out a search query and perform a regression analysis on the first result. The script can be called like so:

bash show_ratings.sh "true detective"

This will open an interactive plot window showing ratings for all three seasons of the series True Detective, with a regression line for each season as well as the series overall. Instead of passing a search string, the script also directly accepts an IMDB ID as parameter:

bash show_ratings.sh tt0944947

Calling the script this way will be faster, since a search for the IMDB ID does not need to be perfomed before fetching episode information.

To ensure all necessary dependencies are installed correctly, run the following pip command:

pip3 install -r requirements.txt