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.
55 lines
1.9 KiB
55 lines
1.9 KiB
from misc.log import logger
|
|
|
|
log = logger.get_logger(__name__)
|
|
|
|
|
|
def movies_to_tmdb_dict(radarr_movies):
|
|
movies = {}
|
|
try:
|
|
for tmp in radarr_movies:
|
|
if 'tmdbId' not in tmp:
|
|
log.debug("Could not handle movie: %s", tmp['title'])
|
|
continue
|
|
movies[tmp['tmdbId']] = tmp
|
|
return movies
|
|
except Exception:
|
|
log.exception("Exception processing Radarr movies to TMDB dict: ")
|
|
return None
|
|
|
|
|
|
def remove_existing_movies(radarr_movies, trakt_movies, callback=None):
|
|
new_movies_list = []
|
|
|
|
if not radarr_movies or not trakt_movies:
|
|
log.error("Inappropriate parameters were supplied")
|
|
return None
|
|
|
|
try:
|
|
# turn radarr movies result into a dict with tmdb id as keys
|
|
processed_movies = movies_to_tmdb_dict(radarr_movies)
|
|
if not processed_movies:
|
|
return None
|
|
|
|
# loop list adding to movies that do not already exist
|
|
for tmp in trakt_movies:
|
|
if 'movie' not in tmp or 'ids' not in tmp['movie'] or 'tmdb' not in tmp['movie']['ids']:
|
|
log.debug("Skipping movie because it did not have required fields: %s", tmp)
|
|
if callback:
|
|
callback('movie', tmp)
|
|
continue
|
|
# check if movie exists in processed_movies
|
|
if tmp['movie']['ids']['tmdb'] in processed_movies:
|
|
log.debug("Removing existing movie: %s", tmp['movie']['title'])
|
|
if callback:
|
|
callback('movie', tmp)
|
|
continue
|
|
|
|
new_movies_list.append(tmp)
|
|
|
|
log.debug("Filtered %d Trakt movies to %d movies that weren't already in Radarr", len(trakt_movies),
|
|
len(new_movies_list))
|
|
return new_movies_list
|
|
except Exception:
|
|
log.exception("Exception removing existing movies from Trakt list: ")
|
|
return None
|