mirror of https://github.com/l3uddz/traktarr
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
3.2 KiB
82 lines
3.2 KiB
from misc.log import logger
|
|
import json
|
|
import requests
|
|
|
|
log = logger.get_logger(__name__)
|
|
|
|
|
|
def get_rating(omdb_api_key, movie_title, movie_year, movie_imdb_id):
|
|
"""
|
|
Lookup movie ratings via OMDb
|
|
|
|
:param omdb_api_key: OMDb API Key
|
|
:param movie_title: Movie Title
|
|
:param movie_year: Movie Year
|
|
:param movie_imdb_id: Movie IMDb ID
|
|
:return: Rating score (int) or False if no rating found
|
|
"""
|
|
|
|
ratings_exist = False
|
|
|
|
if movie_imdb_id:
|
|
log.debug("Requesting info from OMDb for: \'%s (%s)\' [IMDb ID: %s]",
|
|
movie_title,
|
|
movie_year,
|
|
movie_imdb_id)
|
|
r = requests.get('http://www.omdbapi.com/?i=' + movie_imdb_id + '&apikey=' + omdb_api_key)
|
|
if r.status_code == 200 and json.loads(r.text)["Response"] == 'True':
|
|
log.debug("Successfully requested ratings from OMDB for \'%s (%s)\' [IMDb ID: %s]",
|
|
movie_title,
|
|
movie_year,
|
|
movie_imdb_id)
|
|
for source in json.loads(r.text)["Ratings"]:
|
|
if source['Source'] == 'Rotten Tomatoes':
|
|
# noinspection PyUnusedLocal
|
|
ratings_exist = True
|
|
log.debug("Rotten Tomatoes score for \'%s (%s)\' [IMDb ID: %s]: %s",
|
|
movie_title,
|
|
movie_year,
|
|
movie_imdb_id,
|
|
source['Value'])
|
|
return int(source['Value'].split('%')[0])
|
|
if not ratings_exist:
|
|
log.debug("No Rotten Tomatoes score found for: \'%s (%s)\' [IMDb ID: %s]",
|
|
movie_title,
|
|
movie_year,
|
|
movie_imdb_id)
|
|
else:
|
|
log.debug("Error encountered when requesting ratings from OMDb for: \'%s (%s)\' [IMDb ID: %s]",
|
|
movie_title,
|
|
movie_year,
|
|
movie_imdb_id)
|
|
else:
|
|
log.debug("Skipping OMDb ratings lookup because no IMDb ID was found for: \'%s (%s)\'",
|
|
movie_title,
|
|
movie_year)
|
|
|
|
return False
|
|
|
|
|
|
def does_movie_have_min_req_rating(omdb_api_key, sorted_movie, req_rating):
|
|
|
|
movie_title = sorted_movie['movie']['title']
|
|
movie_year = str(sorted_movie['movie']['year']) \
|
|
if sorted_movie['movie']['year'] else '????'
|
|
movie_imdb_id = sorted_movie['movie']['ids']['imdb']
|
|
|
|
# pull RT score
|
|
movie_rating = get_rating(omdb_api_key, movie_title, movie_year, movie_imdb_id)
|
|
|
|
if not movie_rating:
|
|
log.info("SKIPPING: \'%s (%s)\' because a Rotten Tomatoes score could not be found.", movie_title,
|
|
movie_year)
|
|
return False
|
|
elif movie_rating < req_rating:
|
|
log.info("SKIPPING: \'%s (%s)\' because its Rotten Tomatoes score of %d%% is below the required score of %d%%.",
|
|
movie_title, movie_year, movie_rating, req_rating)
|
|
return False
|
|
elif movie_rating >= req_rating:
|
|
log.info("ADDING: \'%s (%s)\' because its Rotten Tomatoes score of %d%% is above the required score of %d%%.",
|
|
movie_title, movie_year, movie_rating, req_rating)
|
|
return True
|