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

Sonarr: Get Profile Tags only when needed

Browse Source
pull/105/head
desimaniac 5 years ago
parent f99129ddb1
commit 32d411fcbc
1 changed files with 44 additions and 17 deletions
Show Stats Download Patch File Download Diff File

61
traktarr.py
Unescape View File

@ -161,9 +161,6 @@ def show(show_id, folder=None, no_search=False):
validate_trakt(trakt, False) validate_trakt(trakt, False)
validate_pvr(sonarr, 'Sonarr', False) validate_pvr(sonarr, 'Sonarr', False)
profile_id = get_profile_id(sonarr, cfg.sonarr.profile)
profile_tags = get_profile_tags(sonarr)
# get trakt show # get trakt show
trakt_show = trakt.get_show(show_id) trakt_show = trakt.get_show(show_id)
@ -177,17 +174,34 @@ def show(show_id, folder=None, no_search=False):
log.info("Retrieved Trakt show information for \'%s\': \'%s (%s)\'", show_id, trakt_show['title'], log.info("Retrieved Trakt show information for \'%s\': \'%s (%s)\'", show_id, trakt_show['title'],
series_year) series_year)
# determine which tags to use when adding this series # profile id
use_tags = sonarr_helper.series_tag_id_from_network(profile_tags, cfg.sonarr.tags, trakt_show['network']) profile_id = get_profile_id(sonarr, cfg.sonarr.profile)
# profile tags
profile_tags = None
use_tags = None
readable_tags = None
if cfg.sonarr.tags:
profile_tags = get_profile_tags(sonarr)
# determine which tags to use when adding this series
use_tags = sonarr_helper.series_tag_id_from_network(profile_tags, cfg.sonarr.tags, trakt_show['network'])
readable_tags = sonarr_helper.readable_tag_from_ids(profile_tags, use_tags)
# add show to sonarr # add show to sonarr
if sonarr.add_series(trakt_show['ids']['tvdb'], trakt_show['title'], trakt_show['ids']['slug'], profile_id, if sonarr.add_series(trakt_show['ids']['tvdb'], trakt_show['title'], trakt_show['ids']['slug'], profile_id,
cfg.sonarr.root_folder, use_tags, not no_search): cfg.sonarr.root_folder, use_tags, not no_search):
log.info("ADDED: \'%s (%s)\' with Sonarr Tags: %s", trakt_show['title'], series_year, if profile_tags is not None:
sonarr_helper.readable_tag_from_ids(profile_tags, use_tags)) log.info("ADDED: \'%s (%s)\' with Sonarr Tags: %s", trakt_show['title'], series_year,
readable_tags)
else:
log.info("ADDED: \'%s (%s)\'", trakt_show['title'], series_year)
else: else:
log.error("FAILED ADDING: \'%s (%s)\' with Sonarr Tags: %s", trakt_show['title'], series_year, if profile_tags is not None:
sonarr_helper.readable_tag_from_ids(profile_tags, use_tags)) log.error("FAILED ADDING: \'%s (%s)\' with Sonarr Tags: %s", trakt_show['title'], series_year,
readable_tags)
else:
log.info("FAILED ADDING: \'%s (%s)\'", trakt_show['title'], series_year)
return return
@ -283,7 +297,7 @@ def shows(list_type, add_limit=0, add_delay=2.5, sort='votes', genre=None, folde
validate_pvr(sonarr, 'Sonarr', notifications) validate_pvr(sonarr, 'Sonarr', notifications)
profile_id = get_profile_id(sonarr, cfg.sonarr.profile) profile_id = get_profile_id(sonarr, cfg.sonarr.profile)
profile_tags = get_profile_tags(sonarr) profile_tags = get_profile_tags(sonarr) if cfg.sonarr.tags else None
pvr_objects_list = get_objects(sonarr, 'Sonarr', notifications) pvr_objects_list = get_objects(sonarr, 'Sonarr', notifications)
@ -403,21 +417,34 @@ def shows(list_type, add_limit=0, add_delay=2.5, sort='votes', genre=None, folde
series_genres, series_genres,
series['show']['network']) series['show']['network'])
# determine which tags to use when adding this series # profile tags
use_tags = sonarr_helper.series_tag_id_from_network(profile_tags, cfg.sonarr.tags, use_tags = None
series['show']['network']) readable_tags = None
if profile_tags is not None:
# determine which tags to use when adding this series
use_tags = sonarr_helper.series_tag_id_from_network(profile_tags, cfg.sonarr.tags,
series['show']['network'])
readable_tags = sonarr_helper.readable_tag_from_ids(profile_tags, use_tags)
# add show to sonarr # add show to sonarr
if sonarr.add_series(series['show']['ids']['tvdb'], series['show']['title'], if sonarr.add_series(series['show']['ids']['tvdb'], series['show']['title'],
series['show']['ids']['slug'], profile_id, cfg.sonarr.root_folder, use_tags, series['show']['ids']['slug'], profile_id, cfg.sonarr.root_folder, use_tags,
not no_search): not no_search):
log.info("ADDED: \'%s (%s)\' with tags: %s", series['show']['title'], series_year, if profile_tags is not None:
sonarr_helper.readable_tag_from_ids(profile_tags, use_tags)) log.info("ADDED: \'%s (%s)\' with Sonarr Tags: %s", series['show']['title'], series_year,
readable_tags)
else:
log.info("ADDED: \'%s (%s)\'", series['show']['title'], series_year)
if notifications: if notifications:
callback_notify({'event': 'add_show', 'list_type': list_type, 'show': series['show']}) callback_notify({'event': 'add_show', 'list_type': list_type, 'show': series['show']})
added_shows += 1 added_shows += 1
else: else:
log.error("FAILED ADDING: \'%s (%s)\' with tags: %s", series['show']['title'], series_year, if profile_tags is not None:
sonarr_helper.readable_tag_from_ids(profile_tags, use_tags)) log.error("FAILED ADDING: \'%s (%s)\' with Sonarr Tags: %s", series['show']['title'], series_year,
readable_tags)
else:
log.info("FAILED ADDING: \'%s (%s)\'", series['show']['title'], series_year)
continue continue
else: else:

Write Preview
Loading…
Cancel
Save