2021-02-20 05:41:45 +00:00
import glob , logging , os , re
from datetime import datetime , timedelta
2021-04-05 02:25:41 +00:00
from modules import anidb , anilist , imdb , letterboxd , mal , plex , radarr , sonarr , tautulli , tmdb , trakttv , tvdb , util
2021-02-20 05:41:45 +00:00
from modules . util import Failed
2021-03-01 20:59:10 +00:00
from plexapi . exceptions import BadRequest , NotFound
2021-05-05 18:01:41 +00:00
from urllib . parse import quote
2021-02-20 05:41:45 +00:00
logger = logging . getLogger ( " Plex Meta Manager " )
2021-05-05 18:01:41 +00:00
string_filters = [ " title " , " episode_title " , " studio " ]
2021-03-30 05:50:53 +00:00
image_file_details = [ " file_poster " , " file_background " , " asset_directory " ]
2021-04-21 12:49:27 +00:00
advance_new_agent = [ " item_metadata_language " , " item_use_original_title " ]
advance_show = [ " item_episode_sorting " , " item_keep_episodes " , " item_delete_episodes " , " item_season_display " , " item_episode_sorting " ]
2021-03-30 05:50:53 +00:00
method_alias = {
" actors " : " actor " , " role " : " actor " , " roles " : " actor " ,
" content_ratings " : " content_rating " , " contentRating " : " content_rating " , " contentRatings " : " content_rating " ,
" countries " : " country " ,
" decades " : " decade " ,
" directors " : " director " ,
" genres " : " genre " ,
" labels " : " label " ,
2021-04-01 16:32:41 +00:00
" rating " : " critic_rating " ,
2021-04-01 15:34:02 +00:00
" studios " : " studio " ,
" networks " : " network " ,
2021-03-30 05:50:53 +00:00
" producers " : " producer " ,
" writers " : " writer " ,
" years " : " year "
}
2021-05-05 18:01:41 +00:00
modifier_alias = { " .greater " : " .gt " , " .less " : " .lt " }
2021-03-30 05:50:53 +00:00
all_builders = anidb . builders + anilist . builders + imdb . builders + letterboxd . builders + mal . builders + plex . builders + tautulli . builders + tmdb . builders + trakttv . builders + tvdb . builders
dictionary_builders = [
" filters " ,
" anilist_genre " ,
" anilist_season " ,
" anilist_tag " ,
" mal_season " ,
" mal_userlist " ,
" plex_collectionless " ,
" plex_search " ,
" tautulli_popular " ,
" tautulli_watched " ,
" tmdb_discover "
]
show_only_builders = [
" tmdb_network " ,
" tmdb_show " ,
" tmdb_show_details " ,
" tvdb_show " ,
" tvdb_show_details "
]
movie_only_builders = [
" letterboxd_list " ,
" letterboxd_list_details " ,
" tmdb_collection " ,
" tmdb_collection_details " ,
" tmdb_movie " ,
" tmdb_movie_details " ,
" tmdb_now_playing " ,
" tvdb_movie " ,
" tvdb_movie_details "
]
numbered_builders = [
" anidb_popular " ,
" anilist_popular " ,
" anilist_top_rated " ,
" mal_all " ,
" mal_airing " ,
" mal_upcoming " ,
" mal_tv " ,
" mal_ova " ,
" mal_movie " ,
" mal_special " ,
" mal_popular " ,
" mal_favorite " ,
" mal_suggested " ,
" tmdb_popular " ,
" tmdb_top_rated " ,
" tmdb_now_playing " ,
" tmdb_trending_daily " ,
" tmdb_trending_weekly " ,
" trakt_trending " ,
" trakt_popular " ,
" trakt_recommended " ,
" trakt_watched " ,
" trakt_collected "
]
2021-05-02 04:22:48 +00:00
smart_collection_invalid = [ " collection_mode " , " collection_order " ]
smart_url_collection_invalid = [
" item_label " , " item_label.sync " , " item_episode_sorting " , " item_keep_episodes " , " item_delete_episodes " ,
" item_season_display " , " item_episode_ordering " , " item_metadata_language " , " item_use_original_title " ,
" run_again " , " sync_mode " , " show_filtered " , " show_missing " , " save_missing " , " smart_label " ,
" radarr_add " , " radarr_folder " , " radarr_monitor " , " radarr_availability " ,
" radarr_quality " , " radarr_tag " , " radarr_search " ,
" sonarr_add " , " sonarr_folder " , " sonarr_monitor " , " sonarr_quality " , " sonarr_language " ,
2021-05-05 18:01:41 +00:00
" sonarr_series " , " sonarr_season " , " sonarr_tag " , " sonarr_search " , " sonarr_cutoff_search " ,
" filters "
2021-05-02 04:22:48 +00:00
]
2021-03-30 05:50:53 +00:00
all_details = [
2021-04-05 02:25:41 +00:00
" sort_title " , " content_rating " , " collection_mode " , " collection_order " ,
" summary " , " tmdb_summary " , " tmdb_description " , " tmdb_biography " , " tvdb_summary " ,
" tvdb_description " , " trakt_description " , " letterboxd_description " ,
2021-03-30 05:50:53 +00:00
" url_poster " , " tmdb_poster " , " tmdb_profile " , " tvdb_poster " , " file_poster " ,
" url_background " , " tmdb_background " , " tvdb_background " , " file_background " ,
2021-04-05 02:25:41 +00:00
" name_mapping " , " label " , " show_filtered " , " show_missing " , " save_missing "
2021-03-30 05:50:53 +00:00
]
collectionless_details = [
" sort_title " , " content_rating " ,
" summary " , " tmdb_summary " , " tmdb_description " , " tmdb_biography " ,
" collection_order " , " plex_collectionless " ,
" url_poster " , " tmdb_poster " , " tmdb_profile " , " file_poster " ,
" url_background " , " file_background " ,
" name_mapping " , " label " , " label_sync_mode " , " test "
]
ignored_details = [
2021-05-05 18:01:41 +00:00
" smart_filter " ,
2021-05-02 04:22:48 +00:00
" smart_label " ,
" smart_url " ,
2021-03-30 05:50:53 +00:00
" run_again " ,
" schedule " ,
" sync_mode " ,
" template " ,
" test " ,
" tmdb_person "
]
boolean_details = [
" show_filtered " ,
" show_missing " ,
" save_missing "
]
all_filters = [
" actor " , " actor.not " ,
" audio_language " , " audio_language.not " ,
" audio_track_title " , " audio_track_title.not " ,
" collection " , " collection.not " ,
" content_rating " , " content_rating.not " ,
" country " , " country.not " ,
" director " , " director.not " ,
2021-04-28 18:42:55 +00:00
" filepath " , " filepath.not " ,
2021-03-30 05:50:53 +00:00
" genre " , " genre.not " ,
" max_age " ,
" originally_available.gte " , " originally_available.lte " ,
" tmdb_vote_count.gte " , " tmdb_vote_count.lte " ,
" duration.gte " , " duration.lte " ,
" original_language " , " original_language.not " ,
2021-04-08 03:45:37 +00:00
" user_rating.gte " , " user_rating.lte " ,
2021-04-01 16:32:41 +00:00
" audience_rating.gte " , " audience_rating.lte " ,
" critic_rating.gte " , " critic_rating.lte " ,
2021-03-30 05:50:53 +00:00
" studio " , " studio.not " ,
" subtitle_language " , " subtitle_language.not " ,
" video_resolution " , " video_resolution.not " ,
" writer " , " writer.not " ,
" year " , " year.gte " , " year.lte " , " year.not "
]
movie_only_filters = [
" audio_language " , " audio_language.not " ,
" audio_track_title " , " audio_track_title.not " ,
" country " , " country.not " ,
" director " , " director.not " ,
" duration.gte " , " duration.lte " ,
" original_language " , " original_language.not " ,
" subtitle_language " , " subtitle_language.not " ,
" video_resolution " , " video_resolution.not " ,
" writer " , " writer.not "
]
2021-05-05 18:01:41 +00:00
def split_attribute ( text ) :
attribute , modifier = os . path . splitext ( str ( text ) . lower ( ) )
attribute = method_alias [ attribute ] if attribute in method_alias else attribute
modifier = modifier_alias [ modifier ] if modifier in modifier_alias else modifier
final = f " { attribute } { modifier } "
if text != final :
logger . warning ( f " Collection Warning: { text } plex search attribute will run as { final } " )
return attribute , modifier , final
2021-02-20 05:41:45 +00:00
class CollectionBuilder :
def __init__ ( self , config , library , name , data ) :
self . config = config
self . library = library
self . name = name
self . data = data
2021-02-20 06:41:40 +00:00
self . details = {
" show_filtered " : library . show_filtered ,
" show_missing " : library . show_missing ,
" save_missing " : library . save_missing
}
2021-04-21 12:49:27 +00:00
self . item_details = { }
2021-04-05 02:25:41 +00:00
self . radarr_options = { }
self . sonarr_options = { }
2021-03-01 20:59:10 +00:00
self . missing_movies = [ ]
self . missing_shows = [ ]
2021-02-20 05:41:45 +00:00
self . methods = [ ]
self . filters = [ ]
2021-02-28 23:06:04 +00:00
self . posters = { }
self . backgrounds = { }
self . summaries = { }
2021-02-26 18:15:05 +00:00
self . schedule = " "
2021-03-01 20:59:10 +00:00
self . rating_key_map = { }
2021-04-05 02:25:41 +00:00
self . add_to_radarr = None
self . add_to_sonarr = None
2021-02-25 21:59:56 +00:00
current_time = datetime . now ( )
current_year = current_time . year
2021-02-20 05:41:45 +00:00
2021-03-21 23:00:10 +00:00
methods = { m . lower ( ) : m for m in self . data }
if " template " in methods :
2021-02-20 05:41:45 +00:00
if not self . library . templates :
raise Failed ( " Collection Error: No templates found " )
2021-03-21 23:00:10 +00:00
elif not self . data [ methods [ " template " ] ] :
2021-02-20 05:41:45 +00:00
raise Failed ( " Collection Error: template attribute is blank " )
else :
2021-03-21 23:00:10 +00:00
for data_template in util . get_list ( self . data [ methods [ " template " ] ] , split = False ) :
2021-02-20 05:41:45 +00:00
if not isinstance ( data_template , dict ) :
raise Failed ( " Collection Error: template attribute is not a dictionary " )
elif " name " not in data_template :
raise Failed ( " Collection Error: template sub-attribute name is required " )
elif not data_template [ " name " ] :
raise Failed ( " Collection Error: template sub-attribute name is blank " )
elif data_template [ " name " ] not in self . library . templates :
2021-02-24 06:44:06 +00:00
raise Failed ( f " Collection Error: template { data_template [ ' name ' ] } not found " )
2021-02-20 05:41:45 +00:00
elif not isinstance ( self . library . templates [ data_template [ " name " ] ] , dict ) :
2021-02-24 06:44:06 +00:00
raise Failed ( f " Collection Error: template { data_template [ ' name ' ] } is not a dictionary " )
2021-02-20 05:41:45 +00:00
else :
for tm in data_template :
if not data_template [ tm ] :
2021-03-30 20:50:34 +00:00
raise Failed ( f " Collection Error: template sub-attribute { tm } is blank " )
2021-02-20 05:41:45 +00:00
template_name = data_template [ " name " ]
template = self . library . templates [ template_name ]
2021-03-02 00:30:33 +00:00
2021-02-20 05:41:45 +00:00
default = { }
if " default " in template :
if template [ " default " ] :
if isinstance ( template [ " default " ] , dict ) :
for dv in template [ " default " ] :
2021-03-02 00:30:33 +00:00
if template [ " default " ] [ dv ] :
default [ dv ] = template [ " default " ] [ dv ]
else :
raise Failed ( f " Collection Error: template default sub-attribute { dv } is blank " )
2021-02-20 05:41:45 +00:00
else :
raise Failed ( " Collection Error: template sub-attribute default is not a dictionary " )
else :
raise Failed ( " Collection Error: template sub-attribute default is blank " )
2021-03-02 00:30:33 +00:00
optional = [ ]
if " optional " in template :
if template [ " optional " ] :
if isinstance ( template [ " optional " ] , list ) :
for op in template [ " optional " ] :
if op not in default :
optional . append ( op )
else :
logger . warning ( f " Template Warning: variable { op } cannot be optional if it has a default " )
else :
optional . append ( str ( template [ " optional " ] ) )
else :
raise Failed ( " Collection Error: template sub-attribute optional is blank " )
2021-03-26 22:17:24 +00:00
2021-03-26 15:49:43 +00:00
for method_name , attr_data in template . items ( ) :
2021-03-25 15:07:54 +00:00
if method_name not in self . data and method_name not in [ " default " , " optional " ] :
if attr_data :
2021-02-20 05:41:45 +00:00
def replace_txt ( txt ) :
txt = str ( txt )
2021-03-02 00:30:33 +00:00
for option in optional :
if option not in data_template and f " << { option } >> " in txt :
raise Failed ( " remove attribute " )
2021-02-24 06:42:58 +00:00
for template_method in data_template :
2021-03-30 20:50:34 +00:00
if template_method != " name " and txt == f " << { template_method } >> " :
2021-04-01 20:11:35 +00:00
return data_template [ template_method ]
2021-03-30 20:50:34 +00:00
elif template_method != " name " and f " << { template_method } >> " in txt :
2021-02-24 06:44:06 +00:00
txt = txt . replace ( f " << { template_method } >> " , str ( data_template [ template_method ] ) )
2021-02-20 05:41:45 +00:00
if " <<collection_name>> " in txt :
txt = txt . replace ( " <<collection_name>> " , str ( self . name ) )
for dm in default :
2021-03-30 20:50:34 +00:00
if txt == f " << { dm } >> " :
txt = default [ dm ]
elif f " << { dm } >> " in txt :
2021-02-24 06:44:06 +00:00
txt = txt . replace ( f " << { dm } >> " , str ( default [ dm ] ) )
2021-02-20 05:41:45 +00:00
if txt in [ " true " , " True " ] : return True
elif txt in [ " false " , " False " ] : return False
else :
try : return int ( txt )
2021-03-30 20:50:34 +00:00
except ( ValueError , TypeError ) : return txt
2021-02-28 18:18:21 +00:00
try :
2021-03-25 15:07:54 +00:00
if isinstance ( attr_data , dict ) :
2021-03-26 22:17:24 +00:00
final_data = { }
2021-03-25 15:07:54 +00:00
for sm in attr_data :
if isinstance ( attr_data [ sm ] , list ) :
2021-02-28 18:18:21 +00:00
temp_list = [ ]
2021-03-25 15:07:54 +00:00
for li in attr_data [ sm ] :
2021-02-28 18:18:21 +00:00
temp_list . append ( replace_txt ( li ) )
2021-03-26 22:17:24 +00:00
final_data [ sm ] = temp_list
2021-02-28 18:18:21 +00:00
else :
2021-03-26 22:17:24 +00:00
final_data [ sm ] = replace_txt ( attr_data [ sm ] )
2021-03-25 15:07:54 +00:00
elif isinstance ( attr_data , list ) :
2021-03-26 22:17:24 +00:00
final_data = [ ]
2021-03-25 15:07:54 +00:00
for li in attr_data :
2021-02-28 18:18:21 +00:00
if isinstance ( li , dict ) :
temp_dict = { }
for sm in li :
temp_dict [ sm ] = replace_txt ( li [ sm ] )
2021-03-26 22:17:24 +00:00
final_data . append ( temp_dict )
2021-02-28 18:18:21 +00:00
else :
2021-03-26 22:17:24 +00:00
final_data . append ( replace_txt ( li ) )
2021-02-28 18:18:21 +00:00
else :
2021-03-26 22:17:24 +00:00
final_data = replace_txt ( attr_data )
2021-02-28 18:18:21 +00:00
except Failed :
continue
2021-03-26 22:17:24 +00:00
self . data [ method_name ] = final_data
methods [ method_name . lower ( ) ] = method_name
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: template attribute { method_name } is blank " )
2021-02-20 05:41:45 +00:00
skip_collection = True
2021-03-21 23:00:10 +00:00
if " schedule " not in methods :
2021-02-20 05:41:45 +00:00
skip_collection = False
2021-03-21 23:00:10 +00:00
elif not self . data [ methods [ " schedule " ] ] :
2021-02-20 05:41:45 +00:00
logger . error ( " Collection Error: schedule attribute is blank. Running daily " )
skip_collection = False
else :
2021-03-21 23:00:10 +00:00
schedule_list = util . get_list ( self . data [ methods [ " schedule " ] ] )
2021-02-20 05:41:45 +00:00
next_month = current_time . replace ( day = 28 ) + timedelta ( days = 4 )
last_day = next_month - timedelta ( days = next_month . day )
for schedule in schedule_list :
run_time = str ( schedule ) . lower ( )
if run_time . startswith ( " day " ) or run_time . startswith ( " daily " ) :
skip_collection = False
2021-04-03 20:48:27 +00:00
elif run_time . startswith ( " week " ) or run_time . startswith ( " month " ) or run_time . startswith ( " year " ) :
2021-02-20 05:41:45 +00:00
match = re . search ( " \\ (([^)]+) \\ ) " , run_time )
if match :
param = match . group ( 1 )
if run_time . startswith ( " week " ) :
if param . lower ( ) in util . days_alias :
weekday = util . days_alias [ param . lower ( ) ]
2021-02-24 06:44:06 +00:00
self . schedule + = f " \n Scheduled weekly on { util . pretty_days [ weekday ] } "
2021-02-20 05:41:45 +00:00
if weekday == current_time . weekday ( ) :
skip_collection = False
else :
2021-02-24 06:44:06 +00:00
logger . error ( f " Collection Error: weekly schedule attribute { schedule } invalid must be a day of the week i.e. weekly(Monday) " )
2021-02-20 05:41:45 +00:00
elif run_time . startswith ( " month " ) :
try :
if 1 < = int ( param ) < = 31 :
2021-02-24 06:44:06 +00:00
self . schedule + = f " \n Scheduled monthly on the { util . make_ordinal ( param ) } "
2021-02-20 05:41:45 +00:00
if current_time . day == int ( param ) or ( current_time . day == last_day . day and int ( param ) > last_day . day ) :
skip_collection = False
else :
2021-02-24 06:44:06 +00:00
logger . error ( f " Collection Error: monthly schedule attribute { schedule } invalid must be between 1 and 31 " )
2021-02-20 05:41:45 +00:00
except ValueError :
2021-02-24 06:44:06 +00:00
logger . error ( f " Collection Error: monthly schedule attribute { schedule } invalid must be an integer " )
2021-02-20 05:41:45 +00:00
elif run_time . startswith ( " year " ) :
match = re . match ( " ^(1[0-2]|0?[1-9])/(3[01]|[12][0-9]|0?[1-9])$ " , param )
if match :
month = int ( match . group ( 1 ) )
day = int ( match . group ( 2 ) )
2021-02-24 06:44:06 +00:00
self . schedule + = f " \n Scheduled yearly on { util . pretty_months [ month ] } { util . make_ordinal ( day ) } "
2021-02-20 05:41:45 +00:00
if current_time . month == month and ( current_time . day == day or ( current_time . day == last_day . day and day > last_day . day ) ) :
skip_collection = False
else :
2021-02-24 06:44:06 +00:00
logger . error ( f " Collection Error: yearly schedule attribute { schedule } invalid must be in the MM/DD format i.e. yearly(11/22) " )
2021-02-20 05:41:45 +00:00
else :
2021-02-24 06:44:06 +00:00
logger . error ( f " Collection Error: failed to parse schedule: { schedule } " )
2021-02-20 05:41:45 +00:00
else :
2021-02-24 06:44:06 +00:00
logger . error ( f " Collection Error: schedule attribute { schedule } invalid " )
2021-02-26 19:55:02 +00:00
if len ( self . schedule ) == 0 :
2021-02-20 05:41:45 +00:00
skip_collection = False
if skip_collection :
2021-02-26 19:55:02 +00:00
raise Failed ( f " { self . schedule } \n \n Collection { self . name } not scheduled to run " )
2021-02-20 05:41:45 +00:00
2021-02-24 06:44:06 +00:00
logger . info ( f " Scanning { self . name } Collection " )
2021-02-20 05:41:45 +00:00
2021-03-21 23:00:10 +00:00
self . run_again = " run_again " in methods
2021-05-02 04:22:48 +00:00
self . collectionless = " plex_collectionless " in methods
2021-05-05 18:01:41 +00:00
if " tmdb_person " in methods :
if self . data [ methods [ " tmdb_person " ] ] :
valid_names = [ ]
for tmdb_id in util . get_int_list ( self . data [ methods [ " tmdb_person " ] ] , " TMDb Person ID " ) :
person = config . TMDb . get_person ( tmdb_id )
valid_names . append ( person . name )
if hasattr ( person , " biography " ) and person . biography :
self . summaries [ " tmdb_person " ] = person . biography
if hasattr ( person , " profile_path " ) and person . profile_path :
self . posters [ " tmdb_person " ] = f " { config . TMDb . image_url } { person . profile_path } "
if len ( valid_names ) > 0 :
self . details [ " tmdb_person " ] = valid_names
else :
raise Failed ( f " Collection Error: No valid TMDb Person IDs in { self . data [ methods [ ' tmdb_person ' ] ] } " )
else :
raise Failed ( " Collection Error: tmdb_person attribute is blank " )
self . smart_sort = " random "
2021-05-02 04:22:48 +00:00
self . smart_label_collection = False
if " smart_label " in methods :
self . smart_label_collection = True
if self . data [ methods [ " smart_label " ] ] :
2021-05-05 18:01:41 +00:00
if ( self . library . is_movie and str ( self . data [ methods [ " smart_label " ] ] ) . lower ( ) in plex . movie_smart_sorts ) \
or ( self . library . is_show and str ( self . data [ methods [ " smart_label " ] ] ) . lower ( ) in plex . show_smart_sorts ) :
2021-05-02 04:22:48 +00:00
self . smart_sort = str ( self . data [ methods [ " smart_label " ] ] ) . lower ( )
else :
logger . info ( " " )
2021-05-05 18:01:41 +00:00
logger . warning ( f " Collection Error: smart_label attribute: { self . data [ methods [ ' smart_label ' ] ] } is invalid defaulting to random " )
2021-05-02 04:22:48 +00:00
else :
logger . info ( " " )
2021-05-05 18:01:41 +00:00
logger . warning ( " Collection Error: smart_label attribute is blank defaulting to random " )
2021-05-02 04:22:48 +00:00
self . smart_url = None
2021-05-05 18:01:41 +00:00
self . smart_type_key = None
2021-05-02 04:22:48 +00:00
if " smart_url " in methods :
if self . data [ methods [ " smart_url " ] ] :
try :
2021-05-05 18:01:41 +00:00
self . smart_url , self . smart_type_key = library . get_smart_filter_from_uri ( self . data [ methods [ " smart_url " ] ] )
2021-05-02 04:22:48 +00:00
except ValueError :
raise Failed ( " Collection Error: smart_url is incorrectly formatted " )
else :
raise Failed ( " Collection Error: smart_url attribute is blank " )
2021-05-05 18:01:41 +00:00
if " smart_filter " in methods :
logger . info ( " " )
smart_filter = self . data [ methods [ " smart_filter " ] ]
if smart_filter is None :
raise Failed ( f " Collection Error: smart_filter attribute is blank " )
if not isinstance ( smart_filter , dict ) :
raise Failed ( f " Collection Error: smart_filter must be a dictionary: { smart_filter } " )
smart_methods = { m . lower ( ) : m for m in smart_filter }
if " any " in smart_methods and " all " in smart_methods :
raise Failed ( f " Collection Error: Cannot have more then one base " )
if " any " not in smart_methods and " all " not in smart_methods :
raise Failed ( f " Collection Error: Must have either any or all as a base for the filter " )
2021-05-02 04:22:48 +00:00
2021-05-05 18:01:41 +00:00
if " type " in smart_methods and self . library . is_show :
if smart_filter [ smart_methods [ " type " ] ] not in [ " shows " , " seasons " , " episodes " ] :
raise Failed ( f " Collection Error: type: { smart_filter [ smart_methods [ ' type ' ] ] } is invalid, must be either shows, season, or episodes " )
smart_type = smart_filter [ smart_methods [ " type " ] ]
elif self . library . is_show :
smart_type = " shows "
else :
smart_type = " movies "
logger . info ( f " Smart { smart_type . capitalize ( ) [ : - 1 ] } Filter " )
self . smart_type_key , smart_sorts = plex . smart_types [ smart_type ]
2021-05-02 04:22:48 +00:00
2021-05-05 18:01:41 +00:00
smart_sort = " random "
if " sort_by " in smart_methods :
if smart_filter [ smart_methods [ " sort_by " ] ] is None :
raise Failed ( f " Collection Error: sort_by attribute is blank " )
if smart_filter [ smart_methods [ " sort_by " ] ] not in smart_sorts :
raise Failed ( f " Collection Error: sort_by: { smart_filter [ smart_methods [ ' sort_by ' ] ] } is invalid " )
smart_sort = smart_filter [ smart_methods [ " sort_by " ] ]
logger . info ( f " Sort By: { smart_sort } " )
2021-02-20 05:41:45 +00:00
2021-05-05 18:01:41 +00:00
limit = None
if " limit " in smart_methods :
if smart_filter [ smart_methods [ " limit " ] ] is None :
raise Failed ( " Collection Error: limit attribute is blank " )
if not isinstance ( smart_filter [ smart_methods [ " limit " ] ] , int ) or smart_filter [ smart_methods [ " limit " ] ] < 1 :
raise Failed ( " Collection Error: limit attribute must be an integer greater then 0 " )
limit = smart_filter [ smart_methods [ " limit " ] ]
logger . info ( f " Limit: { limit } " )
def _filter ( filter_dict , is_all = True , level = 1 ) :
output = " "
display = f " \n { ' ' * level } Match { ' all ' if is_all else ' any ' } of the following: "
level + = 1
indent = f " \n { ' ' * level } "
conjunction = f " { ' and ' if is_all else ' or ' } =1& "
for smart_key , smart_data in filter_dict . items ( ) :
smart , smart_mod , smart_final = split_attribute ( smart_key )
def build_url_arg ( arg , mod = None , arg_s = None , mod_s = None ) :
arg_key = plex . search_translation [ smart ] if smart in plex . search_translation else smart
if mod is None :
mod = plex . modifier_translation [ smart_mod ] if smart_mod in plex . search_translation else smart_mod
if arg_s is None :
arg_s = arg
if smart in string_filters and smart_mod in [ " " , " .not " ] :
mod_s = " does not contain " if smart_mod == " .not " else " contains "
elif mod_s is None :
mod_s = plex . mod_displays [ smart_mod ]
display_line = f " { indent } { smart . title ( ) . replace ( ' _ ' , ' ' ) } { mod_s } { arg_s } "
return f " { arg_key } { mod } = { arg } & " , display_line
if smart_final in plex . movie_only_smart_searches and self . library . is_show :
raise Failed ( f " Collection Error: { smart_final } smart filter attribute only works for movie libraries " )
elif smart_final in plex . show_only_smart_searches and self . library . is_movie :
raise Failed ( f " Collection Error: { smart_final } smart filter attribute only works for show libraries " )
elif smart_final not in plex . smart_searches :
raise Failed ( f " Collection Error: { smart_final } is not a valid smart filter attribute " )
elif smart_data is None :
raise Failed ( f " Collection Error: { smart_final } smart filter attribute is blank " )
elif smart in [ " all " , " any " ] :
dicts = util . get_list ( smart_data )
results = " "
display_add = " "
for dict_data in dicts :
if not isinstance ( dict_data , dict ) :
raise Failed ( f " Collection Error: { smart } must be either a dictionary or list of dictionaries " )
inside_filter , inside_display = _filter ( dict_data , is_all = smart == " all " , level = level )
display_add + = inside_display
results + = f " { conjunction if len ( results ) > 0 else ' ' } push=1& { inside_filter } pop=1& "
elif smart in [ " year " , " episode_year " ] and smart_mod in [ " .gt " , " .gte " , " .lt " , " .lte " ] :
results , display_add = build_url_arg ( util . check_year ( smart_data , current_year , smart_final ) )
elif smart in [ " added " , " episode_added " , " originally_available " , " episode_originally_available " ] and smart_mod in [ " .before " , " .after " ] :
results , display_add = build_url_arg ( util . check_date ( smart_data , smart_final , return_string = True , plex_date = True ) )
elif smart in [ " added " , " episode_added " , " originally_available " , " episode_originally_available " ] and smart_mod in [ " " , " .not " ] :
in_the_last = util . check_number ( smart_data , smart_final , minimum = 1 )
last_text = " is not in the last " if smart_mod == " .not " else " is in the last "
last_mod = " %3E %3E " if smart_mod == " " else " % 3C % 3C "
results , display_add = build_url_arg ( f " - { in_the_last } d " , mod = last_mod , arg_s = f " { in_the_last } Days " , mod_s = last_text )
elif smart in [ " duration " ] and smart_mod in [ " .gt " , " .gte " , " .lt " , " .lte " ] :
results , display_add = build_url_arg ( util . check_number ( smart_data , smart_final , minimum = 1 ) * 60000 )
elif smart in [ " plays " , " episode_plays " ] and smart_mod in [ " .gt " , " .gte " , " .lt " , " .lte " ] :
results , display_add = build_url_arg ( util . check_number ( smart_data , smart_final , minimum = 1 ) )
elif smart in [ " user_rating " , " episode_user_rating " , " critic_rating " , " audience_rating " ] and smart_mod in [ " .gt " , " .gte " , " .lt " , " .lte " ] :
results , display_add = build_url_arg ( util . check_number ( smart_data , smart_final , number_type = " float " , minimum = 0 , maximum = 10 ) )
else :
if smart in [ " title " , " episode_title " ] and smart_mod in [ " " , " .not " , " .begins " , " .ends " ] :
results_list = [ ( t , t ) for t in util . get_list ( smart_data , split = False ) ]
elif smart in plex . tags and smart_mod in [ " " , " .not " , " .begins " , " .ends " ] :
if smart_final in plex . tmdb_searches :
final_tmdb_values = [ ]
for tmdb_value in util . get_list ( smart_data ) :
if tmdb_value . lower ( ) == " tmdb " and " tmdb_person " in self . details :
for tmdb_name in self . details [ " tmdb_person " ] :
final_tmdb_values . append ( tmdb_name )
else :
final_tmdb_values . append ( tmdb_value )
elif smart == " studio " :
final_tmdb_values = util . get_list ( smart_data , split = False )
else :
final_tmdb_values = util . get_list ( smart_data )
results_list = self . library . validate_search_list ( final_tmdb_values , smart , fail = True , title = False , pairs = True )
elif smart in [ " decade " , " year " , " episode_year " ] and smart_mod in [ " " , " .not " ] :
results_list = [ ( y , y ) for y in util . get_year_list ( smart_data , current_year , smart_final ) ]
else :
raise Failed ( f " Collection Error: modifier: { smart_mod } not supported with the { smart } plex search attribute " )
results = " "
display_add = " "
for og_value , result in results_list :
built_arg = build_url_arg ( quote ( result ) if smart in string_filters else result , arg_s = og_value )
display_add + = built_arg [ 1 ]
results + = f " { conjunction if len ( results ) > 0 else ' ' } { built_arg [ 0 ] } "
display + = display_add
output + = f " { conjunction if len ( output ) > 0 else ' ' } { results } "
return output , display
base = " all " if " all " in smart_methods else " any "
base_all = base == " all "
if smart_filter [ smart_methods [ base ] ] is None :
raise Failed ( f " Collection Error: { base } attribute is blank " )
if not isinstance ( smart_filter [ smart_methods [ base ] ] , dict ) :
raise Failed ( f " Collection Error: { base } must be a dictionary: { smart_filter [ smart_methods [ base ] ] } " )
built_filter , filter_text = _filter ( smart_filter [ smart_methods [ base ] ] , is_all = base_all )
util . print_multiline ( f " Filter: { filter_text } " )
final_filter = built_filter [ : - 1 ] if base_all else f " push=1& { built_filter } pop=1 "
self . smart_url = f " ?type= { self . smart_type_key } & { f ' limit= { limit } & ' if limit else ' ' } sort= { smart_sorts [ smart_sort ] } & { final_filter } "
def cant_interact ( attr1 , attr2 , fail = False ) :
if getattr ( self , attr1 ) and getattr ( self , attr2 ) :
message = f " Collection Error: { attr1 } & { attr2 } attributes cannot go together "
if fail :
raise Failed ( message )
2021-05-02 04:22:48 +00:00
else :
2021-05-05 18:01:41 +00:00
setattr ( self , attr2 , False )
logger . info ( " " )
logger . warning ( f " { message } removing { attr2 } " )
cant_interact ( " smart_label_collection " , " collectionless " )
cant_interact ( " smart_url " , " collectionless " )
cant_interact ( " smart_url " , " run_again " )
cant_interact ( " smart_label_collection " , " smart_url " , fail = True )
self . smart = self . smart_url or self . smart_label_collection
2021-02-20 05:41:45 +00:00
2021-04-05 02:25:41 +00:00
for method_key , method_data in self . data . items ( ) :
if " trakt " in method_key . lower ( ) and not config . Trakt : raise Failed ( f " Collection Error: { method_key } requires Trakt todo be configured " )
elif " imdb " in method_key . lower ( ) and not config . IMDb : raise Failed ( f " Collection Error: { method_key } requires TMDb or Trakt to be configured " )
elif " radarr " in method_key . lower ( ) and not self . library . Radarr : raise Failed ( f " Collection Error: { method_key } requires Radarr to be configured " )
elif " sonarr " in method_key . lower ( ) and not self . library . Sonarr : raise Failed ( f " Collection Error: { method_key } requires Sonarr to be configured " )
elif " tautulli " in method_key . lower ( ) and not self . library . Tautulli : raise Failed ( f " Collection Error: { method_key } requires Tautulli to be configured " )
elif " mal " in method_key . lower ( ) and not config . MyAnimeList : raise Failed ( f " Collection Error: { method_key } requires MyAnimeList to be configured " )
2021-03-25 15:07:54 +00:00
elif method_data is not None :
2021-02-20 05:41:45 +00:00
logger . debug ( " " )
2021-04-05 02:25:41 +00:00
logger . debug ( f " Validating Method: { method_key } " )
2021-03-25 15:07:54 +00:00
logger . debug ( f " Value: { method_data } " )
2021-04-05 02:25:41 +00:00
if method_key . lower ( ) in method_alias :
method_name = method_alias [ method_key . lower ( ) ]
logger . warning ( f " Collection Warning: { method_key } attribute will run as { method_name } " )
elif method_key . lower ( ) == " add_to_arr " :
method_name = " radarr_add " if self . library . is_movie else " sonarr_add "
logger . warning ( f " Collection Warning: { method_key } attribute will run as { method_name } " )
elif method_key . lower ( ) in [ " arr_tag " , " arr_folder " ] :
method_name = f " { ' rad ' if self . library . is_movie else ' son ' } { method_key . lower ( ) } "
logger . warning ( f " Collection Warning: { method_key } attribute will run as { method_name } " )
2021-02-20 05:41:45 +00:00
else :
2021-04-05 02:25:41 +00:00
method_name = method_key . lower ( )
2021-03-30 05:50:53 +00:00
if method_name in show_only_builders and self . library . is_movie :
2021-02-24 06:44:06 +00:00
raise Failed ( f " Collection Error: { method_name } attribute only works for show libraries " )
2021-03-30 05:50:53 +00:00
elif method_name in movie_only_builders and self . library . is_show :
2021-02-24 06:44:06 +00:00
raise Failed ( f " Collection Error: { method_name } attribute only works for movie libraries " )
2021-03-30 05:50:53 +00:00
elif method_name in plex . movie_only_searches and self . library . is_show :
2021-02-24 06:44:06 +00:00
raise Failed ( f " Collection Error: { method_name } plex search only works for movie libraries " )
2021-04-01 15:34:02 +00:00
elif method_name in plex . show_only_searches and self . library . is_movie :
raise Failed ( f " Collection Error: { method_name } plex search only works for show libraries " )
2021-05-02 04:22:48 +00:00
elif method_name in smart_collection_invalid and self . smart :
raise Failed ( f " Collection Error: { method_name } attribute only works with normal collections " )
2021-03-30 05:50:53 +00:00
elif method_name not in collectionless_details and self . collectionless :
2021-02-24 06:44:06 +00:00
raise Failed ( f " Collection Error: { method_name } attribute does not work for Collectionless collection " )
2021-05-05 18:01:41 +00:00
elif self . smart_url and method_name in all_builders :
2021-05-02 04:22:48 +00:00
raise Failed ( f " Collection Error: { method_name } builder not allowed when using smart_url " )
2021-05-05 18:01:41 +00:00
elif self . smart_url and method_name in smart_url_collection_invalid :
2021-05-02 04:22:48 +00:00
raise Failed ( f " Collection Error: { method_name } detail not allowed when using smart_url " )
2021-02-28 23:06:04 +00:00
elif method_name == " summary " :
2021-03-25 15:07:54 +00:00
self . summaries [ method_name ] = method_data
2021-02-20 05:41:45 +00:00
elif method_name == " tmdb_summary " :
2021-03-25 15:07:54 +00:00
self . summaries [ method_name ] = config . TMDb . get_movie_show_or_collection ( util . regex_first_int ( method_data , " TMDb ID " ) , self . library . is_movie ) . overview
2021-02-20 05:41:45 +00:00
elif method_name == " tmdb_description " :
2021-03-25 15:07:54 +00:00
self . summaries [ method_name ] = config . TMDb . get_list ( util . regex_first_int ( method_data , " TMDb List ID " ) ) . description
2021-02-20 05:41:45 +00:00
elif method_name == " tmdb_biography " :
2021-03-25 15:07:54 +00:00
self . summaries [ method_name ] = config . TMDb . get_person ( util . regex_first_int ( method_data , " TMDb Person ID " ) ) . biography
2021-03-08 19:53:05 +00:00
elif method_name == " tvdb_summary " :
2021-03-25 15:07:54 +00:00
self . summaries [ method_name ] = config . TVDb . get_movie_or_show ( method_data , self . library . Plex . language , self . library . is_movie ) . summary
2021-03-08 19:53:05 +00:00
elif method_name == " tvdb_description " :
2021-03-25 15:07:54 +00:00
self . summaries [ method_name ] = config . TVDb . get_list_description ( method_data , self . library . Plex . language )
2021-03-08 19:53:05 +00:00
elif method_name == " trakt_description " :
2021-03-27 07:30:07 +00:00
self . summaries [ method_name ] = config . Trakt . standard_list ( config . Trakt . validate_trakt ( util . get_list ( method_data ) ) [ 0 ] ) . description
2021-03-08 21:54:19 +00:00
elif method_name == " letterboxd_description " :
2021-03-25 15:07:54 +00:00
self . summaries [ method_name ] = config . Letterboxd . get_list_description ( method_data , self . library . Plex . language )
2021-02-20 05:41:45 +00:00
elif method_name == " collection_mode " :
2021-03-25 15:07:54 +00:00
if str ( method_data ) . lower ( ) == " default " :
2021-03-21 23:00:10 +00:00
self . details [ method_name ] = " default "
2021-03-25 15:07:54 +00:00
elif str ( method_data ) . lower ( ) == " hide " :
2021-03-21 23:00:10 +00:00
self . details [ method_name ] = " hide "
2021-03-25 15:07:54 +00:00
elif str ( method_data ) . lower ( ) in [ " hide_items " , " hideitems " ] :
2021-03-21 23:00:10 +00:00
self . details [ method_name ] = " hideItems "
2021-03-25 15:07:54 +00:00
elif str ( method_data ) . lower ( ) in [ " show_items " , " showitems " ] :
2021-03-21 23:00:10 +00:00
self . details [ method_name ] = " showItems "
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_data } collection_mode invalid \n \t default (Library default) \n \t hide (Hide Collection) \n \t hide_items (Hide Items in this Collection) \n \t show_items (Show this Collection and its Items) " )
2021-02-20 05:41:45 +00:00
elif method_name == " collection_order " :
2021-03-25 15:07:54 +00:00
if str ( method_data ) . lower ( ) == " release " :
2021-03-21 23:00:10 +00:00
self . details [ method_name ] = " release "
2021-03-25 15:07:54 +00:00
elif str ( method_data ) . lower ( ) == " alpha " :
2021-03-27 03:36:57 +00:00
self . details [ method_name ] = " alpha "
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_data } collection_order invalid \n \t release (Order Collection by release dates) \n \t alpha (Order Collection Alphabetically) " )
2021-02-20 05:41:45 +00:00
elif method_name == " url_poster " :
2021-03-25 15:07:54 +00:00
self . posters [ method_name ] = method_data
2021-02-20 05:41:45 +00:00
elif method_name == " tmdb_poster " :
2021-03-25 15:07:54 +00:00
self . posters [ method_name ] = f " { config . TMDb . image_url } { config . TMDb . get_movie_show_or_collection ( util . regex_first_int ( method_data , ' TMDb ID ' ) , self . library . is_movie ) . poster_path } "
2021-02-20 05:41:45 +00:00
elif method_name == " tmdb_profile " :
2021-03-25 15:07:54 +00:00
self . posters [ method_name ] = f " { config . TMDb . image_url } { config . TMDb . get_person ( util . regex_first_int ( method_data , ' TMDb Person ID ' ) ) . profile_path } "
2021-03-08 19:53:05 +00:00
elif method_name == " tvdb_poster " :
2021-03-25 15:07:54 +00:00
self . posters [ method_name ] = f " { config . TVDb . get_movie_or_series ( method_data , self . library . Plex . language , self . library . is_movie ) . poster_path } "
2021-02-20 05:41:45 +00:00
elif method_name == " file_poster " :
2021-03-25 15:07:54 +00:00
if os . path . exists ( method_data ) :
self . posters [ method_name ] = os . path . abspath ( method_data )
2021-03-21 23:00:10 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: Poster Path Does Not Exist: { os . path . abspath ( method_data ) } " )
2021-02-20 05:41:45 +00:00
elif method_name == " url_background " :
2021-03-25 15:07:54 +00:00
self . backgrounds [ method_name ] = method_data
2021-02-20 05:41:45 +00:00
elif method_name == " tmdb_background " :
2021-03-25 15:07:54 +00:00
self . backgrounds [ method_name ] = f " { config . TMDb . image_url } { config . TMDb . get_movie_show_or_collection ( util . regex_first_int ( method_data , ' TMDb ID ' ) , self . library . is_movie ) . poster_path } "
2021-03-08 19:53:05 +00:00
elif method_name == " tvdb_background " :
2021-03-25 15:07:54 +00:00
self . posters [ method_name ] = f " { config . TVDb . get_movie_or_series ( method_data , self . library . Plex . language , self . library . is_movie ) . background_path } "
2021-02-20 05:41:45 +00:00
elif method_name == " file_background " :
2021-03-25 15:07:54 +00:00
if os . path . exists ( method_data ) : self . backgrounds [ method_name ] = os . path . abspath ( method_data )
else : raise Failed ( f " Collection Error: Background Path Does Not Exist: { os . path . abspath ( method_data ) } " )
2021-03-01 20:59:10 +00:00
elif method_name == " sync_mode " :
2021-03-25 15:07:54 +00:00
if str ( method_data ) . lower ( ) in [ " append " , " sync " ] : self . details [ method_name ] = method_data . lower ( )
2021-03-01 20:59:10 +00:00
else : raise Failed ( " Collection Error: sync_mode attribute must be either ' append ' or ' sync ' " )
2021-04-12 04:01:01 +00:00
elif method_name in [ " label " , " label.sync " ] :
if " label " in self . data and " label.sync " in self . data :
raise Failed ( f " Collection Error: Cannot use label and label.sync together " )
if method_name == " label " and " label_sync_mode " in self . data and self . data [ " label_sync_mode " ] == " sync " :
self . details [ " label.sync " ] = util . get_list ( method_data )
else :
self . details [ method_name ] = util . get_list ( method_data )
2021-04-21 12:49:27 +00:00
elif method_name in [ " item_label " , " item_label.sync " ] :
if " item_label " in self . data and " item_label.sync " in self . data :
raise Failed ( f " Collection Error: Cannot use item_label and item_label.sync together " )
self . item_details [ method_name ] = util . get_list ( method_data )
elif method_name in plex . item_advance_keys :
key , options = plex . item_advance_keys [ method_name ]
if method_name in advance_new_agent and self . library . agent not in plex . new_plex_agents :
logger . error ( f " Metadata Error: { method_name } attribute only works for with the New Plex Movie Agent and New Plex TV Agent " )
elif method_name in advance_show and not self . library . is_show :
logger . error ( f " Metadata Error: { method_name } attribute only works for show libraries " )
elif str ( method_data ) . lower ( ) not in options :
logger . error ( f " Metadata Error: { method_data } { method_name } attribute invalid " )
else :
2021-05-04 15:56:30 +00:00
self . item_details [ method_name ] = options [ str ( method_data ) . lower ( ) ]
2021-03-30 05:50:53 +00:00
elif method_name in boolean_details :
2021-04-11 15:31:59 +00:00
self . details [ method_name ] = util . get_bool ( method_name , method_data )
2021-03-30 05:50:53 +00:00
elif method_name in all_details :
2021-03-25 15:07:54 +00:00
self . details [ method_name ] = method_data
2021-04-05 02:25:41 +00:00
elif method_name == " radarr_add " :
2021-04-11 15:31:59 +00:00
self . add_to_radarr = util . get_bool ( method_name , method_data )
2021-04-05 02:25:41 +00:00
elif method_name == " radarr_folder " :
self . radarr_options [ " folder " ] = method_data
elif method_name in [ " radarr_monitor " , " radarr_search " ] :
2021-04-11 15:31:59 +00:00
self . radarr_options [ method_name [ 7 : ] ] = util . get_bool ( method_name , method_data )
2021-04-05 02:25:41 +00:00
elif method_name == " radarr_availability " :
if str ( method_data ) . lower ( ) in radarr . availability_translation :
self . radarr_options [ " availability " ] = str ( method_data ) . lower ( )
else :
raise Failed ( f " Collection Error: { method_name } attribute must be either announced, cinemas, released or db " )
elif method_name == " radarr_quality " :
self . library . Radarr . get_profile_id ( method_data )
self . radarr_options [ " quality " ] = method_data
elif method_name == " radarr_tag " :
self . radarr_options [ " tag " ] = util . get_list ( method_data )
elif method_name == " sonarr_add " :
2021-04-11 15:31:59 +00:00
self . add_to_sonarr = util . get_bool ( method_name , method_data )
2021-04-05 02:25:41 +00:00
elif method_name == " sonarr_folder " :
self . sonarr_options [ " folder " ] = method_data
elif method_name == " sonarr_monitor " :
if str ( method_data ) . lower ( ) in sonarr . monitor_translation :
self . sonarr_options [ " monitor " ] = str ( method_data ) . lower ( )
else :
raise Failed ( f " Collection Error: { method_name } attribute must be either all, future, missing, existing, pilot, first, latest or none " )
elif method_name == " sonarr_quality " :
self . library . Sonarr . get_profile_id ( method_data , " quality_profile " )
self . sonarr_options [ " quality " ] = method_data
elif method_name == " sonarr_language " :
self . library . Sonarr . get_profile_id ( method_data , " language_profile " )
self . sonarr_options [ " language " ] = method_data
elif method_name == " sonarr_series " :
if str ( method_data ) . lower ( ) in sonarr . series_type :
self . sonarr_options [ " series " ] = str ( method_data ) . lower ( )
else :
raise Failed ( f " Collection Error: { method_name } attribute must be either standard, daily, or anime " )
elif method_name in [ " sonarr_season " , " sonarr_search " , " sonarr_cutoff_search " ] :
2021-04-11 15:31:59 +00:00
self . sonarr_options [ method_name [ 7 : ] ] = util . get_bool ( method_name , method_data )
2021-04-05 02:25:41 +00:00
elif method_name == " sonarr_tag " :
self . sonarr_options [ " tag " ] = util . get_list ( method_data )
2021-03-26 05:43:11 +00:00
elif method_name in [ " title " , " title.and " , " title.not " , " title.begins " , " title.ends " ] :
self . methods . append ( ( " plex_search " , [ { method_name : util . get_list ( method_data , split = False ) } ] ) )
2021-05-05 18:01:41 +00:00
elif method_name in [ " year.gt " , " year.gte " , " year.lt " , " year.lte " ] :
2021-04-02 04:48:40 +00:00
self . methods . append ( ( " plex_search " , [ { method_name : util . check_year ( method_data , current_year , method_name ) } ] ) )
2021-03-26 05:43:11 +00:00
elif method_name in [ " added.before " , " added.after " , " originally_available.before " , " originally_available.after " ] :
2021-04-02 04:48:40 +00:00
self . methods . append ( ( " plex_search " , [ { method_name : util . check_date ( method_data , method_name , return_string = True , plex_date = True ) } ] ) )
2021-05-05 18:01:41 +00:00
elif method_name in [ " added " , " added.not " , " originally_available " , " originally_available.not " , " duration.gt " , " duration.gte " , " duration.lt " , " duration.lte " ] :
2021-04-02 04:48:40 +00:00
self . methods . append ( ( " plex_search " , [ { method_name : util . check_number ( method_data , method_name , minimum = 1 ) } ] ) )
2021-05-05 18:01:41 +00:00
elif method_name in [ " user_rating.gt " , " user_rating.gte " , " user_rating.lt " , " user_rating.lte " , " critic_rating.gt " , " critic_rating.gte " , " critic_rating.lt " , " critic_rating.lte " , " audience_rating.gt " , " audience_rating.gte " , " audience_rating.lt " , " audience_rating.lte " ] :
2021-04-03 18:00:05 +00:00
self . methods . append ( ( " plex_search " , [ { method_name : util . check_number ( method_data , method_name , number_type = " float " , minimum = 0 , maximum = 10 ) } ] ) )
2021-04-02 04:48:40 +00:00
elif method_name in [ " decade " , " year " , " year.not " ] :
2021-03-26 05:43:11 +00:00
self . methods . append ( ( " plex_search " , [ { method_name : util . get_year_list ( method_data , current_year , method_name ) } ] ) )
2021-03-30 05:50:53 +00:00
elif method_name in plex . searches :
if method_name in plex . tmdb_searches :
2021-03-26 05:43:11 +00:00
final_values = [ ]
for value in util . get_list ( method_data ) :
if value . lower ( ) == " tmdb " and " tmdb_person " in self . details :
for name in self . details [ " tmdb_person " ] :
final_values . append ( name )
else :
final_values . append ( value )
else :
final_values = method_data
2021-03-31 04:09:56 +00:00
search = os . path . splitext ( method_name ) [ 0 ]
valid_values = self . library . validate_search_list ( final_values , search )
2021-04-14 20:55:35 +00:00
if len ( valid_values ) > 0 :
2021-03-31 04:09:56 +00:00
self . methods . append ( ( " plex_search " , [ { method_name : valid_values } ] ) )
else :
logger . warning ( f " Collection Warning: No valid { search } values found in { final_values } " )
2021-02-20 05:41:45 +00:00
elif method_name == " plex_all " :
self . methods . append ( ( method_name , [ " " ] ) )
elif method_name == " plex_collection " :
2021-03-25 15:07:54 +00:00
self . methods . append ( ( method_name , self . library . validate_collections ( method_data if isinstance ( method_data , list ) else [ method_data ] ) ) )
2021-02-20 05:41:45 +00:00
elif method_name == " anidb_popular " :
2021-03-25 15:07:54 +00:00
list_count = util . regex_first_int ( method_data , " List Size " , default = 40 )
2021-02-20 05:41:45 +00:00
if 1 < = list_count < = 30 :
self . methods . append ( ( method_name , [ list_count ] ) )
else :
logger . warning ( " Collection Error: anidb_popular must be an integer between 1 and 30 defaulting to 30 " )
self . methods . append ( ( method_name , [ 30 ] ) )
elif method_name == " mal_id " :
2021-03-25 15:07:54 +00:00
self . methods . append ( ( method_name , util . get_int_list ( method_data , " MyAnimeList ID " ) ) )
2021-02-20 05:41:45 +00:00
elif method_name in [ " anidb_id " , " anidb_relation " ] :
2021-03-25 15:07:54 +00:00
self . methods . append ( ( method_name , config . AniDB . validate_anidb_list ( util . get_int_list ( method_data , " AniDB ID " ) , self . library . Plex . language ) ) )
2021-03-10 16:51:27 +00:00
elif method_name in [ " anilist_id " , " anilist_relations " , " anilist_studio " ] :
2021-03-25 15:07:54 +00:00
self . methods . append ( ( method_name , config . AniList . validate_anilist_ids ( util . get_int_list ( method_data , " AniList ID " ) , studio = method_name == " anilist_studio " ) ) )
2021-02-20 05:41:45 +00:00
elif method_name == " trakt_list " :
2021-03-27 07:30:07 +00:00
self . methods . append ( ( method_name , config . Trakt . validate_trakt ( util . get_list ( method_data ) ) ) )
2021-03-08 15:25:35 +00:00
elif method_name == " trakt_list_details " :
2021-03-27 07:30:07 +00:00
valid_list = config . Trakt . validate_trakt ( util . get_list ( method_data ) )
2021-03-08 15:25:35 +00:00
item = config . Trakt . standard_list ( valid_list [ 0 ] )
if hasattr ( item , " description " ) and item . description :
self . summaries [ method_name ] = item . description
self . methods . append ( ( method_name [ : - 8 ] , valid_list ) )
2021-03-27 07:30:07 +00:00
elif method_name in [ " trakt_watchlist " , " trakt_collection " ] :
2021-04-01 19:07:12 +00:00
self . methods . append ( ( method_name , config . Trakt . validate_trakt ( util . get_list ( method_data ) , trakt_type = method_name [ 6 : ] , is_movie = self . library . is_movie ) ) )
2021-02-20 05:41:45 +00:00
elif method_name == " imdb_list " :
new_list = [ ]
2021-03-25 15:07:54 +00:00
for imdb_list in util . get_list ( method_data , split = False ) :
2021-02-20 05:41:45 +00:00
if isinstance ( imdb_list , dict ) :
2021-03-21 23:00:10 +00:00
dict_methods = { dm . lower ( ) : dm for dm in imdb_list }
if " url " in dict_methods and imdb_list [ dict_methods [ " url " ] ] :
2021-04-15 18:39:47 +00:00
imdb_url = config . IMDb . validate_imdb_url ( imdb_list [ dict_methods [ " url " ] ] , self . library . Plex . language )
2021-03-21 23:00:10 +00:00
else :
raise Failed ( " Collection Error: imdb_list attribute url is required " )
2021-03-30 05:49:10 +00:00
if " limit " in dict_methods and imdb_list [ dict_methods [ " limit " ] ] :
list_count = util . regex_first_int ( imdb_list [ dict_methods [ " limit " ] ] , " List Limit " , default = 0 )
else :
list_count = 0
2021-02-20 05:41:45 +00:00
else :
2021-04-15 18:39:47 +00:00
imdb_url = config . IMDb . validate_imdb_url ( str ( imdb_list ) , self . library . Plex . language )
2021-02-20 05:41:45 +00:00
list_count = 0
new_list . append ( { " url " : imdb_url , " limit " : list_count } )
self . methods . append ( ( method_name , new_list ) )
2021-03-04 20:05:51 +00:00
elif method_name == " letterboxd_list " :
2021-03-25 15:07:54 +00:00
self . methods . append ( ( method_name , util . get_list ( method_data , split = False ) ) )
2021-03-08 21:54:19 +00:00
elif method_name == " letterboxd_list_details " :
2021-03-25 15:07:54 +00:00
values = util . get_list ( method_data , split = False )
2021-03-08 21:54:19 +00:00
self . summaries [ method_name ] = config . Letterboxd . get_list_description ( values [ 0 ] , self . library . Plex . language )
self . methods . append ( ( method_name [ : - 8 ] , values ) )
2021-03-30 05:50:53 +00:00
elif method_name in dictionary_builders :
2021-03-25 15:07:54 +00:00
if isinstance ( method_data , dict ) :
2021-03-21 23:00:10 +00:00
def get_int ( parent , method , data_in , methods_in , default_in , minimum = 1 , maximum = None ) :
if method not in methods_in :
2021-04-01 20:11:35 +00:00
logger . warning ( f " Collection Warning: { parent } { method } attribute not found using { default_in } as default " )
2021-03-21 23:00:10 +00:00
elif not data_in [ methods_in [ method ] ] :
logger . warning ( f " Collection Warning: { parent } { methods_in [ method ] } attribute is blank using { default_in } as default " )
elif isinstance ( data_in [ methods_in [ method ] ] , int ) and data_in [ methods_in [ method ] ] > = minimum :
if maximum is None or data_in [ methods_in [ method ] ] < = maximum :
return data_in [ methods_in [ method ] ]
else :
logger . warning ( f " Collection Warning: { parent } { methods_in [ method ] } attribute { data_in [ methods_in [ method ] ] } invalid must an integer <= { maximum } using { default_in } as default " )
else :
logger . warning ( f " Collection Warning: { parent } { methods_in [ method ] } attribute { data_in [ methods_in [ method ] ] } invalid must an integer >= { minimum } using { default_in } as default " )
2021-02-24 06:42:58 +00:00
return default_in
2021-02-20 05:41:45 +00:00
if method_name == " filters " :
2021-03-25 15:07:54 +00:00
for filter_name , filter_data in method_data . items ( ) :
2021-04-01 16:32:41 +00:00
modifier = filter_name [ - 4 : ] . lower ( )
2021-04-05 20:22:57 +00:00
modifier = modifier if modifier in [ " .not " , " .lte " , " .gte " ] else " "
2021-04-01 16:32:41 +00:00
method = filter_name [ : - 4 ] . lower ( ) if modifier in [ " .not " , " .lte " , " .gte " ] else filter_name . lower ( )
if method in method_alias :
filter_method = f " { method_alias [ method ] } { modifier } "
2021-03-25 15:07:54 +00:00
logger . warning ( f " Collection Warning: { filter_name } filter will run as { filter_method } " )
2021-02-20 05:41:45 +00:00
else :
2021-04-01 16:32:41 +00:00
filter_method = f " { method } { modifier } "
2021-03-30 05:50:53 +00:00
if filter_method in movie_only_filters and self . library . is_show :
2021-02-25 21:59:56 +00:00
raise Failed ( f " Collection Error: { filter_method } filter only works for movie libraries " )
2021-03-25 15:07:54 +00:00
elif filter_data is None :
2021-02-25 21:59:56 +00:00
raise Failed ( f " Collection Error: { filter_method } filter is blank " )
elif filter_method == " year " :
2021-03-25 15:07:54 +00:00
valid_data = util . get_year_list ( filter_data , current_year , f " { filter_method } filter " )
2021-02-28 08:36:40 +00:00
elif filter_method in [ " max_age " , " duration.gte " , " duration.lte " , " tmdb_vote_count.gte " , " tmdb_vote_count.lte " ] :
2021-03-25 15:07:54 +00:00
valid_data = util . check_number ( filter_data , f " { filter_method } filter " , minimum = 1 )
2021-02-25 21:59:56 +00:00
elif filter_method in [ " year.gte " , " year.lte " ] :
2021-03-25 15:07:54 +00:00
valid_data = util . check_year ( filter_data , current_year , f " { filter_method } filter " )
2021-04-08 03:45:37 +00:00
elif filter_method in [ " user_rating.gte " , " user_rating.lte " , " audience_rating.gte " , " audience_rating.lte " , " critic_rating.gte " , " critic_rating.lte " ] :
2021-03-25 15:07:54 +00:00
valid_data = util . check_number ( filter_data , f " { filter_method } filter " , number_type = " float " , minimum = 0.1 , maximum = 10 )
2021-02-25 21:59:56 +00:00
elif filter_method in [ " originally_available.gte " , " originally_available.lte " ] :
2021-03-25 15:07:54 +00:00
valid_data = util . check_date ( filter_data , f " { filter_method } filter " )
2021-04-28 18:42:55 +00:00
elif filter_method in [ " original_language " , " original_language.not " ] :
2021-03-25 15:07:54 +00:00
valid_data = util . get_list ( filter_data , lower = True )
2021-04-28 18:42:55 +00:00
elif filter_method in [ " collection " , " collection.not " ] :
2021-03-25 15:07:54 +00:00
valid_data = filter_data if isinstance ( filter_data , list ) else [ filter_data ]
2021-03-30 05:50:53 +00:00
elif filter_method in all_filters :
2021-03-25 15:07:54 +00:00
valid_data = util . get_list ( filter_data )
2021-02-25 21:59:56 +00:00
else :
raise Failed ( f " Collection Error: { filter_method } filter not supported " )
2021-03-25 15:07:54 +00:00
self . filters . append ( ( filter_method , valid_data ) )
2021-02-20 05:41:45 +00:00
elif method_name == " plex_collectionless " :
new_dictionary = { }
2021-03-25 15:07:54 +00:00
dict_methods = { dm . lower ( ) : dm for dm in method_data }
2021-02-20 05:41:45 +00:00
prefix_list = [ ]
2021-03-25 15:07:54 +00:00
if " exclude_prefix " in dict_methods and method_data [ dict_methods [ " exclude_prefix " ] ] :
if isinstance ( method_data [ dict_methods [ " exclude_prefix " ] ] , list ) :
2021-04-08 00:20:06 +00:00
prefix_list . extend ( [ exclude for exclude in method_data [ dict_methods [ " exclude_prefix " ] ] if exclude ] )
2021-03-21 23:00:10 +00:00
else :
2021-03-25 15:07:54 +00:00
prefix_list . append ( str ( method_data [ dict_methods [ " exclude_prefix " ] ] ) )
2021-02-20 05:41:45 +00:00
exact_list = [ ]
2021-03-25 15:07:54 +00:00
if " exclude " in dict_methods and method_data [ dict_methods [ " exclude " ] ] :
if isinstance ( method_data [ dict_methods [ " exclude " ] ] , list ) :
2021-04-08 00:20:06 +00:00
exact_list . extend ( [ exclude for exclude in method_data [ dict_methods [ " exclude " ] ] if exclude ] )
2021-03-21 23:00:10 +00:00
else :
2021-03-25 15:07:54 +00:00
exact_list . append ( str ( method_data [ dict_methods [ " exclude " ] ] ) )
2021-03-21 23:00:10 +00:00
if len ( prefix_list ) == 0 and len ( exact_list ) == 0 :
raise Failed ( " Collection Error: you must have at least one exclusion " )
2021-04-28 19:08:48 +00:00
exact_list . append ( self . name )
2021-02-20 05:41:45 +00:00
new_dictionary [ " exclude_prefix " ] = prefix_list
new_dictionary [ " exclude " ] = exact_list
self . methods . append ( ( method_name , [ new_dictionary ] ) )
elif method_name == " plex_search " :
2021-03-26 05:43:11 +00:00
searches = { }
2021-03-26 16:22:25 +00:00
for search_name , search_data in method_data . items ( ) :
2021-05-05 18:01:41 +00:00
search , modifier , search_final = split_attribute ( search_name )
if search_name != search_final :
logger . warning ( f " Collection Warning: { search_name } plex search attribute will run as { search_final } " )
2021-03-30 05:50:53 +00:00
if search_final in plex . movie_only_searches and self . library . is_show :
2021-03-26 05:43:11 +00:00
raise Failed ( f " Collection Error: { search_final } plex search attribute only works for movie libraries " )
2021-05-05 18:01:41 +00:00
elif search_final in plex . show_only_searches and self . library . is_movie :
2021-04-01 15:34:02 +00:00
raise Failed ( f " Collection Error: { search_final } plex search attribute only works for show libraries " )
2021-05-05 18:01:41 +00:00
elif search_final not in plex . searches :
raise Failed ( f " Collection Error: { search_final } is not a valid plex search attribute " )
2021-03-26 05:43:11 +00:00
elif search_data is None :
raise Failed ( f " Collection Error: { search_final } plex search attribute is blank " )
elif search == " sort_by " :
2021-03-30 05:50:53 +00:00
if str ( search_data ) . lower ( ) in plex . sorts :
2021-03-26 05:43:11 +00:00
searches [ search ] = str ( search_data ) . lower ( )
else :
logger . warning ( f " Collection Error: { search_data } is not a valid plex search sort defaulting to title.asc " )
elif search == " limit " :
if not search_data :
raise Failed ( f " Collection Warning: plex search limit attribute is blank " )
elif not isinstance ( search_data , int ) and search_data > 0 :
raise Failed ( f " Collection Warning: plex search limit attribute: { search_data } must be an integer greater then 0 " )
else :
searches [ search ] = search_data
elif search == " title " and modifier in [ " " , " .and " , " .not " , " .begins " , " .ends " ] :
searches [ search_final ] = util . get_list ( search_data , split = False )
2021-05-05 18:01:41 +00:00
elif search in plex . tags and modifier in [ " " , " .and " , " .not " , " .begins " , " .ends " ] :
2021-03-30 05:50:53 +00:00
if search_final in plex . tmdb_searches :
2021-03-26 05:43:11 +00:00
final_values = [ ]
for value in util . get_list ( search_data ) :
if value . lower ( ) == " tmdb " and " tmdb_person " in self . details :
for name in self . details [ " tmdb_person " ] :
final_values . append ( name )
else :
final_values . append ( value )
else :
2021-05-05 18:01:41 +00:00
final_values = util . get_list ( search_data )
2021-03-31 04:09:56 +00:00
valid_values = self . library . validate_search_list ( final_values , search )
if valid_values :
searches [ search_final ] = valid_values
else :
logger . warning ( f " Collection Warning: No valid { search } values found in { final_values } " )
2021-05-05 18:01:41 +00:00
elif search == " year " and modifier in [ " .gt " , " .gte " , " .lt " , " .lte " ] :
2021-04-02 04:48:40 +00:00
searches [ search_final ] = util . check_year ( search_data , current_year , search_final )
elif search in [ " added " , " originally_available " ] and modifier in [ " .before " , " .after " ] :
searches [ search_final ] = util . check_date ( search_data , search_final , return_string = True , plex_date = True )
2021-05-05 18:01:41 +00:00
elif search in [ " added " , " originally_available " , " duration " ] and modifier in [ " " , " .not " , " .gt " , " .gte " , " .lt " , " .lte " ] :
2021-04-02 04:48:40 +00:00
searches [ search_final ] = util . check_number ( search_data , search_final , minimum = 1 )
2021-05-05 18:01:41 +00:00
elif search in [ " user_rating " , " critic_rating " , " audience_rating " ] and modifier in [ " .gt " , " .gte " , " .lt " , " .lte " ] :
2021-04-03 18:00:05 +00:00
searches [ search_final ] = util . check_number ( search_data , search_final , number_type = " float " , minimum = 0 , maximum = 10 )
2021-05-05 18:01:41 +00:00
elif search in [ " decade " , " year " ] and modifier in [ " " , " .not " ] :
2021-03-26 05:43:11 +00:00
searches [ search_final ] = util . get_year_list ( search_data , current_year , search_final )
2021-02-20 05:41:45 +00:00
else :
2021-05-05 18:01:41 +00:00
raise Failed ( f " Collection Error: modifier: { modifier } not supported with the { search } plex search attribute " )
2021-04-08 02:15:52 +00:00
if len ( searches ) > 0 :
self . methods . append ( ( method_name , [ searches ] ) )
else :
raise Failed ( " Collection Error: no valid plex search attributes " )
2021-02-20 05:41:45 +00:00
elif method_name == " tmdb_discover " :
new_dictionary = { " limit " : 100 }
2021-03-26 22:17:24 +00:00
for discover_name , discover_data in method_data . items ( ) :
2021-03-25 15:07:54 +00:00
discover_final = discover_name . lower ( )
if discover_data :
2021-03-30 05:50:53 +00:00
if ( self . library . is_movie and discover_final in tmdb . discover_movie ) or ( self . library . is_show and discover_final in tmdb . discover_tv ) :
2021-03-25 15:07:54 +00:00
if discover_final == " language " :
if re . compile ( " ([a-z] {2} )-([A-Z] {2} ) " ) . match ( str ( discover_data ) ) :
new_dictionary [ discover_final ] = str ( discover_data )
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_name } attribute { discover_final } : { discover_data } must match pattern ([a-z] {{ 2 }} )-([A-Z] {{ 2 }} ) e.g. en-US " )
elif discover_final == " region " :
if re . compile ( " ^[A-Z] {2} $ " ) . match ( str ( discover_data ) ) :
new_dictionary [ discover_final ] = str ( discover_data )
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_name } attribute { discover_final } : { discover_data } must match pattern ^[A-Z] {{ 2 }} $ e.g. US " )
elif discover_final == " sort_by " :
2021-03-30 05:50:53 +00:00
if ( self . library . is_movie and discover_data in tmdb . discover_movie_sort ) or ( self . library . is_show and discover_data in tmdb . discover_tv_sort ) :
2021-03-25 15:07:54 +00:00
new_dictionary [ discover_final ] = discover_data
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_name } attribute { discover_final } : { discover_data } is invalid " )
elif discover_final == " certification_country " :
if " certification " in method_data or " certification.lte " in method_data or " certification.gte " in method_data :
new_dictionary [ discover_final ] = discover_data
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_name } attribute { discover_final } : must be used with either certification, certification.lte, or certification.gte " )
elif discover_final in [ " certification " , " certification.lte " , " certification.gte " ] :
if " certification_country " in method_data :
new_dictionary [ discover_final ] = discover_data
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_name } attribute { discover_final } : must be used with certification_country " )
elif discover_final in [ " include_adult " , " include_null_first_air_dates " , " screened_theatrically " ] :
if discover_data is True :
new_dictionary [ discover_final ] = discover_data
2021-03-30 05:50:53 +00:00
elif discover_final in tmdb . discover_dates :
2021-03-25 15:07:54 +00:00
new_dictionary [ discover_final ] = util . check_date ( discover_data , f " { method_name } attribute { discover_final } " , return_string = True )
elif discover_final in [ " primary_release_year " , " year " , " first_air_date_year " ] :
new_dictionary [ discover_final ] = util . check_number ( discover_data , f " { method_name } attribute { discover_final } " , minimum = 1800 , maximum = current_year + 1 )
elif discover_final in [ " vote_count.gte " , " vote_count.lte " , " vote_average.gte " , " vote_average.lte " , " with_runtime.gte " , " with_runtime.lte " ] :
new_dictionary [ discover_final ] = util . check_number ( discover_data , f " { method_name } attribute { discover_final } " , minimum = 1 )
elif discover_final in [ " with_cast " , " with_crew " , " with_people " , " with_companies " , " with_networks " , " with_genres " , " without_genres " , " with_keywords " , " without_keywords " , " with_original_language " , " timezone " ] :
new_dictionary [ discover_final ] = discover_data
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_name } attribute { discover_final } not supported " )
elif discover_final == " limit " :
if isinstance ( discover_data , int ) and discover_data > 0 :
new_dictionary [ discover_final ] = discover_data
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_name } attribute { discover_final } : must be a valid number greater then 0 " )
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_name } attribute { discover_final } not supported " )
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_name } parameter { discover_final } is blank " )
2021-02-20 05:41:45 +00:00
if len ( new_dictionary ) > 1 :
self . methods . append ( ( method_name , [ new_dictionary ] ) )
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_name } had no valid fields " )
2021-02-20 05:41:45 +00:00
elif " tautulli " in method_name :
new_dictionary = { }
2021-03-21 23:00:10 +00:00
if method_name == " tautulli_popular " :
new_dictionary [ " list_type " ] = " popular "
elif method_name == " tautulli_watched " :
new_dictionary [ " list_type " ] = " watched "
else :
raise Failed ( f " Collection Error: { method_name } attribute not supported " )
2021-03-25 15:07:54 +00:00
dict_methods = { dm . lower ( ) : dm for dm in method_data }
new_dictionary [ " list_days " ] = get_int ( method_name , " list_days " , method_data , dict_methods , 30 )
new_dictionary [ " list_size " ] = get_int ( method_name , " list_size " , method_data , dict_methods , 10 )
new_dictionary [ " list_buffer " ] = get_int ( method_name , " list_buffer " , method_data , dict_methods , 20 )
2021-02-20 05:41:45 +00:00
self . methods . append ( ( method_name , [ new_dictionary ] ) )
elif method_name == " mal_season " :
new_dictionary = { " sort_by " : " anime_num_list_users " }
2021-03-25 15:07:54 +00:00
dict_methods = { dm . lower ( ) : dm for dm in method_data }
2021-03-21 23:00:10 +00:00
if " sort_by " not in dict_methods :
logger . warning ( " Collection Warning: mal_season sort_by attribute not found using members as default " )
2021-03-25 15:07:54 +00:00
elif not method_data [ dict_methods [ " sort_by " ] ] :
2021-03-21 23:00:10 +00:00
logger . warning ( " Collection Warning: mal_season sort_by attribute is blank using members as default " )
2021-03-30 05:50:53 +00:00
elif method_data [ dict_methods [ " sort_by " ] ] not in mal . season_sort :
2021-03-25 15:07:54 +00:00
logger . warning ( f " Collection Warning: mal_season sort_by attribute { method_data [ dict_methods [ ' sort_by ' ] ] } invalid must be either ' members ' or ' score ' using members as default " )
2021-03-21 23:00:10 +00:00
else :
2021-03-30 05:50:53 +00:00
new_dictionary [ " sort_by " ] = mal . season_sort [ method_data [ dict_methods [ " sort_by " ] ] ]
2021-02-20 05:41:45 +00:00
if current_time . month in [ 1 , 2 , 3 ] : new_dictionary [ " season " ] = " winter "
elif current_time . month in [ 4 , 5 , 6 ] : new_dictionary [ " season " ] = " spring "
elif current_time . month in [ 7 , 8 , 9 ] : new_dictionary [ " season " ] = " summer "
elif current_time . month in [ 10 , 11 , 12 ] : new_dictionary [ " season " ] = " fall "
2021-03-21 23:00:10 +00:00
if " season " not in dict_methods :
logger . warning ( f " Collection Warning: mal_season season attribute not found using the current season: { new_dictionary [ ' season ' ] } as default " )
2021-03-25 15:07:54 +00:00
elif not method_data [ dict_methods [ " season " ] ] :
2021-03-21 23:00:10 +00:00
logger . warning ( f " Collection Warning: mal_season season attribute is blank using the current season: { new_dictionary [ ' season ' ] } as default " )
2021-03-25 15:07:54 +00:00
elif method_data [ dict_methods [ " season " ] ] not in util . pretty_seasons :
logger . warning ( f " Collection Warning: mal_season season attribute { method_data [ dict_methods [ ' season ' ] ] } invalid must be either ' winter ' , ' spring ' , ' summer ' or ' fall ' using the current season: { new_dictionary [ ' season ' ] } as default " )
2021-03-21 23:00:10 +00:00
else :
2021-03-25 15:07:54 +00:00
new_dictionary [ " season " ] = method_data [ dict_methods [ " season " ] ]
2021-02-20 05:41:45 +00:00
2021-03-25 15:07:54 +00:00
new_dictionary [ " year " ] = get_int ( method_name , " year " , method_data , dict_methods , current_time . year , minimum = 1917 , maximum = current_time . year + 1 )
new_dictionary [ " limit " ] = get_int ( method_name , " limit " , method_data , dict_methods , 100 , maximum = 500 )
2021-02-20 05:41:45 +00:00
self . methods . append ( ( method_name , [ new_dictionary ] ) )
elif method_name == " mal_userlist " :
new_dictionary = { " status " : " all " , " sort_by " : " list_score " }
2021-03-25 15:07:54 +00:00
dict_methods = { dm . lower ( ) : dm for dm in method_data }
2021-03-21 23:00:10 +00:00
if " username " not in dict_methods :
raise Failed ( " Collection Error: mal_userlist username attribute is required " )
2021-03-25 15:07:54 +00:00
elif not method_data [ dict_methods [ " username " ] ] :
2021-03-21 23:00:10 +00:00
raise Failed ( " Collection Error: mal_userlist username attribute is blank " )
else :
2021-03-25 15:07:54 +00:00
new_dictionary [ " username " ] = method_data [ dict_methods [ " username " ] ]
2021-02-20 05:41:45 +00:00
2021-03-21 23:00:10 +00:00
if " status " not in dict_methods :
logger . warning ( " Collection Warning: mal_season status attribute not found using all as default " )
2021-03-25 15:07:54 +00:00
elif not method_data [ dict_methods [ " status " ] ] :
2021-03-21 23:00:10 +00:00
logger . warning ( " Collection Warning: mal_season status attribute is blank using all as default " )
2021-03-30 05:50:53 +00:00
elif method_data [ dict_methods [ " status " ] ] not in mal . userlist_status :
2021-03-25 15:07:54 +00:00
logger . warning ( f " Collection Warning: mal_season status attribute { method_data [ dict_methods [ ' status ' ] ] } invalid must be either ' all ' , ' watching ' , ' completed ' , ' on_hold ' , ' dropped ' or ' plan_to_watch ' using all as default " )
2021-03-21 23:00:10 +00:00
else :
2021-03-30 05:50:53 +00:00
new_dictionary [ " status " ] = mal . userlist_status [ method_data [ dict_methods [ " status " ] ] ]
2021-02-20 05:41:45 +00:00
2021-03-21 23:00:10 +00:00
if " sort_by " not in dict_methods :
logger . warning ( " Collection Warning: mal_season sort_by attribute not found using score as default " )
2021-03-25 15:07:54 +00:00
elif not method_data [ dict_methods [ " sort_by " ] ] :
2021-03-21 23:00:10 +00:00
logger . warning ( " Collection Warning: mal_season sort_by attribute is blank using score as default " )
2021-03-30 05:50:53 +00:00
elif method_data [ dict_methods [ " sort_by " ] ] not in mal . userlist_sort :
2021-03-25 15:07:54 +00:00
logger . warning ( f " Collection Warning: mal_season sort_by attribute { method_data [ dict_methods [ ' sort_by ' ] ] } invalid must be either ' score ' , ' last_updated ' , ' title ' or ' start_date ' using score as default " )
2021-03-21 23:00:10 +00:00
else :
2021-03-30 05:50:53 +00:00
new_dictionary [ " sort_by " ] = mal . userlist_sort [ method_data [ dict_methods [ " sort_by " ] ] ]
2021-02-20 05:41:45 +00:00
2021-03-25 15:07:54 +00:00
new_dictionary [ " limit " ] = get_int ( method_name , " limit " , method_data , dict_methods , 100 , maximum = 1000 )
2021-02-20 05:41:45 +00:00
self . methods . append ( ( method_name , [ new_dictionary ] ) )
2021-03-20 04:22:51 +00:00
elif " anilist " in method_name :
2021-03-10 16:51:27 +00:00
new_dictionary = { " sort_by " : " score " }
2021-03-25 15:07:54 +00:00
dict_methods = { dm . lower ( ) : dm for dm in method_data }
2021-03-20 04:22:51 +00:00
if method_name == " anilist_season " :
if current_time . month in [ 12 , 1 , 2 ] : new_dictionary [ " season " ] = " winter "
elif current_time . month in [ 3 , 4 , 5 ] : new_dictionary [ " season " ] = " spring "
elif current_time . month in [ 6 , 7 , 8 ] : new_dictionary [ " season " ] = " summer "
elif current_time . month in [ 9 , 10 , 11 ] : new_dictionary [ " season " ] = " fall "
2021-03-10 16:51:27 +00:00
2021-03-21 23:00:10 +00:00
if " season " not in dict_methods :
logger . warning ( f " Collection Warning: anilist_season season attribute not found using the current season: { new_dictionary [ ' season ' ] } as default " )
2021-03-25 15:07:54 +00:00
elif not method_data [ dict_methods [ " season " ] ] :
2021-03-21 23:00:10 +00:00
logger . warning ( f " Collection Warning: anilist_season season attribute is blank using the current season: { new_dictionary [ ' season ' ] } as default " )
2021-03-25 15:07:54 +00:00
elif method_data [ dict_methods [ " season " ] ] not in util . pretty_seasons :
logger . warning ( f " Collection Warning: anilist_season season attribute { method_data [ dict_methods [ ' season ' ] ] } invalid must be either ' winter ' , ' spring ' , ' summer ' or ' fall ' using the current season: { new_dictionary [ ' season ' ] } as default " )
2021-03-21 23:00:10 +00:00
else :
2021-03-25 15:07:54 +00:00
new_dictionary [ " season " ] = method_data [ dict_methods [ " season " ] ]
2021-03-10 16:51:27 +00:00
2021-03-25 15:07:54 +00:00
new_dictionary [ " year " ] = get_int ( method_name , " year " , method_data , dict_methods , current_time . year , minimum = 1917 , maximum = current_time . year + 1 )
2021-03-20 04:22:51 +00:00
elif method_name == " anilist_genre " :
2021-03-21 23:00:10 +00:00
if " genre " not in dict_methods :
raise Failed ( f " Collection Warning: anilist_genre genre attribute not found " )
2021-03-25 15:07:54 +00:00
elif not method_data [ dict_methods [ " genre " ] ] :
2021-03-21 23:00:10 +00:00
raise Failed ( f " Collection Warning: anilist_genre genre attribute is blank " )
else :
2021-03-25 15:07:54 +00:00
new_dictionary [ " genre " ] = self . config . AniList . validate_genre ( method_data [ dict_methods [ " genre " ] ] )
2021-03-20 04:22:51 +00:00
elif method_name == " anilist_tag " :
2021-03-21 23:00:10 +00:00
if " tag " not in dict_methods :
raise Failed ( f " Collection Warning: anilist_tag tag attribute not found " )
2021-03-25 15:07:54 +00:00
elif not method_data [ dict_methods [ " tag " ] ] :
2021-03-21 23:00:10 +00:00
raise Failed ( f " Collection Warning: anilist_tag tag attribute is blank " )
else :
2021-03-25 15:07:54 +00:00
new_dictionary [ " tag " ] = self . config . AniList . validate_tag ( method_data [ dict_methods [ " tag " ] ] )
2021-03-20 04:22:51 +00:00
2021-03-21 23:00:10 +00:00
if " sort_by " not in dict_methods :
logger . warning ( f " Collection Warning: { method_name } sort_by attribute not found using score as default " )
2021-03-25 15:07:54 +00:00
elif not method_data [ dict_methods [ " sort_by " ] ] :
2021-03-21 23:00:10 +00:00
logger . warning ( f " Collection Warning: { method_name } sort_by attribute is blank using score as default " )
2021-03-25 15:07:54 +00:00
elif str ( method_data [ dict_methods [ " sort_by " ] ] ) . lower ( ) not in [ " score " , " popular " ] :
logger . warning ( f " Collection Warning: { method_name } sort_by attribute { method_data [ dict_methods [ ' sort_by ' ] ] } invalid must be either ' score ' or ' popular ' using score as default " )
2021-03-21 23:00:10 +00:00
else :
2021-03-25 15:07:54 +00:00
new_dictionary [ " sort_by " ] = method_data [ dict_methods [ " sort_by " ] ]
2021-03-10 16:51:27 +00:00
2021-03-25 15:07:54 +00:00
new_dictionary [ " limit " ] = get_int ( method_name , " limit " , method_data , dict_methods , 0 , maximum = 500 )
2021-03-20 04:22:51 +00:00
2021-03-10 16:51:27 +00:00
self . methods . append ( ( method_name , [ new_dictionary ] ) )
2021-02-20 05:41:45 +00:00
else :
2021-03-25 15:07:54 +00:00
raise Failed ( f " Collection Error: { method_name } attribute is not a dictionary: { method_data } " )
2021-03-30 05:50:53 +00:00
elif method_name in numbered_builders :
2021-03-25 15:07:54 +00:00
list_count = util . regex_first_int ( method_data , " List Size " , default = 10 )
2021-02-20 05:41:45 +00:00
if list_count < 1 :
2021-03-09 07:50:44 +00:00
logger . warning ( f " Collection Warning: { method_name } must be an integer greater then 0 defaulting to 10 " )
list_count = 10
2021-02-20 05:41:45 +00:00
self . methods . append ( ( method_name , [ list_count ] ) )
2021-03-08 17:02:40 +00:00
elif " tvdb " in method_name :
2021-03-25 15:07:54 +00:00
values = util . get_list ( method_data )
2021-03-08 17:02:40 +00:00
if method_name [ - 8 : ] == " _details " :
if method_name == " tvdb_movie_details " :
2021-03-08 19:53:05 +00:00
item = config . TVDb . get_movie ( self . library . Plex . language , values [ 0 ] )
2021-03-08 17:02:40 +00:00
if hasattr ( item , " description " ) and item . description :
self . summaries [ method_name ] = item . description
if hasattr ( item , " background_path " ) and item . background_path :
self . backgrounds [ method_name ] = f " { config . TMDb . image_url } { item . background_path } "
if hasattr ( item , " poster_path " ) and item . poster_path :
self . posters [ method_name ] = f " { config . TMDb . image_url } { item . poster_path } "
elif method_name == " tvdb_show_details " :
2021-03-08 19:53:05 +00:00
item = config . TVDb . get_series ( self . library . Plex . language , values [ 0 ] )
2021-03-08 17:02:40 +00:00
if hasattr ( item , " description " ) and item . description :
self . summaries [ method_name ] = item . description
if hasattr ( item , " background_path " ) and item . background_path :
self . backgrounds [ method_name ] = f " { config . TMDb . image_url } { item . background_path } "
if hasattr ( item , " poster_path " ) and item . poster_path :
self . posters [ method_name ] = f " { config . TMDb . image_url } { item . poster_path } "
elif method_name == " tvdb_list_details " :
2021-03-08 19:53:05 +00:00
self . summaries [ method_name ] = config . TVDb . get_list_description ( values [ 0 ] , self . library . Plex . language )
2021-03-08 17:02:40 +00:00
self . methods . append ( ( method_name [ : - 8 ] , values ) )
else :
self . methods . append ( ( method_name , values ) )
2021-03-30 05:50:53 +00:00
elif method_name in tmdb . builders :
values = config . TMDb . validate_tmdb_list ( util . get_int_list ( method_data , f " TMDb { tmdb . type_map [ method_name ] } ID " ) , tmdb . type_map [ method_name ] )
2021-02-20 05:41:45 +00:00
if method_name [ - 8 : ] == " _details " :
if method_name in [ " tmdb_collection_details " , " tmdb_movie_details " , " tmdb_show_details " ] :
item = config . TMDb . get_movie_show_or_collection ( values [ 0 ] , self . library . is_movie )
2021-02-28 23:06:04 +00:00
if hasattr ( item , " overview " ) and item . overview :
self . summaries [ method_name ] = item . overview
if hasattr ( item , " backdrop_path " ) and item . backdrop_path :
self . backgrounds [ method_name ] = f " { config . TMDb . image_url } { item . backdrop_path } "
if hasattr ( item , " poster_path " ) and item . poster_path :
self . posters [ method_name ] = f " { config . TMDb . image_url } { item . poster_path } "
2021-03-02 05:28:32 +00:00
elif method_name in [ " tmdb_actor_details " , " tmdb_crew_details " , " tmdb_director_details " , " tmdb_producer_details " , " tmdb_writer_details " ] :
item = config . TMDb . get_person ( values [ 0 ] )
if hasattr ( item , " biography " ) and item . biography :
self . summaries [ method_name ] = item . biography
if hasattr ( item , " profile_path " ) and item . profile_path :
self . posters [ method_name ] = f " { config . TMDb . image_url } { item . profile_path } "
2021-02-20 05:41:45 +00:00
else :
item = config . TMDb . get_list ( values [ 0 ] )
2021-02-28 23:06:04 +00:00
if hasattr ( item , " description " ) and item . description :
self . summaries [ method_name ] = item . description
2021-02-20 05:41:45 +00:00
self . methods . append ( ( method_name [ : - 8 ] , values ) )
else :
self . methods . append ( ( method_name , values ) )
2021-03-30 05:50:53 +00:00
elif method_name in all_builders :
2021-03-25 15:07:54 +00:00
self . methods . append ( ( method_name , util . get_list ( method_data ) ) )
2021-03-30 05:50:53 +00:00
elif method_name not in ignored_details :
2021-02-24 06:44:06 +00:00
raise Failed ( f " Collection Error: { method_name } attribute not supported " )
2021-04-05 02:25:41 +00:00
elif method_key . lower ( ) in all_builders or method_key . lower ( ) in method_alias or method_key . lower ( ) in plex . searches :
raise Failed ( f " Collection Error: { method_key } attribute is blank " )
2021-03-05 19:19:22 +00:00
else :
2021-04-05 02:25:41 +00:00
logger . warning ( f " Collection Warning: { method_key } attribute is blank " )
2021-02-20 05:41:45 +00:00
2021-03-01 20:59:10 +00:00
self . sync = self . library . sync_mode == " sync "
2021-03-21 23:00:10 +00:00
if " sync_mode " in methods :
if not self . data [ methods [ " sync_mode " ] ] :
logger . warning ( f " Collection Warning: sync_mode attribute is blank using general: { self . library . sync_mode } " )
elif self . data [ methods [ " sync_mode " ] ] . lower ( ) not in [ " append " , " sync " ] :
logger . warning ( f " Collection Warning: { self . data [ methods [ ' sync_mode ' ] ] } sync_mode invalid using general: { self . library . sync_mode } " )
else :
self . sync = self . data [ methods [ " sync_mode " ] ] . lower ( ) == " sync "
2021-03-01 20:59:10 +00:00
2021-04-05 02:25:41 +00:00
if self . add_to_radarr is None :
self . add_to_radarr = self . library . Radarr . add if self . library . Radarr else False
if self . add_to_sonarr is None :
self . add_to_sonarr = self . library . Sonarr . add if self . library . Sonarr else False
2021-05-02 04:22:48 +00:00
2021-05-05 18:01:41 +00:00
if self . smart_url :
2021-05-02 04:22:48 +00:00
self . add_to_radarr = False
self . add_to_sonarr = False
2021-02-20 05:41:45 +00:00
2021-03-16 15:49:05 +00:00
if self . collectionless :
2021-04-05 02:25:41 +00:00
self . add_to_radarr = False
self . add_to_sonarr = False
2021-03-16 15:49:05 +00:00
self . details [ " collection_mode " ] = " hide "
self . sync = True
2021-02-24 06:42:58 +00:00
def run_methods ( self , collection_obj , collection_name , rating_key_map , movie_map , show_map ) :
2021-02-20 05:41:45 +00:00
items_found = 0
for method , values in self . methods :
logger . debug ( " " )
2021-02-24 06:44:06 +00:00
logger . debug ( f " Method: { method } " )
logger . debug ( f " Values: { values } " )
2021-02-20 05:41:45 +00:00
for value in values :
items = [ ]
missing_movies = [ ]
missing_shows = [ ]
def check_map ( input_ids ) :
movie_ids , show_ids = input_ids
items_found_inside = 0
if len ( movie_ids ) > 0 :
items_found_inside + = len ( movie_ids )
for movie_id in movie_ids :
2021-04-03 20:48:27 +00:00
if movie_id in movie_map :
2021-04-22 03:13:40 +00:00
items . extend ( movie_map [ movie_id ] )
2021-04-03 20:48:27 +00:00
else :
missing_movies . append ( movie_id )
2021-02-20 05:41:45 +00:00
if len ( show_ids ) > 0 :
items_found_inside + = len ( show_ids )
for show_id in show_ids :
2021-04-22 03:13:40 +00:00
if show_id in show_map :
items . extend ( show_map [ show_id ] )
else :
missing_shows . append ( show_id )
2021-02-20 05:41:45 +00:00
return items_found_inside
2021-05-05 18:01:41 +00:00
logger . debug ( " " )
2021-02-24 06:44:06 +00:00
logger . debug ( f " Value: { value } " )
2021-05-05 18:01:41 +00:00
logger . info ( " " )
2021-04-03 18:00:05 +00:00
if " plex " in method :
items = self . library . get_items ( method , value )
2021-02-20 05:41:45 +00:00
items_found + = len ( items )
elif " tautulli " in method :
items = self . library . Tautulli . get_items ( self . library , time_range = value [ " list_days " ] , stats_count = value [ " list_size " ] , list_type = value [ " list_type " ] , stats_count_buffer = value [ " list_buffer " ] )
items_found + = len ( items )
2021-03-05 20:33:24 +00:00
elif " anidb " in method : items_found + = check_map ( self . config . AniDB . get_items ( method , value , self . library . Plex . language ) )
2021-04-15 20:45:35 +00:00
elif " anilist " in method : items_found + = check_map ( self . config . AniList . get_items ( method , value , self . library . Plex . language ) )
elif " mal " in method : items_found + = check_map ( self . config . MyAnimeList . get_items ( method , value , self . library . Plex . language ) )
2021-02-20 05:41:45 +00:00
elif " tvdb " in method : items_found + = check_map ( self . config . TVDb . get_items ( method , value , self . library . Plex . language ) )
2021-03-05 20:33:24 +00:00
elif " imdb " in method : items_found + = check_map ( self . config . IMDb . get_items ( method , value , self . library . Plex . language ) )
2021-03-04 20:05:51 +00:00
elif " letterboxd " in method : items_found + = check_map ( self . config . Letterboxd . get_items ( method , value , self . library . Plex . language ) )
2021-02-20 05:41:45 +00:00
elif " tmdb " in method : items_found + = check_map ( self . config . TMDb . get_items ( method , value , self . library . is_movie ) )
elif " trakt " in method : items_found + = check_map ( self . config . Trakt . get_items ( method , value , self . library . is_movie ) )
2021-02-24 06:44:06 +00:00
else : logger . error ( f " Collection Error: { method } method not supported " )
2021-02-20 05:41:45 +00:00
2021-05-05 18:01:41 +00:00
logger . info ( " " )
2021-05-02 04:22:48 +00:00
if len ( items ) > 0 :
rating_key_map = self . library . add_to_collection ( collection_obj if collection_obj else collection_name , items , self . filters , self . details [ " show_filtered " ] , self . smart_label_collection , rating_key_map , movie_map , show_map )
else :
logger . error ( " No items found to add to this collection " )
2021-02-20 05:41:45 +00:00
if len ( missing_movies ) > 0 or len ( missing_shows ) > 0 :
logger . info ( " " )
2021-02-28 08:36:40 +00:00
arr_filters = [ ]
for filter_method , filter_data in self . filters :
if ( filter_method . startswith ( " original_language " ) and self . library . is_movie ) or filter_method . startswith ( " tmdb_vote_count " ) :
arr_filters . append ( ( filter_method , filter_data ) )
2021-02-20 05:41:45 +00:00
if len ( missing_movies ) > 0 :
missing_movies_with_names = [ ]
for missing_id in missing_movies :
try :
movie = self . config . TMDb . get_movie ( missing_id )
except Failed as e :
logger . error ( e )
2021-02-28 08:36:40 +00:00
continue
match = True
for filter_method , filter_data in arr_filters :
if ( filter_method == " original_language " and movie . original_language not in filter_data ) \
or ( filter_method == " original_language.not " and movie . original_language in filter_data ) \
or ( filter_method == " tmdb_vote_count.gte " and movie . vote_count < filter_data ) \
or ( filter_method == " tmdb_vote_count.lte " and movie . vote_count > filter_data ) :
match = False
break
if match :
missing_movies_with_names . append ( ( movie . title , missing_id ) )
if self . details [ " show_missing " ] is True :
logger . info ( f " { collection_name } Collection | ? | { movie . title } (TMDb: { missing_id } ) " )
elif self . details [ " show_filtered " ] is True :
logger . info ( f " { collection_name } Collection | X | { movie . title } (TMDb: { missing_id } ) " )
2021-02-24 06:44:06 +00:00
logger . info ( f " { len ( missing_movies_with_names ) } Movie { ' s ' if len ( missing_movies_with_names ) > 1 else ' ' } Missing " )
2021-02-20 06:41:40 +00:00
if self . details [ " save_missing " ] is True :
self . library . add_missing ( collection_name , missing_movies_with_names , True )
2021-05-02 04:22:48 +00:00
if ( self . add_to_radarr and self . library . Radarr ) or self . run_again :
missing_tmdb_ids = [ missing_id for title , missing_id in missing_movies_with_names ]
if self . add_to_radarr and self . library . Radarr :
try :
self . library . Radarr . add_tmdb ( missing_tmdb_ids , * * self . radarr_options )
except Failed as e :
logger . error ( e )
if self . run_again :
self . missing_movies . extend ( missing_tmdb_ids )
2021-02-20 05:41:45 +00:00
if len ( missing_shows ) > 0 and self . library . is_show :
missing_shows_with_names = [ ]
for missing_id in missing_shows :
try :
2021-03-08 19:53:05 +00:00
title = str ( self . config . TVDb . get_series ( self . library . Plex . language , missing_id ) . title . encode ( " ascii " , " replace " ) . decode ( ) )
2021-02-28 08:36:40 +00:00
except Failed as e :
logger . error ( e )
continue
match = True
if arr_filters :
show = self . config . TMDb . get_show ( self . config . TMDb . convert_tvdb_to_tmdb ( missing_id ) )
for filter_method , filter_data in arr_filters :
if ( filter_method == " tmdb_vote_count.gte " and show . vote_count < filter_data ) \
or ( filter_method == " tmdb_vote_count.lte " and show . vote_count > filter_data ) :
match = False
break
if match :
2021-02-20 05:41:45 +00:00
missing_shows_with_names . append ( ( title , missing_id ) )
2021-02-20 06:41:40 +00:00
if self . details [ " show_missing " ] is True :
2021-02-24 06:44:06 +00:00
logger . info ( f " { collection_name } Collection | ? | { title } (TVDB: { missing_id } ) " )
2021-02-28 08:36:40 +00:00
elif self . details [ " show_filtered " ] is True :
2021-03-01 20:59:10 +00:00
logger . info ( f " { collection_name } Collection | X | { title } (TVDb: { missing_id } ) " )
2021-02-24 06:44:06 +00:00
logger . info ( f " { len ( missing_shows_with_names ) } Show { ' s ' if len ( missing_shows_with_names ) > 1 else ' ' } Missing " )
2021-02-20 06:41:40 +00:00
if self . details [ " save_missing " ] is True :
2021-02-22 02:13:27 +00:00
self . library . add_missing ( collection_name , missing_shows_with_names , False )
2021-05-02 04:22:48 +00:00
if ( self . add_to_sonarr and self . library . Sonarr ) or self . run_again :
missing_tvdb_ids = [ missing_id for title , missing_id in missing_shows_with_names ]
if self . add_to_sonarr and self . library . Sonarr :
try :
self . library . Sonarr . add_tvdb ( missing_tvdb_ids , * * self . sonarr_options )
except Failed as e :
logger . error ( e )
if self . run_again :
self . missing_shows . extend ( missing_tvdb_ids )
2021-02-20 05:41:45 +00:00
if self . sync and items_found > 0 :
logger . info ( " " )
count_removed = 0
2021-02-24 06:42:58 +00:00
for ratingKey , item in rating_key_map . items ( ) :
2021-02-20 05:41:45 +00:00
if item is not None :
2021-02-24 06:44:06 +00:00
logger . info ( f " { collection_name } Collection | - | { item . title } " )
2021-05-02 04:22:48 +00:00
if self . smart_label_collection :
self . library . query_data ( item . removeLabel , collection_name )
else :
self . library . query_data ( item . removeCollection , collection_name )
2021-02-20 05:41:45 +00:00
count_removed + = 1
2021-02-24 06:44:06 +00:00
logger . info ( f " { count_removed } { ' Movie ' if self . library . is_movie else ' Show ' } { ' s ' if count_removed == 1 else ' ' } Removed " )
2021-02-20 05:41:45 +00:00
logger . info ( " " )
def update_details ( self , collection ) :
2021-05-02 04:22:48 +00:00
if self . smart_url and self . smart_url != self . library . smart_filter ( collection ) :
self . library . update_smart_collection ( collection , self . smart_url )
logger . info ( f " Detail: Smart Filter updated to { self . smart_url } " )
2021-02-20 05:41:45 +00:00
edits = { }
2021-02-28 23:06:04 +00:00
def get_summary ( summary_method , summaries ) :
2021-03-01 04:16:08 +00:00
logger . info ( f " Detail: { summary_method } updated Collection Summary " )
2021-02-28 23:06:04 +00:00
return summaries [ summary_method ]
if " summary " in self . summaries : summary = get_summary ( " summary " , self . summaries )
elif " tmdb_description " in self . summaries : summary = get_summary ( " tmdb_description " , self . summaries )
2021-03-08 21:54:19 +00:00
elif " letterboxd_description " in self . summaries : summary = get_summary ( " letterboxd_description " , self . summaries )
2021-02-28 23:06:04 +00:00
elif " tmdb_summary " in self . summaries : summary = get_summary ( " tmdb_summary " , self . summaries )
2021-03-08 19:53:05 +00:00
elif " tvdb_summary " in self . summaries : summary = get_summary ( " tvdb_summary " , self . summaries )
2021-02-28 23:06:04 +00:00
elif " tmdb_biography " in self . summaries : summary = get_summary ( " tmdb_biography " , self . summaries )
elif " tmdb_person " in self . summaries : summary = get_summary ( " tmdb_person " , self . summaries )
elif " tmdb_collection_details " in self . summaries : summary = get_summary ( " tmdb_collection_details " , self . summaries )
2021-03-08 15:25:35 +00:00
elif " trakt_list_details " in self . summaries : summary = get_summary ( " trakt_list_details " , self . summaries )
2021-02-28 23:06:04 +00:00
elif " tmdb_list_details " in self . summaries : summary = get_summary ( " tmdb_list_details " , self . summaries )
2021-03-02 05:28:32 +00:00
elif " tmdb_actor_details " in self . summaries : summary = get_summary ( " tmdb_actor_details " , self . summaries )
elif " tmdb_crew_details " in self . summaries : summary = get_summary ( " tmdb_crew_details " , self . summaries )
elif " tmdb_director_details " in self . summaries : summary = get_summary ( " tmdb_director_details " , self . summaries )
elif " tmdb_producer_details " in self . summaries : summary = get_summary ( " tmdb_producer_details " , self . summaries )
elif " tmdb_writer_details " in self . summaries : summary = get_summary ( " tmdb_writer_details " , self . summaries )
2021-02-28 23:06:04 +00:00
elif " tmdb_movie_details " in self . summaries : summary = get_summary ( " tmdb_movie_details " , self . summaries )
2021-03-08 17:02:40 +00:00
elif " tvdb_movie_details " in self . summaries : summary = get_summary ( " tvdb_movie_details " , self . summaries )
elif " tvdb_show_details " in self . summaries : summary = get_summary ( " tvdb_show_details " , self . summaries )
2021-02-28 23:06:04 +00:00
elif " tmdb_show_details " in self . summaries : summary = get_summary ( " tmdb_show_details " , self . summaries )
else : summary = None
if summary :
2021-05-02 04:22:48 +00:00
if str ( summary ) != str ( collection . summary ) :
edits [ " summary.value " ] = summary
edits [ " summary.locked " ] = 1
2021-03-01 04:16:08 +00:00
2021-02-20 05:41:45 +00:00
if " sort_title " in self . details :
2021-05-02 04:22:48 +00:00
if str ( self . details [ " sort_title " ] ) != str ( collection . titleSort ) :
edits [ " titleSort.value " ] = self . details [ " sort_title " ]
edits [ " titleSort.locked " ] = 1
logger . info ( f " Detail: sort_title updated Collection Sort Title to { self . details [ ' sort_title ' ] } " )
2021-03-01 04:16:08 +00:00
2021-02-20 05:41:45 +00:00
if " content_rating " in self . details :
2021-05-02 04:22:48 +00:00
if str ( self . details [ " content_rating " ] ) != str ( collection . contentRating ) :
edits [ " contentRating.value " ] = self . details [ " content_rating " ]
edits [ " contentRating.locked " ] = 1
logger . info ( f " Detail: content_rating updated Collection Content Rating to { self . details [ ' content_rating ' ] } " )
2021-03-01 04:16:08 +00:00
2021-02-20 05:41:45 +00:00
if " collection_mode " in self . details :
2021-05-02 04:22:48 +00:00
if int ( collection . collectionMode ) not in plex . collection_mode_keys \
or plex . collection_mode_keys [ int ( collection . collectionMode ) ] != self . details [ " collection_mode " ] :
self . library . collection_mode_query ( collection , self . details [ " collection_mode " ] )
logger . info ( f " Detail: collection_mode updated Collection Mode to { self . details [ ' collection_mode ' ] } " )
2021-03-01 04:16:08 +00:00
2021-02-20 05:41:45 +00:00
if " collection_order " in self . details :
2021-05-02 04:22:48 +00:00
if int ( collection . collectionSort ) not in plex . collection_order_keys \
or plex . collection_order_keys [ int ( collection . collectionSort ) ] != self . details [ " collection_order " ] :
self . library . collection_order_query ( collection , self . details [ " collection_order " ] )
logger . info ( f " Detail: collection_order updated Collection Order to { self . details [ ' collection_order ' ] } " )
2021-02-20 05:41:45 +00:00
2021-04-12 04:01:01 +00:00
if " label " in self . details or " label.sync " in self . details :
2021-02-20 17:05:11 +00:00
item_labels = [ label . tag for label in collection . labels ]
2021-04-12 04:01:01 +00:00
labels = util . get_list ( self . details [ " label " if " label " in self . details else " label.sync " ] )
if " label.sync " in self . details :
2021-02-24 06:42:58 +00:00
for label in ( la for la in item_labels if la not in labels ) :
2021-05-02 04:22:48 +00:00
self . library . query_data ( collection . removeLabel , label )
2021-02-24 06:44:06 +00:00
logger . info ( f " Detail: Label { label } removed " )
2021-02-24 06:42:58 +00:00
for label in ( la for la in labels if la not in item_labels ) :
2021-05-02 04:22:48 +00:00
self . library . query_data ( collection . addLabel , label )
2021-02-24 06:44:06 +00:00
logger . info ( f " Detail: Label { label } added " )
2021-02-20 17:05:11 +00:00
2021-04-21 12:49:27 +00:00
if len ( self . item_details ) > 0 :
labels = None
if " item_label " in self . item_details or " item_label.sync " in self . item_details :
labels = util . get_list ( self . item_details [ " item_label " if " item_label " in self . item_details else " item_label.sync " ] )
2021-05-02 04:22:48 +00:00
for item in self . library . get_collection_items ( collection , self . smart_label_collection ) :
2021-04-21 12:49:27 +00:00
if labels is not None :
item_labels = [ label . tag for label in item . labels ]
if " item_label.sync " in self . item_details :
for label in ( la for la in item_labels if la not in labels ) :
2021-05-02 04:22:48 +00:00
self . library . query_data ( item . removeLabel , label )
2021-04-21 12:49:27 +00:00
logger . info ( f " Detail: Label { label } removed from { item . title } " )
for label in ( la for la in labels if la not in item_labels ) :
2021-05-02 04:22:48 +00:00
self . library . query_data ( item . addLabel , label )
2021-04-21 12:49:27 +00:00
logger . info ( f " Detail: Label { label } added to { item . title } " )
advance_edits = { }
for method_name , method_data in self . item_details . items ( ) :
if method_name in plex . item_advance_keys :
key , options = plex . item_advance_keys [ method_name ]
if getattr ( item , key ) != options [ method_data ] :
advance_edits [ key ] = options [ method_data ]
2021-04-21 21:10:34 +00:00
self . library . edit_item ( item , item . title , " Movie " if self . library . is_movie else " Show " , advance_edits , advanced = True )
2021-04-21 12:49:27 +00:00
2021-03-01 04:16:08 +00:00
if len ( edits ) > 0 :
logger . debug ( edits )
2021-05-03 04:10:12 +00:00
self . library . edit_query ( collection , edits )
2021-03-01 04:16:08 +00:00
logger . info ( " Details: have been updated " )
2021-02-20 05:41:45 +00:00
if self . library . asset_directory :
name_mapping = self . name
if " name_mapping " in self . details :
2021-02-28 23:06:04 +00:00
if self . details [ " name_mapping " ] : name_mapping = self . details [ " name_mapping " ]
else : logger . error ( " Collection Error: name_mapping attribute is blank " )
2021-02-21 08:13:07 +00:00
for ad in self . library . asset_directory :
2021-02-24 06:44:06 +00:00
path = os . path . join ( ad , f " { name_mapping } " )
2021-04-21 21:29:36 +00:00
if self . library . asset_folders :
2021-04-30 16:11:48 +00:00
if not os . path . isdir ( path ) :
continue
2021-05-02 07:40:39 +00:00
poster_filter = os . path . join ( ad , name_mapping , " poster.* " )
background_filter = os . path . join ( ad , name_mapping , " background.* " )
2021-04-21 21:29:36 +00:00
else :
2021-05-02 07:40:39 +00:00
poster_filter = os . path . join ( ad , f " { name_mapping } .* " )
background_filter = os . path . join ( ad , f " { name_mapping } _background.* " )
matches = glob . glob ( poster_filter )
2021-02-21 08:13:07 +00:00
if len ( matches ) > 0 :
2021-04-21 21:29:36 +00:00
self . posters [ " asset_directory " ] = os . path . abspath ( matches [ 0 ] )
2021-05-02 07:40:39 +00:00
matches = glob . glob ( background_filter )
if len ( matches ) > 0 :
self . backgrounds [ " asset_directory " ] = os . path . abspath ( matches [ 0 ] )
for item in self . library . query ( collection . items ) :
self . library . update_item_from_assets ( item , dirs = [ path ] )
2021-02-21 08:13:07 +00:00
2021-02-28 23:06:04 +00:00
def set_image ( image_method , images , is_background = False ) :
2021-03-30 05:49:10 +00:00
message = f " { ' background ' if is_background else ' poster ' } to [ { ' File ' if image_method in image_file_details else ' URL ' } ] { images [ image_method ] } "
try :
2021-05-02 04:22:48 +00:00
self . library . upload_image ( collection , images [ image_method ] , poster = not is_background , url = image_method not in image_file_details )
2021-03-30 05:49:10 +00:00
logger . info ( f " Detail: { image_method } updated collection { message } " )
except BadRequest :
logger . error ( f " Detail: { image_method } failed to update { message } " )
2021-02-28 23:06:04 +00:00
2021-04-08 00:20:06 +00:00
if len ( self . posters ) > 0 :
logger . info ( " " )
2021-02-28 23:06:04 +00:00
if len ( self . posters ) > 1 :
logger . info ( f " { len ( self . posters ) } posters found: " )
for p in self . posters :
logger . info ( f " Method: { p } Poster: { self . posters [ p ] } " )
if " url_poster " in self . posters : set_image ( " url_poster " , self . posters )
elif " file_poster " in self . posters : set_image ( " file_poster " , self . posters )
elif " tmdb_poster " in self . posters : set_image ( " tmdb_poster " , self . posters )
elif " tmdb_profile " in self . posters : set_image ( " tmdb_profile " , self . posters )
2021-03-08 19:53:05 +00:00
elif " tvdb_poster " in self . posters : set_image ( " tvdb_poster " , self . posters )
2021-02-28 23:06:04 +00:00
elif " asset_directory " in self . posters : set_image ( " asset_directory " , self . posters )
2021-03-02 05:28:32 +00:00
elif " tmdb_person " in self . posters : set_image ( " tmdb_person " , self . posters )
2021-03-06 16:44:56 +00:00
elif " tmdb_collection_details " in self . posters : set_image ( " tmdb_collection_details " , self . posters )
2021-03-02 05:28:32 +00:00
elif " tmdb_actor_details " in self . posters : set_image ( " tmdb_actor_details " , self . posters )
elif " tmdb_crew_details " in self . posters : set_image ( " tmdb_crew_details " , self . posters )
elif " tmdb_director_details " in self . posters : set_image ( " tmdb_director_details " , self . posters )
elif " tmdb_producer_details " in self . posters : set_image ( " tmdb_producer_details " , self . posters )
elif " tmdb_writer_details " in self . posters : set_image ( " tmdb_writer_details " , self . posters )
2021-03-06 16:44:56 +00:00
elif " tmdb_movie_details " in self . posters : set_image ( " tmdb_movie_details " , self . posters )
2021-03-08 17:02:40 +00:00
elif " tvdb_movie_details " in self . posters : set_image ( " tvdb_movie_details " , self . posters )
elif " tvdb_show_details " in self . posters : set_image ( " tvdb_show_details " , self . posters )
2021-03-06 16:44:56 +00:00
elif " tmdb_show_details " in self . posters : set_image ( " tmdb_show_details " , self . posters )
2021-02-28 23:06:04 +00:00
else : logger . info ( " No poster to update " )
2021-04-08 00:20:06 +00:00
if len ( self . backgrounds ) > 0 :
logger . info ( " " )
2021-02-28 23:06:04 +00:00
if len ( self . backgrounds ) > 1 :
logger . info ( f " { len ( self . backgrounds ) } backgrounds found: " )
for b in self . backgrounds :
logger . info ( f " Method: { b } Background: { self . backgrounds [ b ] } " )
if " url_background " in self . backgrounds : set_image ( " url_background " , self . backgrounds , is_background = True )
2021-03-08 19:53:05 +00:00
elif " file_background " in self . backgrounds : set_image ( " file_background " , self . backgrounds , is_background = True )
elif " tmdb_background " in self . backgrounds : set_image ( " tmdb_background " , self . backgrounds , is_background = True )
elif " tvdb_background " in self . backgrounds : set_image ( " tvdb_background " , self . backgrounds , is_background = True )
2021-02-28 23:06:04 +00:00
elif " asset_directory " in self . backgrounds : set_image ( " asset_directory " , self . backgrounds , is_background = True )
2021-03-06 16:44:56 +00:00
elif " tmdb_collection_details " in self . backgrounds : set_image ( " tmdb_collection_details " , self . backgrounds , is_background = True )
elif " tmdb_movie_details " in self . backgrounds : set_image ( " tmdb_movie_details " , self . backgrounds , is_background = True )
2021-03-08 17:02:40 +00:00
elif " tvdb_movie_details " in self . backgrounds : set_image ( " tvdb_movie_details " , self . backgrounds , is_background = True )
elif " tvdb_show_details " in self . backgrounds : set_image ( " tvdb_show_details " , self . backgrounds , is_background = True )
2021-03-06 16:44:56 +00:00
elif " tmdb_show_details " in self . backgrounds : set_image ( " tmdb_show_details " , self . backgrounds , is_background = True )
2021-03-01 20:59:10 +00:00
else : logger . info ( " No background to update " )
2021-03-05 16:04:28 +00:00
def run_collections_again ( self , collection_obj , movie_map , show_map ) :
2021-05-02 04:22:48 +00:00
name , collection_items = self . library . get_collection_name_and_items ( collection_obj , self . smart_label_collection )
2021-04-22 03:13:40 +00:00
rating_keys = [ ]
for mm in self . missing_movies :
if mm in movie_map :
rating_keys . extend ( movie_map [ mm ] )
2021-03-05 16:04:28 +00:00
if self . library . is_show :
2021-04-22 03:13:40 +00:00
for sm in self . missing_shows :
if sm in show_map :
rating_keys . extend ( show_map [ sm ] )
2021-03-01 20:59:10 +00:00
if len ( rating_keys ) > 0 :
for rating_key in rating_keys :
try :
2021-03-05 16:04:28 +00:00
current = self . library . fetchItem ( int ( rating_key ) )
2021-03-01 20:59:10 +00:00
except ( BadRequest , NotFound ) :
logger . error ( f " Plex Error: Item { rating_key } not found " )
continue
if current in collection_items :
logger . info ( f " { name } Collection | = | { current . title } " )
2021-05-02 04:22:48 +00:00
elif self . smart_label_collection :
self . library . query_data ( current . addLabel , name )
2021-03-01 20:59:10 +00:00
else :
2021-05-02 04:22:48 +00:00
self . library . query_data ( current . addCollection , name )
2021-03-01 20:59:10 +00:00
logger . info ( f " { name } Collection | + | { current . title } " )
2021-03-05 16:04:28 +00:00
logger . info ( f " { len ( rating_keys ) } { ' Movie ' if self . library . is_movie else ' Show ' } { ' s ' if len ( rating_keys ) > 1 else ' ' } Processed " )
2021-03-01 20:59:10 +00:00
if len ( self . missing_movies ) > 0 :
logger . info ( " " )
for missing_id in self . missing_movies :
if missing_id not in movie_map :
try :
movie = self . config . TMDb . get_movie ( missing_id )
except Failed as e :
logger . error ( e )
continue
if self . details [ " show_missing " ] is True :
logger . info ( f " { name } Collection | ? | { movie . title } (TMDb: { missing_id } ) " )
logger . info ( " " )
logger . info ( f " { len ( self . missing_movies ) } Movie { ' s ' if len ( self . missing_movies ) > 1 else ' ' } Missing " )
2021-03-05 16:04:28 +00:00
if len ( self . missing_shows ) > 0 and self . library . is_show :
2021-03-01 20:59:10 +00:00
logger . info ( " " )
for missing_id in self . missing_shows :
if missing_id not in show_map :
try :
2021-03-08 19:53:05 +00:00
title = str ( self . config . TVDb . get_series ( self . library . Plex . language , missing_id ) . title . encode ( " ascii " , " replace " ) . decode ( ) )
2021-03-01 20:59:10 +00:00
except Failed as e :
logger . error ( e )
continue
if self . details [ " show_missing " ] is True :
logger . info ( f " { name } Collection | ? | { title } (TVDb: { missing_id } ) " )
logger . info ( f " { len ( self . missing_shows ) } Show { ' s ' if len ( self . missing_shows ) > 1 else ' ' } Missing " )
