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.
43 lines
1.7 KiB
43 lines
1.7 KiB
from misc.log import logger
|
|
import requests
|
|
|
|
log = logger.get_logger(__name__)
|
|
|
|
|
|
def validate_movie_tmdb_id(movie_title, movie_year, movie_tmdb_id):
|
|
try:
|
|
if not movie_tmdb_id or not isinstance(movie_tmdb_id, int):
|
|
log.debug("SKIPPING: \'%s (%s)\' blacklisted it has an invalid TMDb ID", movie_title, movie_year)
|
|
return False
|
|
else:
|
|
return True
|
|
except Exception:
|
|
log.exception("Exception validating TMDb ID for \'%s (%s)\'.", movie_title, movie_year)
|
|
return False
|
|
|
|
|
|
def verify_movie_exists_on_tmdb(movie_title, movie_year, movie_tmdb_id):
|
|
try:
|
|
headers = {"User-Agent":"Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/51.0.2704.103 Safari/537.36"}
|
|
req = requests.get('https://www.themoviedb.org/movie/%s' % movie_tmdb_id, headers=headers)
|
|
if req.status_code == 200:
|
|
log.debug("\'%s (%s)\' [TMDb ID: %s] exists on TMDb.", movie_title, movie_year, movie_tmdb_id)
|
|
return True
|
|
else:
|
|
log.debug("SKIPPING: \'%s (%s)\' [TMDb ID: %s] because it does not exist on TMDb.", movie_title, movie_year,
|
|
movie_tmdb_id)
|
|
return False
|
|
except Exception:
|
|
log.exception("Exception verifying TMDb ID for \'%s (%s)\'.", movie_title, movie_year)
|
|
return False
|
|
|
|
|
|
def check_movie_tmdb_id(movie_title, movie_year, movie_tmdb_id):
|
|
try:
|
|
if validate_movie_tmdb_id(movie_title, movie_year, movie_tmdb_id) and \
|
|
verify_movie_exists_on_tmdb(movie_title, movie_year, movie_tmdb_id):
|
|
return True
|
|
except Exception:
|
|
log.exception("Exception verifying/validating TMDb ID for \'%s (%s)\'.", movie_title, movie_year)
|
|
return False
|