Plex
/
traktarr
mirror of https://github.com/l3uddz/traktarr
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Helpers: Reorganized trakt functions

Browse Source
pull/105/head
desimaniac 5 years ago
parent f486ae21a3
commit 854de957c5
1 changed files with 98 additions and 98 deletions
Show Stats Download Patch File Download Diff File

196
helpers/trakt.py
Unescape View File

@ -4,29 +4,32 @@ from misc.log import logger
log = logger.get_logger(__name__)
def blacklisted_show_genre(show, genres):
def extract_list_user_and_key_from_url(list_url):
try:
import re
list_user = re.search('\/users\/([^/]*)', list_url).group(1)
list_key = re.search('\/lists\/([^/]*)', list_url).group(1)
return list_user, list_key
except:
log.error('The URL "%s" is not in the correct format', list_url)
exit()
def blacklisted_show_id(show, blacklisted_ids):
blacklisted = False
blacklisted_ids = sorted(map(int, blacklisted_ids))
try:
# ["ignore"] - add show item even if it is missing a genre
if len(genres) == 1 and genres[0].lower() == 'ignore':
log.debug("Skipping valid genre check for: \'%s\'", show['show']['title'])
elif not show['show']['genres']:
log.debug("\'%s\' was blacklisted because it had no genre", show['show']['title'])
if not show['show']['ids']['tvdb'] or not isinstance(show['show']['ids']['tvdb'], int):
log.debug("\'%s\' was blacklisted because it had an invalid TVDB ID", show['show']['title'])
blacklisted = True
elif show['show']['ids']['tvdb'] in blacklisted_ids:
log.debug("\'%s\' was blacklisted because it had a blacklisted TVDB ID of: %d", show['show']['title'],
show['show']['ids']['tvdb'])
blacklisted = True
# [] - add show item with any valid genre
elif not genres:
log.debug("Skipping blacklisted genre check for: \'%s\'", show['show']['title'])
# List provided - skip adding show item if the genre is blacklisted
else:
for genre in genres:
if genre.lower() in show['show']['genres']:
log.debug("\'%s\' was blacklisted because it was from the genre: %s", show['show']['title'],
genre.title())
blacklisted = True
break
except Exception:
log.exception("Exception determining if show has a blacklisted genre %s: ", show)
log.exception("Exception determining if show had a blacklisted TVDB ID %s: ", show)
return blacklisted
@ -46,6 +49,25 @@ def blacklisted_show_year(show, earliest_year, latest_year):
return blacklisted
def blacklisted_show_network(show, networks):
blacklisted = False
try:
if not show['show']['network']:
log.debug("\'%s\' was blacklisted because it had no network", show['show']['title'])
blacklisted = True
else:
for network in networks:
if network.lower() in show['show']['network'].lower():
log.debug("\'%s\' was blacklisted because it's from the network: %s", show['show']['title'],
show['show']['network'])
blacklisted = True
break
except Exception:
log.exception("Exception determining if show is from a blacklisted network %s: ", show)
return blacklisted
def blacklisted_show_country(show, allowed_countries):
blacklisted = False
try:
@ -70,22 +92,29 @@ def blacklisted_show_country(show, allowed_countries):
return blacklisted
def blacklisted_show_network(show, networks):
def blacklisted_show_genre(show, genres):
blacklisted = False
try:
if not show['show']['network']:
log.debug("\'%s\' was blacklisted because it had no network", show['show']['title'])
# ["ignore"] - add show item even if it is missing a genre
if len(genres) == 1 and genres[0].lower() == 'ignore':
log.debug("Skipping valid genre check for: \'%s\'", show['show']['title'])
elif not show['show']['genres']:
log.debug("\'%s\' was blacklisted because it had no genre", show['show']['title'])
blacklisted = True
# [] - add show item with any valid genre
elif not genres:
log.debug("Skipping blacklisted genre check for: \'%s\'", show['show']['title'])
# List provided - skip adding show item if the genre is blacklisted
else:
for network in networks:
if network.lower() in show['show']['network'].lower():
log.debug("\'%s\' was blacklisted because it's from the network: %s", show['show']['title'],
show['show']['network'])
for genre in genres:
if genre.lower() in show['show']['genres']:
log.debug("\'%s\' was blacklisted because it was from the genre: %s", show['show']['title'],
genre.title())
blacklisted = True
break
except Exception:
log.exception("Exception determining if show is from a blacklisted network %s: ", show)
log.exception("Exception determining if show has a blacklisted genre %s: ", show)
return blacklisted
@ -105,42 +134,25 @@ def blacklisted_show_runtime(show, lowest_runtime):
return blacklisted
def blacklisted_show_id(show, blacklisted_ids):
blacklisted = False
blacklisted_ids = sorted(map(int, blacklisted_ids))
try:
if not show['show']['ids']['tvdb'] or not isinstance(show['show']['ids']['tvdb'], int):
log.debug("\'%s\' was blacklisted because it had an invalid TVDB ID", show['show']['title'])
blacklisted = True
elif show['show']['ids']['tvdb'] in blacklisted_ids:
log.debug("\'%s\' was blacklisted because it had a blacklisted TVDB ID of: %d", show['show']['title'],
show['show']['ids']['tvdb'])
blacklisted = True
except Exception:
log.exception("Exception determining if show had a blacklisted TVDB ID %s: ", show)
return blacklisted
def is_show_blacklisted(show, blacklist_settings, ignore_blacklist, callback=None):
if ignore_blacklist:
return False
blacklisted = False
try:
if blacklisted_show_id(show, blacklist_settings.blacklisted_tvdb_ids):
blacklisted = True
if blacklisted_show_year(show, blacklist_settings.blacklisted_min_year,
blacklist_settings.blacklisted_max_year):
blacklisted = True
if blacklisted_show_network(show, blacklist_settings.blacklisted_networks):
blacklisted = True
if blacklisted_show_country(show, blacklist_settings.allowed_countries):
blacklisted = True
if blacklisted_show_genre(show, blacklist_settings.blacklisted_genres):
blacklisted = True
if blacklisted_show_network(show, blacklist_settings.blacklisted_networks):
blacklisted = True
if blacklisted_show_runtime(show, blacklist_settings.blacklisted_min_runtime):
blacklisted = True
if blacklisted_show_id(show, blacklist_settings.blacklisted_tvdb_ids):
blacklisted = True
if blacklisted and callback:
callback('show', show)
@ -150,29 +162,36 @@ def is_show_blacklisted(show, blacklist_settings, ignore_blacklist, callback=Non
return blacklisted
def blacklisted_movie_genre(movie, genres):
def blacklisted_movie_id(movie, blacklisted_ids):
blacklisted = False
blacklisted_ids = sorted(map(int, blacklisted_ids))
try:
# ["ignore"] - add movie item even if it is missing a genre
if len(genres) == 1 and genres[0].lower() == 'ignore':
log.debug("Skipping valid genre check for: \'%s\'", movie['movie']['title'])
elif not movie['movie']['genres']:
log.debug("\'%s\' was blacklisted because it had no genre", movie['movie']['title'])
if movie['movie']['ids']['tmdb'] in blacklisted_ids:
log.debug("\'%s\' was blacklisted because it had a blacklisted TMDb ID of: %d", movie['movie']['title'],
movie['movie']['ids']['tmdb'])
blacklisted = True
except Exception:
log.exception("Exception determining if movie had a blacklisted TMDb ID %s: ", movie)
return blacklisted
def blacklisted_movie_title(movie, blacklisted_keywords):
blacklisted = False
try:
if not movie['movie']['title']:
log.debug("Blacklisted movie because it had no title: %s", movie)
blacklisted = True
# [] - add movie item with any valid genre
elif not genres:
log.debug("Skipping blacklisted genre check for: \'%s\'", movie['movie']['title'])
# List provided - skip adding movie item if the genre is blacklisted
else:
for genre in genres:
if genre.lower() in movie['movie']['genres']:
log.debug("%s was blacklisted because it was from the genre: %s", movie['movie']['title'],
genre.title())
for keyword in blacklisted_keywords:
if keyword.lower() in movie['movie']['title'].lower():
log.debug("\'%s\' was blacklisted because it had title keyword: %s", movie['movie']['title'],
keyword)
blacklisted = True
break
except Exception:
log.exception("Exception determining if movie has a blacklisted genre %s: ", movie)
log.exception("Exception determining if movie had a blacklisted title %s: ", movie)
return blacklisted
@ -215,22 +234,29 @@ def blacklisted_movie_country(movie, allowed_countries):
return blacklisted
def blacklisted_movie_title(movie, blacklisted_keywords):
def blacklisted_movie_genre(movie, genres):
blacklisted = False
try:
if not movie['movie']['title']:
log.debug("Blacklisted movie because it had no title: %s", movie)
# ["ignore"] - add movie item even if it is missing a genre
if len(genres) == 1 and genres[0].lower() == 'ignore':
log.debug("Skipping valid genre check for: \'%s\'", movie['movie']['title'])
elif not movie['movie']['genres']:
log.debug("\'%s\' was blacklisted because it had no genre", movie['movie']['title'])
blacklisted = True
# [] - add movie item with any valid genre
elif not genres:
log.debug("Skipping blacklisted genre check for: \'%s\'", movie['movie']['title'])
# List provided - skip adding movie item if the genre is blacklisted
else:
for keyword in blacklisted_keywords:
if keyword.lower() in movie['movie']['title'].lower():
log.debug("\'%s\' was blacklisted because it had title keyword: %s", movie['movie']['title'],
keyword)
for genre in genres:
if genre.lower() in movie['movie']['genres']:
log.debug("%s was blacklisted because it was from the genre: %s", movie['movie']['title'],
genre.title())
blacklisted = True
break
except Exception:
log.exception("Exception determining if movie had a blacklisted title %s: ", movie)
log.exception("Exception determining if movie has a blacklisted genre %s: ", movie)
return blacklisted
@ -250,26 +276,14 @@ def blacklisted_movie_runtime(movie, lowest_runtime):
return blacklisted
def blacklisted_movie_id(movie, blacklisted_ids):
blacklisted = False
blacklisted_ids = sorted(map(int, blacklisted_ids))
try:
if movie['movie']['ids']['tmdb'] in blacklisted_ids:
log.debug("\'%s\' was blacklisted because it had a blacklisted TMDb ID of: %d", movie['movie']['title'],
movie['movie']['ids']['tmdb'])
blacklisted = True
except Exception:
log.exception("Exception determining if movie had a blacklisted TMDb ID %s: ", movie)
return blacklisted
def is_movie_blacklisted(movie, blacklist_settings, ignore_blacklist, callback=None):
if ignore_blacklist:
return False
blacklisted = False
try:
if blacklisted_movie_id(movie, blacklist_settings.blacklisted_tmdb_ids):
blacklisted = True
if blacklisted_movie_title(movie, blacklist_settings.blacklist_title_keywords):
blacklisted = True
if blacklisted_movie_year(movie, blacklist_settings.blacklisted_min_year,
@ -281,8 +295,6 @@ def is_movie_blacklisted(movie, blacklist_settings, ignore_blacklist, callback=N
blacklisted = True
if blacklisted_movie_runtime(movie, blacklist_settings.blacklisted_min_runtime):
blacklisted = True
if blacklisted_movie_id(movie, blacklist_settings.blacklisted_tmdb_ids):
blacklisted = True
if blacklisted and callback:
callback('movie', movie)
@ -290,15 +302,3 @@ def is_movie_blacklisted(movie, blacklist_settings, ignore_blacklist, callback=N
except Exception:
log.exception("Exception determining if movie was blacklisted %s: ", movie)
return blacklisted
def extract_list_user_and_key_from_url(list_url):
try:
import re
list_user = re.search('\/users\/([^/]*)', list_url).group(1)
list_key = re.search('\/lists\/([^/]*)', list_url).group(1)
return list_user, list_key
except:
log.error('The URL "%s" is not in the correct format', list_url)
exit()

Write Preview
Loading…
Cancel
Save