2022-04-13 04:30:59 +00:00
import os , re
2022-05-05 13:21:17 +00:00
from datetime import datetime
2023-02-03 02:51:00 +00:00
from modules import plex , util , anidb
2022-10-17 16:06:32 +00:00
from modules . util import Failed , LimitReached , YAML
2023-12-01 21:15:10 +00:00
from plexapi . exceptions import NotFound
2024-03-06 22:04:30 +00:00
from plexapi . video import Movie , Show
2022-04-13 04:30:59 +00:00
logger = util . logger
2022-06-28 20:35:29 +00:00
meta_operations = [
" mass_audience_rating_update " , " mass_user_rating_update " , " mass_critic_rating_update " ,
" mass_episode_audience_rating_update " , " mass_episode_user_rating_update " , " mass_episode_critic_rating_update " ,
2022-10-26 06:57:39 +00:00
" mass_genre_update " , " mass_content_rating_update " , " mass_originally_available_update " , " mass_original_title_update " ,
2023-01-25 07:43:25 +00:00
" mass_poster_update " , " mass_background_update " , " mass_studio_update "
2022-06-28 20:35:29 +00:00
]
2023-12-01 21:15:10 +00:00
name_display = {
" audienceRating " : " Audience Rating " ,
" rating " : " Critic Rating " ,
" userRating " : " User Rating " ,
" originallyAvailableAt " : " Originally Available Date " ,
" contentRating " : " Content Rating "
}
2022-06-28 20:35:29 +00:00
2022-04-13 04:30:59 +00:00
class Operations :
def __init__ ( self , config , library ) :
self . config = config
self . library = library
def run_operations ( self ) :
2022-05-05 13:21:17 +00:00
operation_start = datetime . now ( )
2022-04-13 04:30:59 +00:00
logger . info ( " " )
logger . separator ( f " { self . library . name } Library Operations " )
logger . info ( " " )
logger . debug ( f " Assets For All: { self . library . assets_for_all } " )
2022-11-08 20:59:23 +00:00
logger . debug ( f " Delete Collections: { self . library . delete_collections } " )
2022-08-04 13:34:02 +00:00
logger . debug ( f " Show Unmanaged Collections: { self . library . show_unmanaged } " )
2022-11-09 02:21:46 +00:00
logger . debug ( f " Show Unconfigured Collections: { self . library . show_unconfigured } " )
2022-04-13 04:30:59 +00:00
logger . debug ( f " Mass Genre Update: { self . library . mass_genre_update } " )
logger . debug ( f " Mass Audience Rating Update: { self . library . mass_audience_rating_update } " )
logger . debug ( f " Mass Critic Rating Update: { self . library . mass_critic_rating_update } " )
2022-06-09 14:20:43 +00:00
logger . debug ( f " Mass User Rating Update: { self . library . mass_user_rating_update } " )
logger . debug ( f " Mass Episode Audience Rating Update: { self . library . mass_episode_audience_rating_update } " )
logger . debug ( f " Mass Episode Critic Rating Update: { self . library . mass_episode_critic_rating_update } " )
logger . debug ( f " Mass Episode User Rating Update: { self . library . mass_episode_user_rating_update } " )
2022-04-13 04:30:59 +00:00
logger . debug ( f " Mass Content Rating Update: { self . library . mass_content_rating_update } " )
2022-09-23 15:01:38 +00:00
logger . debug ( f " Mass Original Title Update: { self . library . mass_original_title_update } " )
2022-04-13 04:30:59 +00:00
logger . debug ( f " Mass Originally Available Update: { self . library . mass_originally_available_update } " )
logger . debug ( f " Mass IMDb Parental Labels: { self . library . mass_imdb_parental_labels } " )
2022-10-26 06:57:39 +00:00
logger . debug ( f " Mass Poster Update: { self . library . mass_poster_update } " )
logger . debug ( f " Mass Background Update: { self . library . mass_background_update } " )
2022-04-13 04:30:59 +00:00
logger . debug ( f " Mass Collection Mode Update: { self . library . mass_collection_mode } " )
logger . debug ( f " Split Duplicates: { self . library . split_duplicates } " )
logger . debug ( f " Radarr Add All Existing: { self . library . radarr_add_all_existing } " )
logger . debug ( f " Radarr Remove by Tag: { self . library . radarr_remove_by_tag } " )
logger . debug ( f " Sonarr Add All Existing: { self . library . sonarr_add_all_existing } " )
logger . debug ( f " Sonarr Remove by Tag: { self . library . sonarr_remove_by_tag } " )
logger . debug ( f " Update Blank Track Titles: { self . library . update_blank_track_titles } " )
logger . debug ( f " Update Remove Title Parentheses: { self . library . remove_title_parentheses } " )
logger . debug ( f " Genre Mapper: { self . library . genre_mapper } " )
logger . debug ( f " Content Rating Mapper: { self . library . content_rating_mapper } " )
logger . debug ( f " Metadata Backup: { self . library . metadata_backup } " )
logger . debug ( f " Item Operation: { self . library . items_library_operation } " )
logger . debug ( " " )
if self . library . split_duplicates :
items = self . library . search ( * * { " duplicate " : True } )
for item in items :
item . split ( )
logger . info ( f " { item . title [ : 25 ] : <25 } | Splitting " )
if self . library . update_blank_track_titles :
2022-07-26 19:58:53 +00:00
tracks = self . library . get_all ( builder_level = " track " )
2022-04-13 04:30:59 +00:00
num_edited = 0
for i , track in enumerate ( tracks , 1 ) :
logger . ghost ( f " Processing Track: { i } / { len ( tracks ) } { track . title } " )
if not track . title and track . titleSort :
track . editTitle ( track . titleSort )
num_edited + = 1
logger . info ( f " Track: { track . titleSort } was updated with sort title " )
logger . info ( f " { len ( tracks ) } Tracks Processed; { num_edited } Blank Track Titles Updated " )
if self . library . items_library_operation :
2022-04-20 21:30:31 +00:00
items = self . library . get_all ( )
2022-04-13 04:30:59 +00:00
radarr_adds = [ ]
sonarr_adds = [ ]
2023-12-01 21:15:10 +00:00
label_edits = { " add " : { } , " remove " : { } }
rating_edits = { " audienceRating " : { } , " rating " : { } , " userRating " : { } }
genre_edits = { " add " : { } , " remove " : { } }
content_edits = { }
studio_edits = { }
available_edits = { }
remove_edits = { }
reset_edits = { }
lock_edits = { }
unlock_edits = { }
ep_rating_edits = { " audienceRating " : { } , " rating " : { } , " userRating " : { } }
ep_remove_edits = { }
ep_reset_edits = { }
ep_lock_edits = { }
ep_unlock_edits = { }
2022-04-13 04:30:59 +00:00
reverse_anidb = { }
2022-09-22 20:24:58 +00:00
for k , v in self . library . anidb_map . items ( ) :
reverse_anidb [ v ] = k
reverse_mal = { }
for k , v in self . library . mal_map . items ( ) :
reverse_mal [ v ] = k
2022-04-13 04:30:59 +00:00
2022-05-16 03:08:14 +00:00
if self . library . assets_for_all and not self . library . asset_directory :
logger . error ( " Asset Error: No Asset Directory for Assets For All " )
2022-04-13 04:30:59 +00:00
for i , item in enumerate ( items , 1 ) :
2023-12-01 21:15:10 +00:00
logger . info ( " " )
logger . info ( f " Processing: { i } / { len ( items ) } { item . title } " )
2022-04-22 02:57:26 +00:00
try :
2022-04-22 15:24:44 +00:00
item = self . library . reload ( item )
2022-04-22 02:57:26 +00:00
except Failed as e :
logger . error ( e )
continue
2023-03-01 22:10:46 +00:00
current_labels = [ la . tag for la in self . library . item_labels ( item ) ] if self . library . label_operations else [ ]
2022-04-23 19:57:33 +00:00
2022-06-15 14:45:30 +00:00
if self . library . assets_for_all and self . library . asset_directory :
2022-06-14 15:46:50 +00:00
self . library . find_and_upload_assets ( item , current_labels )
2022-04-23 19:57:33 +00:00
2022-09-21 19:17:11 +00:00
locked_fields = [ f . name for f in item . fields if f . locked ]
2022-04-13 04:30:59 +00:00
tmdb_id , tvdb_id , imdb_id = self . library . get_ids ( item )
2023-12-01 21:15:10 +00:00
item_edits = " "
2022-04-13 04:30:59 +00:00
if self . library . remove_title_parentheses :
if not any ( [ f . name == " title " and f . locked for f in item . fields ] ) and item . title . endswith ( " ) " ) :
new_title = re . sub ( " \\ ( \\ w+ \\ )$ " , " " , item . title )
item . editTitle ( new_title )
2023-12-01 21:15:10 +00:00
item_edits + = f " \n Updated Title: { item . title [ : 25 ] : <25 } | { new_title } "
2022-04-13 04:30:59 +00:00
if self . library . mass_imdb_parental_labels :
try :
2023-12-01 21:15:10 +00:00
if self . library . mass_imdb_parental_labels == " remove " :
parental_labels = [ ]
else :
parental_guide = self . config . IMDb . parental_guide ( imdb_id )
parental_labels = [ f " { k . capitalize ( ) } : { v } " for k , v in parental_guide . items ( ) if v not in util . parental_levels [ self . library . mass_imdb_parental_labels ] ]
2022-04-23 19:57:33 +00:00
add_labels = [ la for la in parental_labels if la not in current_labels ]
remove_labels = [ la for la in current_labels if la in util . parental_labels and la not in parental_labels ]
2023-12-01 21:15:10 +00:00
for label_list , edit_type in [ ( add_labels , " add " ) , ( remove_labels , " remove " ) ] :
if label_list :
for label in label_list :
if label not in label_edits [ edit_type ] :
label_edits [ edit_type ] [ label ] = [ ]
label_edits [ edit_type ] [ label ] . append ( item . ratingKey )
item_edits + = f " \n { edit_type . capitalize ( ) } IMDb Parental Labels (Batched) | { ' , ' . join ( label_list ) } "
2022-04-13 04:30:59 +00:00
except Failed :
pass
2022-06-01 04:50:38 +00:00
if item . locations :
path = os . path . dirname ( str ( item . locations [ 0 ] ) ) if self . library . is_movie else str ( item . locations [ 0 ] )
if self . library . Radarr and self . library . radarr_add_all_existing and tmdb_id :
path = path . replace ( self . library . Radarr . plex_path , self . library . Radarr . radarr_path )
path = path [ : - 1 ] if path . endswith ( ( ' / ' , ' \\ ' ) ) else path
radarr_adds . append ( ( tmdb_id , path ) )
if self . library . Sonarr and self . library . sonarr_add_all_existing and tvdb_id :
path = path . replace ( self . library . Sonarr . plex_path , self . library . Sonarr . sonarr_path )
path = path [ : - 1 ] if path . endswith ( ( " / " , " \\ " ) ) else path
sonarr_adds . append ( ( tvdb_id , path ) )
2022-04-13 04:30:59 +00:00
2024-01-26 05:10:35 +00:00
_trakt_ratings = None
def trakt_ratings ( ) :
nonlocal _trakt_ratings
2024-03-27 13:33:38 +00:00
if _trakt_ratings is None :
2024-01-26 05:10:35 +00:00
_trakt_ratings = self . config . Trakt . user_ratings ( self . library . is_movie )
if not _trakt_ratings :
raise Failed
return _trakt_ratings
_tmdb_obj = None
def tmdb_obj ( ) :
nonlocal _tmdb_obj
if _tmdb_obj is None :
_tmdb_obj = False
2022-06-09 14:20:43 +00:00
try :
2024-01-26 05:10:35 +00:00
_item = self . config . TMDb . get_item ( item , tmdb_id , tvdb_id , imdb_id , is_movie = self . library . is_movie )
if _item :
_tmdb_obj = _item
except Failed as err :
logger . error ( str ( err ) )
if not _tmdb_obj :
raise Failed
return _tmdb_obj
_omdb_obj = None
def omdb_obj ( ) :
nonlocal _omdb_obj
if _omdb_obj is None :
_omdb_obj = False
if self . config . OMDb . limit is not False :
logger . error ( " Daily OMDb Limit Reached " )
elif not imdb_id :
logger . info ( f " No IMDb ID for Guid: { item . guid } " )
else :
try :
_omdb_obj = self . config . OMDb . get_omdb ( imdb_id )
except Failed as err :
logger . error ( str ( err ) )
except Exception :
logger . error ( f " IMDb ID: { imdb_id } " )
raise
if not _omdb_obj :
raise Failed
return _omdb_obj
_tvdb_obj = None
def tvdb_obj ( ) :
nonlocal _tvdb_obj
if _tvdb_obj is None :
_tvdb_obj = False
if tvdb_id :
try :
_tvdb_obj = self . config . TVDb . get_tvdb_obj ( tvdb_id , is_movie = self . library . is_movie )
except Failed as err :
logger . error ( str ( err ) )
else :
logger . info ( f " No TVDb ID for Guid: { item . guid } " )
if not _tvdb_obj :
raise Failed
return _tvdb_obj
_mdb_obj = None
def mdb_obj ( ) :
nonlocal _mdb_obj
if _mdb_obj is None :
_mdb_obj = False
if self . config . Mdblist . limit is False :
if self . library . is_show and tvdb_id :
try :
_mdb_obj = self . config . Mdblist . get_series ( tvdb_id )
except LimitReached as err :
logger . debug ( err )
except Failed as err :
logger . error ( str ( err ) )
except Exception :
logger . trace ( f " TVDb ID: { tvdb_id } " )
raise
if self . library . is_movie and tmdb_id :
try :
_mdb_obj = self . config . Mdblist . get_movie ( tmdb_id )
except LimitReached as err :
logger . debug ( err )
except Failed as err :
logger . error ( str ( err ) )
except Exception :
logger . trace ( f " TMDb ID: { tmdb_id } " )
raise
if imdb_id and not _mdb_obj :
try :
_mdb_obj = self . config . Mdblist . get_imdb ( imdb_id )
except LimitReached as err :
logger . debug ( err )
except Failed as err :
logger . error ( str ( err ) )
except Exception :
logger . trace ( f " IMDb ID: { imdb_id } " )
raise
if not _mdb_obj :
logger . warning ( f " No MdbItem for { item . title } (Guid: { item . guid } ) " )
if not _mdb_obj :
raise Failed
return _mdb_obj
2022-04-13 04:30:59 +00:00
2024-01-26 05:10:35 +00:00
anidb_id = None
def get_anidb_id ( ) :
2022-04-13 04:30:59 +00:00
if item . ratingKey in reverse_anidb :
2024-01-26 05:10:35 +00:00
return reverse_anidb [ item . ratingKey ]
2022-04-13 04:30:59 +00:00
elif tvdb_id in self . config . Convert . _tvdb_to_anidb :
2024-01-26 05:10:35 +00:00
return self . config . Convert . _tvdb_to_anidb [ tvdb_id ]
2022-04-13 04:30:59 +00:00
elif imdb_id in self . config . Convert . _imdb_to_anidb :
2024-01-26 05:10:35 +00:00
return self . config . Convert . _imdb_to_anidb [ imdb_id ]
2022-04-13 04:30:59 +00:00
else :
2024-01-26 05:10:35 +00:00
return False
_anidb_obj = None
def anidb_obj ( ) :
nonlocal anidb_id , _anidb_obj
if _anidb_obj is None :
_anidb_obj = False
if anidb_id is None :
anidb_id = get_anidb_id ( )
2022-11-10 19:24:38 +00:00
if anidb_id :
try :
2024-01-26 05:10:35 +00:00
_anidb_obj = self . config . AniDB . get_anime ( anidb_id )
except Failed as err :
logger . error ( str ( err ) )
2022-11-10 19:24:38 +00:00
else :
logger . warning ( f " No AniDB ID for Guid: { item . guid } " )
2024-01-26 05:10:35 +00:00
if not _anidb_obj :
raise Failed
return _anidb_obj
_mal_obj = None
def mal_obj ( ) :
nonlocal anidb_id , _mal_obj
if _mal_obj is None :
_mal_obj = False
if anidb_id is None :
anidb_id = get_anidb_id ( )
mal_id = None
2022-09-22 20:24:58 +00:00
if item . ratingKey in reverse_mal :
mal_id = reverse_mal [ item . ratingKey ]
2023-01-10 02:11:11 +00:00
elif not anidb_id :
2023-04-20 17:49:18 +00:00
logger . warning ( f " Convert Warning: No AniDB ID to Convert to MyAnimeList ID for Guid: { item . guid } " )
2023-01-10 02:11:11 +00:00
elif anidb_id not in self . config . Convert . _anidb_to_mal :
2023-04-20 17:49:18 +00:00
logger . warning ( f " Convert Warning: No MyAnimeList Found for AniDB ID: { anidb_id } of Guid: { item . guid } " )
2022-09-22 20:24:58 +00:00
else :
mal_id = self . config . Convert . _anidb_to_mal [ anidb_id ]
if mal_id :
try :
2024-01-26 05:10:35 +00:00
_mal_obj = self . config . MyAnimeList . get_anime ( mal_id )
except Failed as err :
logger . error ( str ( err ) )
if not _mal_obj :
raise Failed
return _mal_obj
2022-04-13 04:30:59 +00:00
2023-12-01 21:15:10 +00:00
for attribute , item_attr in [
( self . library . mass_audience_rating_update , " audienceRating " ) ,
( self . library . mass_critic_rating_update , " rating " ) ,
( self . library . mass_user_rating_update , " userRating " )
] :
if attribute :
current = getattr ( item , item_attr )
2024-01-26 05:10:35 +00:00
for option in attribute :
if option in [ " lock " , " remove " ] :
if option == " remove " and current :
if item_attr not in remove_edits :
remove_edits [ item_attr ] = [ ]
remove_edits [ item_attr ] . append ( item . ratingKey )
item_edits + = f " \n Remove { name_display [ item_attr ] } (Batched) "
elif item_attr not in locked_fields :
if item_attr not in lock_edits :
lock_edits [ item_attr ] = [ ]
lock_edits [ item_attr ] . append ( item . ratingKey )
item_edits + = f " \n Lock { name_display [ item_attr ] } (Batched) "
break
elif option in [ " unlock " , " reset " ] :
if option == " reset " and current :
if item_attr not in reset_edits :
reset_edits [ item_attr ] = [ ]
reset_edits [ item_attr ] . append ( item . ratingKey )
item_edits + = f " \n Reset { name_display [ item_attr ] } (Batched) "
elif item_attr in locked_fields :
if item_attr not in unlock_edits :
unlock_edits [ item_attr ] = [ ]
unlock_edits [ item_attr ] . append ( item . ratingKey )
item_edits + = f " \n Unlock { name_display [ item_attr ] } (Batched) "
break
2023-12-01 21:15:10 +00:00
else :
2024-01-26 05:10:35 +00:00
try :
if option == " tmdb " :
found_rating = tmdb_obj ( ) . vote_average # noqa
elif option == " imdb " :
found_rating = self . config . IMDb . get_rating ( imdb_id )
2024-03-27 13:33:38 +00:00
elif option == " omdb " :
2024-03-20 15:34:32 +00:00
found_rating = omdb_obj ( ) . imdb_rating # noqa
2024-01-26 05:10:35 +00:00
elif option == " trakt_user " :
_ratings = trakt_ratings ( )
_id = tmdb_id if self . library . is_movie else tvdb_id
if _id in _ratings :
found_rating = _ratings [ _id ]
else :
raise Failed
elif str ( option ) . startswith ( " mdb " ) :
mdb_item = mdb_obj ( )
if option == " mdb_average " :
found_rating = mdb_item . average / 10 if mdb_item . average else None # noqa
elif option == " mdb_imdb " :
found_rating = mdb_item . imdb_rating if mdb_item . imdb_rating else None # noqa
elif option == " mdb_metacritic " :
found_rating = mdb_item . metacritic_rating / 10 if mdb_item . metacritic_rating else None # noqa
elif option == " mdb_metacriticuser " :
found_rating = mdb_item . metacriticuser_rating if mdb_item . metacriticuser_rating else None # noqa
elif option == " mdb_trakt " :
found_rating = mdb_item . trakt_rating / 10 if mdb_item . trakt_rating else None # noqa
elif option == " mdb_tomatoes " :
found_rating = mdb_item . tomatoes_rating / 10 if mdb_item . tomatoes_rating else None # noqa
elif option == " mdb_tomatoesaudience " :
found_rating = mdb_item . tomatoesaudience_rating / 10 if mdb_item . tomatoesaudience_rating else None # noqa
elif option == " mdb_tmdb " :
found_rating = mdb_item . tmdb_rating / 10 if mdb_item . tmdb_rating else None # noqa
elif option == " mdb_letterboxd " :
found_rating = mdb_item . letterboxd_rating * 2 if mdb_item . letterboxd_rating else None # noqa
elif option == " mdb_myanimelist " :
found_rating = mdb_item . myanimelist_rating if mdb_item . myanimelist_rating else None # noqa
else :
found_rating = mdb_item . score / 10 if mdb_item . score else None # noqa
elif option == " anidb_rating " :
found_rating = anidb_obj ( ) . rating # noqa
elif option == " anidb_average " :
found_rating = anidb_obj ( ) . average # noqa
elif option == " anidb_score " :
found_rating = anidb_obj ( ) . score # noqa
elif option == " mal " :
found_rating = mal_obj ( ) . score # noqa
else :
found_rating = option
if not found_rating :
logger . info ( f " No { option } { name_display [ item_attr ] } Found " )
raise Failed
found_rating = f " { float ( found_rating ) : .1f } "
if str ( current ) != found_rating :
if found_rating not in rating_edits [ item_attr ] :
rating_edits [ item_attr ] [ found_rating ] = [ ]
rating_edits [ item_attr ] [ found_rating ] . append ( item . ratingKey )
item_edits + = f " \n Update { name_display [ item_attr ] } (Batched) | { found_rating } "
break
except Failed :
continue
2022-04-13 04:30:59 +00:00
if self . library . mass_genre_update or self . library . genre_mapper :
2024-03-06 19:31:50 +00:00
new_genres = [ ]
extra_option = None
2024-01-26 05:10:35 +00:00
if self . library . mass_genre_update :
2024-01-26 21:00:10 +00:00
for option in self . library . mass_genre_update :
2024-01-26 05:10:35 +00:00
if option in [ " lock " , " unlock " , " remove " , " reset " ] :
extra_option = option
break
try :
if option == " tmdb " :
new_genres = tmdb_obj ( ) . genres # noqa
elif option == " imdb " :
new_genres = self . config . IMDb . get_genres ( imdb_id )
elif option == " omdb " :
new_genres = omdb_obj ( ) . genres # noqa
elif option == " tvdb " :
new_genres = tvdb_obj ( ) . genres # noqa
2024-01-26 22:08:25 +00:00
elif str ( option ) in anidb . weights :
new_genres = [ str ( t ) . title ( ) for t , w in anidb_obj ( ) . tags . items ( ) if w > = anidb . weights [ str ( option ) ] ] # noqa
2024-01-26 05:10:35 +00:00
elif option == " mal " :
new_genres = mal_obj ( ) . genres # noqa
else :
new_genres = option
if not new_genres :
logger . info ( f " No { option } Genres Found " )
raise Failed
break
except Failed :
continue
2024-03-06 19:31:50 +00:00
item_genres = [ g . tag for g in item . genres ]
if not new_genres and extra_option not in [ " remove " , " reset " ] :
new_genres = item_genres
if self . library . genre_mapper :
mapped_genres = [ ]
for genre in new_genres :
if genre in self . library . genre_mapper :
if self . library . genre_mapper [ genre ] :
mapped_genres . append ( self . library . genre_mapper [ genre ] )
else :
mapped_genres . append ( genre )
new_genres = mapped_genres
_add = list ( set ( new_genres ) - set ( item_genres ) )
_remove = list ( set ( item_genres ) - set ( new_genres ) )
for genre_list , edit_type in [ ( _add , " add " ) , ( _remove , " remove " ) ] :
if genre_list :
for g in genre_list :
if g not in genre_edits [ edit_type ] :
genre_edits [ edit_type ] [ g ] = [ ]
genre_edits [ edit_type ] [ g ] . append ( item . ratingKey )
item_edits + = f " \n { edit_type . capitalize ( ) } Genres (Batched) | { ' , ' . join ( genre_list ) } "
if extra_option in [ " unlock " , " reset " ] and ( " genre " in locked_fields or _add or _remove ) :
if " genre " not in unlock_edits :
unlock_edits [ " genre " ] = [ ]
unlock_edits [ " genre " ] . append ( item . ratingKey )
item_edits + = " \n Unlock Genre (Batched) "
elif extra_option in [ " lock " , " remove " ] and " genre " not in locked_fields and not _add and not _remove :
if " genre " not in lock_edits :
lock_edits [ " genre " ] = [ ]
lock_edits [ " genre " ] . append ( item . ratingKey )
item_edits + = " \n Lock Genre (Batched) "
2022-04-13 04:30:59 +00:00
if self . library . mass_content_rating_update or self . library . content_rating_mapper :
2024-03-06 19:31:50 +00:00
new_rating = None
extra_option = None
2024-01-26 05:10:35 +00:00
if self . library . mass_content_rating_update :
2024-01-26 21:00:10 +00:00
for option in self . library . mass_content_rating_update :
2024-01-26 05:10:35 +00:00
if option in [ " lock " , " unlock " , " remove " , " reset " ] :
extra_option = option
break
try :
if option == " omdb " :
new_rating = omdb_obj ( ) . content_rating # noqa
elif option == " mdb " :
_rating = mdb_obj ( ) . content_rating # noqa
new_rating = _rating if _rating else None
elif str ( option ) . startswith ( " mdb_commonsense " ) :
_rating = mdb_obj ( ) . commonsense # noqa
if not _rating :
new_rating = None
elif option == " mdb_commonsense0 " :
new_rating = str ( _rating ) . rjust ( 2 , " 0 " )
else :
new_rating = _rating
elif str ( option ) . startswith ( " mdb_age_rating " ) :
_rating = mdb_obj ( ) . age_rating # noqa
if not _rating :
new_rating = None
elif option == " mdb_age_rating0 " :
new_rating = str ( _rating ) . rjust ( 2 , " 0 " )
else :
new_rating = _rating
elif option == " mal " :
new_rating = mal_obj ( ) . rating # noqa
else :
new_rating = option
if new_rating is None :
logger . info ( f " No { option } Content Rating Found " )
raise Failed
else :
new_rating = str ( new_rating )
break
except Failed :
continue
2022-11-15 07:00:59 +00:00
2024-03-06 19:31:50 +00:00
is_none = False
do_lock = False
do_unlock = False
current_rating = item . contentRating
if not new_rating :
new_rating = current_rating
if self . library . content_rating_mapper :
if new_rating in self . library . content_rating_mapper :
new_rating = self . library . content_rating_mapper [ new_rating ]
if not new_rating :
is_none = True
if extra_option == " reset " :
if current_rating :
if " contentRating " not in reset_edits :
reset_edits [ " contentRating " ] = [ ]
reset_edits [ " contentRating " ] . append ( item . ratingKey )
item_edits + = " \n Reset Content Rating (Batched) "
elif " contentRating " in locked_fields :
do_unlock = True
elif extra_option == " remove " or is_none :
if current_rating :
if " contentRating " not in remove_edits :
remove_edits [ " contentRating " ] = [ ]
remove_edits [ " contentRating " ] . append ( item . ratingKey )
item_edits + = " \n Remove Content Rating (Batched) "
elif " contentRating " not in locked_fields :
do_lock = True
elif new_rating and new_rating != current_rating :
if new_rating not in content_edits :
content_edits [ new_rating ] = [ ]
content_edits [ new_rating ] . append ( item . ratingKey )
item_edits + = f " \n Update Content Rating (Batched) | { new_rating } "
2024-01-26 05:10:35 +00:00
do_lock = False
2024-03-06 19:31:50 +00:00
if extra_option == " lock " or do_lock :
if " contentRating " not in lock_edits :
lock_edits [ " contentRating " ] = [ ]
lock_edits [ " contentRating " ] . append ( item . ratingKey )
item_edits + = " \n Lock Content Rating (Batched) "
elif extra_option == " unlock " or do_unlock :
if " contentRating " not in unlock_edits :
unlock_edits [ " contentRating " ] = [ ]
unlock_edits [ " contentRating " ] . append ( item . ratingKey )
item_edits + = " \n Unlock Content Rating (Batched) "
2022-09-20 05:21:56 +00:00
if self . library . mass_original_title_update :
2023-12-01 21:15:10 +00:00
current_original = item . originalTitle
2024-01-26 05:10:35 +00:00
for option in self . library . mass_original_title_update :
if option in [ " lock " , " remove " ] :
if option == " remove " and current_original :
if " originalTitle " not in remove_edits :
remove_edits [ " originalTitle " ] = [ ]
remove_edits [ " originalTitle " ] . append ( item . ratingKey )
item_edits + = " \n Remove Original Title (Batched) "
elif " originalTitle " not in locked_fields :
if " originalTitle " not in lock_edits :
lock_edits [ " originalTitle " ] = [ ]
lock_edits [ " originalTitle " ] . append ( item . ratingKey )
item_edits + = " \n Lock Original Title (Batched) "
break
elif option in [ " unlock " , " reset " ] :
if option == " reset " and current_original :
if " originalTitle " not in reset_edits :
reset_edits [ " originalTitle " ] = [ ]
reset_edits [ " originalTitle " ] . append ( item . ratingKey )
item_edits + = " \n Reset Original Title (Batched) "
elif " originalTitle " in locked_fields :
if " originalTitle " not in unlock_edits :
unlock_edits [ " originalTitle " ] = [ ]
unlock_edits [ " originalTitle " ] . append ( item . ratingKey )
item_edits + = " \n Unlock Original Title (Batched) "
break
else :
try :
if option == " anidb " :
new_original_title = anidb_obj ( ) . main_title # noqa
elif option == " anidb_official " :
new_original_title = anidb_obj ( ) . official_title # noqa
elif option == " mal " :
new_original_title = mal_obj ( ) . title # noqa
elif option == " mal_english " :
new_original_title = mal_obj ( ) . title_english # noqa
elif option == " mal_japanese " :
new_original_title = mal_obj ( ) . title_japanese # noqa
else :
new_original_title = option
if not new_original_title :
logger . info ( f " No { option } Original Title Found " )
raise Failed
if str ( current_original ) != str ( new_original_title ) :
item . editOriginalTitle ( new_original_title )
item_edits + = f " \n Updated Original Title | { new_original_title } "
break
except Failed :
continue
2022-09-20 05:21:56 +00:00
2023-01-25 07:43:25 +00:00
if self . library . mass_studio_update :
2023-12-01 21:15:10 +00:00
current_studio = item . studio
2024-01-26 05:10:35 +00:00
for option in self . library . mass_studio_update :
if option in [ " lock " , " remove " ] :
if option == " remove " and current_studio :
if " studio " not in remove_edits :
remove_edits [ " studio " ] = [ ]
remove_edits [ " studio " ] . append ( item . ratingKey )
item_edits + = " \n Remove Studio (Batched) "
elif " studio " not in locked_fields :
if " studio " not in lock_edits :
lock_edits [ " studio " ] = [ ]
lock_edits [ " studio " ] . append ( item . ratingKey )
item_edits + = " \n Lock Studio (Batched) "
break
elif option in [ " unlock " , " reset " ] :
if option == " reset " and current_studio :
if " studio " not in reset_edits :
reset_edits [ " studio " ] = [ ]
reset_edits [ " studio " ] . append ( item . ratingKey )
item_edits + = " \n Reset Studio (Batched) "
elif " studio " in locked_fields :
if " studio " not in unlock_edits :
unlock_edits [ " studio " ] = [ ]
unlock_edits [ " studio " ] . append ( item . ratingKey )
item_edits + = " \n Unlock Studio (Batched) "
break
else :
try :
if option == " tmdb " :
new_studio = tmdb_obj ( ) . studio # noqa
elif option == " anidb " :
new_studio = anidb_obj ( ) . studio # noqa
elif option == " mal " :
new_studio = mal_obj ( ) . studio # noqa
else :
new_studio = option
if not new_studio :
logger . info ( f " No { option } Studio Found " )
raise Failed
if str ( current_studio ) != str ( new_studio ) :
if new_studio not in studio_edits :
studio_edits [ new_studio ] = [ ]
studio_edits [ new_studio ] . append ( item . ratingKey )
item_edits + = f " \n Update Studio (Batched) | { new_studio } "
break
except Failed :
continue
2023-12-01 21:15:10 +00:00
2022-04-13 04:30:59 +00:00
if self . library . mass_originally_available_update :
2023-12-01 21:15:10 +00:00
current_available = item . originallyAvailableAt
2023-12-02 17:20:35 +00:00
if current_available :
current_available = current_available . strftime ( " % Y- % m- %d " )
2024-01-26 05:10:35 +00:00
for option in self . library . mass_originally_available_update :
if option in [ " lock " , " remove " ] :
if option == " remove " and current_available :
if " originallyAvailableAt " not in remove_edits :
remove_edits [ " originallyAvailableAt " ] = [ ]
remove_edits [ " originallyAvailableAt " ] . append ( item . ratingKey )
item_edits + = " \n Remove Originally Available Date (Batched) "
elif " originallyAvailableAt " not in locked_fields :
if " originallyAvailableAt " not in lock_edits :
lock_edits [ " originallyAvailableAt " ] = [ ]
lock_edits [ " originallyAvailableAt " ] . append ( item . ratingKey )
item_edits + = " \n Lock Originally Available Date (Batched) "
break
elif option in [ " unlock " , " reset " ] :
if option == " reset " and current_available :
if " originallyAvailableAt " not in reset_edits :
reset_edits [ " originallyAvailableAt " ] = [ ]
reset_edits [ " originallyAvailableAt " ] . append ( item . ratingKey )
item_edits + = " \n Reset Originally Available Date (Batched) "
elif " originallyAvailableAt " in locked_fields :
if " originallyAvailableAt " not in unlock_edits :
unlock_edits [ " originallyAvailableAt " ] = [ ]
unlock_edits [ " originallyAvailableAt " ] . append ( item . ratingKey )
item_edits + = " \n Unlock Originally Available Date (Batched) "
break
else :
try :
if option == " tmdb " :
new_available = tmdb_obj ( ) . release_date if self . library . is_movie else tmdb_obj ( ) . first_air_date # noqa
elif option == " omdb " :
new_available = omdb_obj ( ) . released # noqa
elif option == " tvdb " :
new_available = tvdb_obj ( ) . release_date # noqa
elif option == " mdb " :
new_available = mdb_obj ( ) . released # noqa
elif option == " mdb_digital " :
new_available = mdb_obj ( ) . released_digital # noqa
elif option == " anidb " :
new_available = anidb_obj ( ) . released # noqa
elif option == " mal " :
new_available = mal_obj ( ) . aired # noqa
else :
new_available = option
if not new_available :
logger . info ( f " No { option } Originally Available Date Found " )
raise Failed
2023-12-02 17:20:35 +00:00
new_available = new_available . strftime ( " % Y- % m- %d " )
if current_available != new_available :
if new_available not in available_edits :
available_edits [ new_available ] = [ ]
available_edits [ new_available ] . append ( item . ratingKey )
item_edits + = f " \n Update Originally Available Date (Batched) | { new_available } "
2024-01-26 05:10:35 +00:00
break
except Failed :
continue
2023-12-01 21:15:10 +00:00
if len ( item_edits ) > 0 :
logger . info ( f " Item Edits { item_edits } " )
else :
logger . info ( " No Item Edits " )
2022-06-09 14:20:43 +00:00
2022-10-26 06:57:39 +00:00
if self . library . mass_poster_update or self . library . mass_background_update :
try :
2022-11-11 22:08:08 +00:00
new_poster , new_background , item_dir , name = self . library . find_item_assets ( item )
2022-10-26 06:57:39 +00:00
except Failed :
2022-11-11 22:08:08 +00:00
item_dir = None
name = None
2022-10-26 06:57:39 +00:00
new_poster = None
new_background = None
2024-01-26 05:10:35 +00:00
try :
tmdb_item = tmdb_obj ( )
except Failed :
tmdb_item = None
2023-02-01 14:33:01 +00:00
if self . library . mass_poster_update :
2024-01-26 05:10:35 +00:00
self . library . poster_update ( item , new_poster , tmdb = tmdb_item . poster_url if tmdb_item else None , title = item . title ) # noqa
2023-02-01 14:33:01 +00:00
if self . library . mass_background_update :
2024-01-26 05:10:35 +00:00
self . library . background_update ( item , new_background , tmdb = tmdb_item . backdrop_url if tmdb_item else None , title = item . title ) # noqa
2022-10-26 06:57:39 +00:00
2023-08-08 19:52:29 +00:00
if self . library . is_show and (
2023-08-14 19:00:14 +00:00
( self . library . mass_poster_update and
( self . library . mass_poster_update [ " seasons " ] or self . library . mass_poster_update [ " episodes " ] ) ) or
( self . library . mass_background_update and
( self . library . mass_background_update [ " seasons " ] or self . library . mass_background_update [ " episodes " ] ) )
2023-08-08 19:52:29 +00:00
) :
2023-02-05 19:10:08 +00:00
real_show = None
try :
2024-01-26 05:10:35 +00:00
real_show = tmdb_item . load_show ( ) if tmdb_item else None # noqa
2023-02-05 19:10:08 +00:00
except Failed as e :
logger . error ( e )
2022-11-13 07:22:16 +00:00
tmdb_seasons = { s . season_number : s for s in real_show . seasons } if real_show else { }
2022-11-11 22:08:08 +00:00
for season in self . library . query ( item . seasons ) :
2023-08-14 19:00:14 +00:00
if ( self . library . mass_poster_update and self . library . mass_poster_update [ " seasons " ] ) or \
( self . library . mass_background_update and self . library . mass_background_update [ " seasons " ] ) :
2022-11-11 22:08:08 +00:00
try :
2023-08-08 19:52:29 +00:00
season_poster , season_background , _ , _ = self . library . find_item_assets ( season , item_asset_directory = item_dir , folder_name = name )
2022-11-11 22:08:08 +00:00
except Failed :
2023-08-08 19:52:29 +00:00
season_poster = None
season_background = None
2023-08-17 13:48:05 +00:00
season_title = f " S { season . seasonNumber } { season . title } "
2023-08-08 19:52:29 +00:00
tmdb_poster = tmdb_seasons [ season . seasonNumber ] . poster_url if season . seasonNumber in tmdb_seasons else None
2023-10-11 20:03:37 +00:00
if self . library . mass_poster_update and self . library . mass_poster_update [ " seasons " ] :
2023-08-17 13:48:05 +00:00
self . library . poster_update ( season , season_poster , tmdb = tmdb_poster , title = season_title if season else None )
2023-10-11 20:03:37 +00:00
if self . library . mass_background_update and self . library . mass_background_update [ " seasons " ] :
2023-08-17 13:48:05 +00:00
self . library . background_update ( season , season_background , title = season_title if season else None )
2023-08-08 19:52:29 +00:00
2023-08-14 19:00:14 +00:00
if ( self . library . mass_poster_update and self . library . mass_poster_update [ " episodes " ] ) or \
( self . library . mass_background_update and self . library . mass_background_update [ " episodes " ] ) :
2023-08-08 19:52:29 +00:00
tmdb_episodes = { }
if season . seasonNumber in tmdb_seasons :
for episode in tmdb_seasons [ season . seasonNumber ] . episodes :
episode . _partial = False
try :
tmdb_episodes [ episode . episode_number ] = episode
except NotFound :
logger . error ( f " TMDb Error: An Episode of Season { season . seasonNumber } was Not Found " )
for episode in self . library . query ( season . episodes ) :
try :
episode_poster , episode_background , _ , _ = self . library . find_item_assets ( episode , item_asset_directory = item_dir , folder_name = name )
except Failed :
episode_poster = None
episode_background = None
2023-08-17 13:48:05 +00:00
episode_title = f " S { season . seasonNumber } E { episode . episodeNumber } { episode . title } "
2023-08-08 19:52:29 +00:00
tmdb_poster = tmdb_episodes [ episode . episodeNumber ] . still_url if episode . episodeNumber in tmdb_episodes else None
2023-10-11 20:03:37 +00:00
if self . library . mass_poster_update and self . library . mass_poster_update [ " episodes " ] :
2023-08-17 13:48:05 +00:00
self . library . poster_update ( episode , episode_poster , tmdb = tmdb_poster , title = episode_title if episode else None )
2023-10-11 20:03:37 +00:00
if self . library . mass_background_update and self . library . mass_background_update [ " episodes " ] :
2023-08-17 13:48:05 +00:00
self . library . background_update ( episode , episode_background , title = episode_title if episode else None )
2022-11-11 22:08:08 +00:00
2023-03-01 22:10:46 +00:00
episode_ops = [
2023-12-01 21:15:10 +00:00
( self . library . mass_episode_audience_rating_update , " audienceRating " ) ,
( self . library . mass_episode_critic_rating_update , " rating " ) ,
( self . library . mass_episode_user_rating_update , " userRating " )
2023-03-01 22:10:46 +00:00
]
2022-06-09 14:20:43 +00:00
2023-12-01 21:15:10 +00:00
if any ( [ x is not None for x , _ in episode_ops ] ) :
2022-06-09 14:20:43 +00:00
2024-03-19 20:31:25 +00:00
if any ( [ " imdb " in x for x , _ in episode_ops if x ] ) and not imdb_id :
2022-07-06 18:38:26 +00:00
logger . info ( f " No IMDb ID for Guid: { item . guid } " )
2022-06-09 14:20:43 +00:00
for ep in item . episodes ( ) :
2023-03-01 22:10:46 +00:00
ep = self . library . reload ( ep )
2022-06-09 14:20:43 +00:00
item_title = self . library . get_item_sort_title ( ep , atr = " title " )
2022-07-06 18:38:26 +00:00
logger . info ( " " )
logger . info ( f " Processing { item_title } " )
2024-01-12 18:42:08 +00:00
item_edits = " "
2022-09-21 19:17:11 +00:00
2023-12-01 21:15:10 +00:00
for attribute , item_attr in episode_ops :
if attribute :
current = getattr ( ep , item_attr )
2024-03-06 19:31:50 +00:00
for option in attribute :
if option in [ " lock " , " remove " ] :
if option == " remove " and current :
if item_attr not in ep_remove_edits :
ep_remove_edits [ item_attr ] = [ ]
2024-03-06 22:04:30 +00:00
ep_remove_edits [ item_attr ] . append ( ep )
2024-03-06 19:31:50 +00:00
item_edits + = f " \n Remove { name_display [ item_attr ] } (Batched) "
elif item_attr not in locked_fields :
if item_attr not in ep_lock_edits :
ep_lock_edits [ item_attr ] = [ ]
2024-03-06 22:04:30 +00:00
ep_lock_edits [ item_attr ] . append ( ep )
2024-03-06 19:31:50 +00:00
item_edits + = f " \n Lock { name_display [ item_attr ] } (Batched) "
break
elif option in [ " unlock " , " reset " ] :
if option == " reset " and current :
if item_attr not in ep_reset_edits :
ep_reset_edits [ item_attr ] = [ ]
2024-03-06 22:04:30 +00:00
ep_reset_edits [ item_attr ] . append ( ep )
2024-03-06 19:31:50 +00:00
item_edits + = f " \n Reset { name_display [ item_attr ] } (Batched) "
elif item_attr in locked_fields :
if item_attr not in ep_unlock_edits :
ep_unlock_edits [ item_attr ] = [ ]
2024-03-06 22:04:30 +00:00
ep_unlock_edits [ item_attr ] . append ( ep )
2024-03-06 19:31:50 +00:00
item_edits + = f " \n Unlock { name_display [ item_attr ] } (Batched) "
break
2023-12-02 07:06:23 +00:00
else :
2024-03-06 19:31:50 +00:00
try :
try :
tmdb_item = tmdb_obj ( )
except Failed :
tmdb_item = None
found_rating = None
2024-03-06 22:04:30 +00:00
if tmdb_item and option == " tmdb " :
2024-03-06 19:31:50 +00:00
try :
found_rating = self . config . TMDb . get_episode ( tmdb_item . tmdb_id , ep . seasonNumber , ep . episodeNumber ) . vote_average # noqa
except Failed as er :
logger . error ( er )
2024-03-06 22:04:30 +00:00
elif imdb_id and option == " imdb " :
2024-03-06 19:31:50 +00:00
found_rating = self . config . IMDb . get_episode_rating ( imdb_id , ep . seasonNumber , ep . episodeNumber )
else :
2024-03-06 22:04:30 +00:00
try :
found_rating = float ( option )
except ValueError :
pass
2024-03-06 19:31:50 +00:00
if not found_rating :
logger . info ( f " No { option } { name_display [ item_attr ] } Found " )
raise Failed
found_rating = f " { float ( found_rating ) : .1f } "
if str ( current ) != found_rating :
if found_rating not in ep_rating_edits [ item_attr ] :
ep_rating_edits [ item_attr ] [ found_rating ] = [ ]
2024-03-06 22:04:30 +00:00
ep_rating_edits [ item_attr ] [ found_rating ] . append ( ep )
2024-03-06 19:31:50 +00:00
item_edits + = f " \n Update { name_display [ item_attr ] } (Batched) | { found_rating } "
break
except Failed :
continue
2023-12-01 21:15:10 +00:00
if len ( item_edits ) > 0 :
logger . info ( f " Item Edits: { item_edits } " )
logger . info ( " " )
logger . separator ( " Batch Updates " , space = False , border = False )
logger . info ( " " )
2023-12-11 16:17:31 +00:00
def get_batch_info ( placement , total , display_attr , total_count , display_value = None , is_episode = False , out_type = None , tag_type = None ) :
return f " Batch { name_display [ display_attr ] if display_attr in name_display else display_attr . capitalize ( ) } Update ( { placement } / { total } ): " \
2023-12-01 21:15:10 +00:00
f " { f ' { out_type . capitalize ( ) } ing ' if out_type else ' ' } " \
f " { f ' Adding { display_value } to ' if tag_type == ' add ' else f ' Removing { display_value } from ' if tag_type == ' remove ' else ' ' } " \
f " { total_count } { ' Episode ' if is_episode else ' Movie ' if self . library . is_movie else ' Show ' } " \
f " { ' s ' if total_count > 1 else ' ' } { ' ' if out_type or tag_type else f ' updated to { display_value } ' } "
for tag_attribute , edit_dict in [ ( " Label " , label_edits ) , ( " Genre " , genre_edits ) ] :
for edit_type , batch_edits in edit_dict . items ( ) :
2023-12-11 16:17:31 +00:00
_size = len ( batch_edits . items ( ) )
for i , ( tag_name , rating_keys ) in enumerate ( sorted ( batch_edits . items ( ) ) , 1 ) :
logger . info ( get_batch_info ( i , _size , tag_attribute , len ( rating_keys ) , display_value = tag_name , tag_type = edit_type ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( self . library . load_list_from_cache ( rating_keys ) )
getattr ( self . library . Plex , f " { edit_type } { tag_attribute } " ) ( tag_name )
self . library . Plex . saveMultiEdits ( )
for item_attr , _edits in rating_edits . items ( ) :
2023-12-11 16:17:31 +00:00
_size = len ( rating_edits . items ( ) )
for i , ( new_rating , rating_keys ) in enumerate ( sorted ( _edits . items ( ) ) , 1 ) :
logger . info ( get_batch_info ( i , _size , item_attr , len ( rating_keys ) , display_value = new_rating ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( self . library . load_list_from_cache ( rating_keys ) )
self . library . Plex . editField ( item_attr , new_rating )
self . library . Plex . saveMultiEdits ( )
2023-12-11 16:17:31 +00:00
_size = len ( content_edits . items ( ) )
for i , ( new_rating , rating_keys ) in enumerate ( sorted ( content_edits . items ( ) ) , 1 ) :
logger . info ( get_batch_info ( i , _size , " contentRating " , len ( rating_keys ) , display_value = new_rating ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( self . library . load_list_from_cache ( rating_keys ) )
self . library . Plex . editContentRating ( new_rating )
self . library . Plex . saveMultiEdits ( )
2023-12-11 16:17:31 +00:00
_size = len ( studio_edits . items ( ) )
for i , ( new_studio , rating_keys ) in enumerate ( sorted ( studio_edits . items ( ) ) , 1 ) :
logger . info ( get_batch_info ( i , _size , " studio " , len ( rating_keys ) , display_value = new_studio ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( self . library . load_list_from_cache ( rating_keys ) )
self . library . Plex . editStudio ( new_studio )
self . library . Plex . saveMultiEdits ( )
2023-12-11 16:17:31 +00:00
_size = len ( available_edits . items ( ) )
for i , ( new_available , rating_keys ) in enumerate ( sorted ( available_edits . items ( ) ) , 1 ) :
logger . info ( get_batch_info ( i , _size , " originallyAvailableAt " , len ( rating_keys ) , display_value = new_available ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( self . library . load_list_from_cache ( rating_keys ) )
self . library . Plex . editOriginallyAvailable ( new_available )
self . library . Plex . saveMultiEdits ( )
2023-12-11 16:17:31 +00:00
_size = len ( remove_edits . items ( ) )
for i , ( field_attr , rating_keys ) in enumerate ( remove_edits . items ( ) , 1 ) :
logger . info ( get_batch_info ( i , _size , field_attr , len ( rating_keys ) , out_type = " remov " ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( self . library . load_list_from_cache ( rating_keys ) )
self . library . Plex . editField ( field_attr , None , locked = True )
self . library . Plex . saveMultiEdits ( )
2023-12-11 16:17:31 +00:00
_size = len ( reset_edits . items ( ) )
for i , ( field_attr , rating_keys ) in enumerate ( reset_edits . items ( ) , 1 ) :
logger . info ( get_batch_info ( i , _size , field_attr , len ( rating_keys ) , out_type = " reset " ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( self . library . load_list_from_cache ( rating_keys ) )
self . library . Plex . editField ( field_attr , None , locked = False )
self . library . Plex . saveMultiEdits ( )
2023-12-11 16:17:31 +00:00
_size = len ( lock_edits . items ( ) )
for i , ( field_attr , rating_keys ) in enumerate ( lock_edits . items ( ) , 1 ) :
logger . info ( get_batch_info ( i , _size , field_attr , len ( rating_keys ) , out_type = " lock " ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( self . library . load_list_from_cache ( rating_keys ) )
self . library . Plex . _edit ( * * { f " { field_attr } .locked " : 1 } )
self . library . Plex . saveMultiEdits ( )
2023-12-11 16:17:31 +00:00
_size = len ( unlock_edits . items ( ) )
for i , ( field_attr , rating_keys ) in enumerate ( unlock_edits . items ( ) , 1 ) :
logger . info ( get_batch_info ( i , _size , field_attr , len ( rating_keys ) , out_type = " unlock " ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( self . library . load_list_from_cache ( rating_keys ) )
self . library . Plex . _edit ( * * { f " { field_attr } .locked " : 0 } )
self . library . Plex . saveMultiEdits ( )
for item_attr , _edits in ep_rating_edits . items ( ) :
2023-12-11 16:17:31 +00:00
_size = len ( _edits . items ( ) )
for i , ( new_rating , rating_keys ) in enumerate ( sorted ( _edits . items ( ) ) , 1 ) :
logger . info ( get_batch_info ( i , _size , item_attr , len ( rating_keys ) , display_value = new_rating , is_episode = True ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( rating_keys )
self . library . Plex . editField ( item_attr , new_rating )
self . library . Plex . saveMultiEdits ( )
2023-12-11 16:17:31 +00:00
_size = len ( ep_remove_edits . items ( ) )
for i , ( field_attr , rating_keys ) in enumerate ( ep_remove_edits . items ( ) , 1 ) :
logger . info ( get_batch_info ( i , _size , field_attr , len ( rating_keys ) , is_episode = True , out_type = " remov " ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( rating_keys )
self . library . Plex . editField ( field_attr , None , locked = True )
self . library . Plex . saveMultiEdits ( )
2023-12-11 16:17:31 +00:00
_size = len ( ep_reset_edits . items ( ) )
for i , ( field_attr , rating_keys ) in enumerate ( ep_reset_edits . items ( ) , 1 ) :
logger . info ( get_batch_info ( i , _size , field_attr , len ( rating_keys ) , is_episode = True , out_type = " reset " ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( rating_keys )
self . library . Plex . editField ( field_attr , None , locked = False )
self . library . Plex . saveMultiEdits ( )
2023-12-11 16:17:31 +00:00
_size = len ( ep_lock_edits . items ( ) )
for i , ( field_attr , rating_keys ) in enumerate ( ep_lock_edits . items ( ) , 1 ) :
logger . info ( get_batch_info ( i , _size , field_attr , len ( rating_keys ) , is_episode = True , out_type = " lock " ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( rating_keys )
self . library . Plex . _edit ( * * { f " { field_attr } .locked " : 1 } )
self . library . Plex . saveMultiEdits ( )
2023-12-11 16:17:31 +00:00
_size = len ( ep_unlock_edits . items ( ) )
for i , ( field_attr , rating_keys ) in enumerate ( ep_unlock_edits . items ( ) , 1 ) :
logger . info ( get_batch_info ( i , _size , field_attr , len ( rating_keys ) , is_episode = True , out_type = " unlock " ) )
2023-12-01 21:15:10 +00:00
self . library . Plex . batchMultiEdits ( rating_keys )
self . library . Plex . _edit ( * * { f " { field_attr } .locked " : 0 } )
self . library . Plex . saveMultiEdits ( )
2022-07-05 06:11:21 +00:00
2022-04-13 04:30:59 +00:00
if self . library . Radarr and self . library . radarr_add_all_existing :
2023-12-01 21:15:10 +00:00
logger . info ( " " )
logger . separator ( f " Radarr Add All Existing: { len ( radarr_adds ) } Movies " , space = False , border = False )
logger . info ( " " )
2022-04-13 04:30:59 +00:00
try :
self . library . Radarr . add_tmdb ( radarr_adds )
except Failed as e :
logger . error ( e )
if self . library . Sonarr and self . library . sonarr_add_all_existing :
2023-12-01 21:15:10 +00:00
logger . info ( " " )
logger . separator ( f " Sonarr Add All Existing: { len ( sonarr_adds ) } Shows " , space = False , border = False )
logger . info ( " " )
2022-04-13 04:30:59 +00:00
try :
self . library . Sonarr . add_tvdb ( sonarr_adds )
except Failed as e :
logger . error ( e )
if self . library . radarr_remove_by_tag :
2023-12-01 21:15:10 +00:00
logger . info ( " " )
2023-12-07 19:49:32 +00:00
logger . separator ( f " Radarr Remove { len ( self . library . radarr_remove_by_tag ) } Movies with Tags: { ' , ' . join ( self . library . radarr_remove_by_tag ) } " , space = False , border = False )
2023-12-01 21:15:10 +00:00
logger . info ( " " )
2022-04-13 04:30:59 +00:00
self . library . Radarr . remove_all_with_tags ( self . library . radarr_remove_by_tag )
if self . library . sonarr_remove_by_tag :
logger . info ( " " )
2023-12-01 21:15:10 +00:00
logger . separator ( f " Sonarr Remove { len ( self . library . sonarr_remove_by_tag ) } Shows with Tags: { ' , ' . join ( self . library . sonarr_remove_by_tag ) } " , space = False , border = False )
2022-04-13 04:30:59 +00:00
logger . info ( " " )
2023-12-01 21:15:10 +00:00
self . library . Sonarr . remove_all_with_tags ( self . library . sonarr_remove_by_tag )
2022-11-08 20:59:23 +00:00
2023-12-01 21:15:10 +00:00
if self . library . delete_collections or self . library . show_unmanaged or self . library . show_unconfigured or self . library . assets_for_all or self . library . mass_collection_mode :
2022-04-13 04:30:59 +00:00
logger . info ( " " )
2023-12-01 21:15:10 +00:00
logger . separator ( " Collection Operations " , space = False , border = False )
2022-04-13 04:30:59 +00:00
logger . info ( " " )
2023-12-01 21:15:10 +00:00
if self . library . delete_collections :
logger . info ( " " )
logger . separator ( " Deleting Collections " , space = False , border = False )
logger . info ( " " )
2022-11-08 20:59:23 +00:00
2023-12-01 21:15:10 +00:00
less = self . library . delete_collections [ " less " ] if self . library . delete_collections and self . library . delete_collections [ " less " ] is not None else None
managed = self . library . delete_collections [ " managed " ] if self . library . delete_collections else None
configured = self . library . delete_collections [ " configured " ] if self . library . delete_collections else None
unmanaged_collections = [ ]
unconfigured_collections = [ ]
all_collections = self . library . get_all_collections ( )
for i , col in enumerate ( all_collections , 1 ) :
logger . ghost ( f " Reading Collection: { i } / { len ( all_collections ) } { col . title } " )
2024-01-09 16:59:17 +00:00
col = self . library . reload ( col , force = True )
2023-12-01 21:15:10 +00:00
labels = [ la . tag for la in self . library . item_labels ( col ) ]
if ( less is not None or managed is not None or configured is not None ) \
and ( less is None or col . childCount < less ) \
and ( managed is None
or ( managed is True and " PMM " in labels )
2024-04-22 14:20:12 +00:00
or ( managed is True and " Kometa " in labels )
or ( managed is False and " PMM " not in labels )
or ( managed is False and " Kometa " not in labels ) ) \
2023-12-01 21:15:10 +00:00
and ( configured is None
or ( configured is True and col . title in self . library . collections )
or ( configured is False and col . title not in self . library . collections ) ) :
try :
self . library . delete ( col )
logger . info ( f " { col . title } Deleted " )
except Failed as e :
logger . error ( e )
else :
2024-04-22 14:20:12 +00:00
if " PMM " not in labels and " Kometa " not in labels :
2023-12-01 21:15:10 +00:00
unmanaged_collections . append ( col )
if col . title not in self . library . collection_names :
unconfigured_collections . append ( col )
if self . library . show_unmanaged and len ( unmanaged_collections ) > 0 :
logger . info ( " " )
logger . separator ( f " Unmanaged Collections in { self . library . name } Library " , space = False , border = False )
logger . info ( " " )
for col in unmanaged_collections :
logger . info ( col . title )
logger . info ( " " )
logger . info ( f " { len ( unmanaged_collections ) } Unmanaged Collection { ' s ' if len ( unmanaged_collections ) > 1 else ' ' } " )
elif self . library . show_unmanaged :
logger . info ( " " )
logger . separator ( f " No Unmanaged Collections in { self . library . name } Library " , space = False , border = False )
logger . info ( " " )
2022-04-25 20:18:04 +00:00
2023-12-01 21:15:10 +00:00
if self . library . show_unconfigured and len ( unconfigured_collections ) > 0 :
logger . info ( " " )
logger . separator ( f " Unconfigured Collections in { self . library . name } Library " , space = False , border = False )
logger . info ( " " )
for col in unconfigured_collections :
logger . info ( col . title )
logger . info ( " " )
logger . info ( f " { len ( unconfigured_collections ) } Unconfigured Collection { ' s ' if len ( unconfigured_collections ) > 1 else ' ' } " )
elif self . library . show_unconfigured :
logger . info ( " " )
logger . separator ( f " No Unconfigured Collections in { self . library . name } Library " , space = False , border = False )
logger . info ( " " )
2023-12-21 20:00:53 +00:00
if self . library . assets_for_all_collections and len ( unconfigured_collections ) > 0 :
2023-12-01 21:15:10 +00:00
logger . info ( " " )
logger . separator ( f " Unconfigured Collection Assets Check for { self . library . name } Library " , space = False , border = False )
logger . info ( " " )
for col in unconfigured_collections :
try :
poster , background , item_dir , name = self . library . find_item_assets ( col )
if poster or background :
self . library . upload_images ( col , poster = poster , background = background )
elif self . library . show_missing_assets :
logger . warning ( f " Asset Warning: No poster or background found in an assets folder for ' { name } ' " )
except Failed as e :
logger . warning ( e )
if self . library . mass_collection_mode :
logger . info ( " " )
logger . separator ( f " Unconfigured Mass Collection Mode to { self . library . mass_collection_mode } for { self . library . name } Library " , space = False , border = False )
logger . info ( " " )
for col in unconfigured_collections :
if int ( col . collectionMode ) not in plex . collection_mode_keys \
or plex . collection_mode_keys [ int ( col . collectionMode ) ] != self . library . mass_collection_mode :
self . library . collection_mode_query ( col , self . library . mass_collection_mode )
logger . info ( f " { col . title } Collection Mode Updated " )
2022-04-13 04:30:59 +00:00
if self . library . metadata_backup :
logger . info ( " " )
logger . separator ( f " Metadata Backup for { self . library . name } Library " , space = False , border = False )
logger . info ( " " )
logger . info ( f " Metadata Backup Path: { self . library . metadata_backup [ ' path ' ] } " )
logger . info ( " " )
2022-05-12 19:10:03 +00:00
yaml = None
2022-04-13 04:30:59 +00:00
if os . path . exists ( self . library . metadata_backup [ " path " ] ) :
try :
2022-05-12 19:10:03 +00:00
yaml = YAML ( path = self . library . metadata_backup [ " path " ] )
except Failed as e :
logger . error ( e )
2022-04-13 04:30:59 +00:00
filename , file_extension = os . path . splitext ( self . library . metadata_backup [ " path " ] )
i = 1
while os . path . exists ( f " { filename } { i } { file_extension } " ) :
i + = 1
os . rename ( self . library . metadata_backup [ " path " ] , f " { filename } { i } { file_extension } " )
logger . error ( f " Backup failed to load saving copy to { filename } { i } { file_extension } " )
2022-05-12 19:10:03 +00:00
if not yaml :
yaml = YAML ( path = self . library . metadata_backup [ " path " ] , create = True )
2022-11-17 06:38:58 +00:00
if " metadata " not in yaml . data or not isinstance ( yaml . data [ " metadata " ] , dict ) :
2022-05-12 19:10:03 +00:00
yaml . data [ " metadata " ] = { }
2022-04-13 04:30:59 +00:00
special_names = { }
2022-05-12 19:10:03 +00:00
for mk , mv in yaml . data [ " metadata " ] . items ( ) :
2022-05-19 19:53:22 +00:00
if mv and " title " in mv :
2022-04-13 04:30:59 +00:00
special_names [ mv [ " title " ] ] = mk
if " year " in mv :
special_names [ f " { mv [ ' title ' ] } ( { mv [ ' year ' ] } ) " ] = mk
items = self . library . get_all ( load = True )
2024-02-02 20:35:25 +00:00
titles = [ ]
year_titles = [ ]
for item in items :
titles . append ( item . title )
2024-03-06 22:04:30 +00:00
if isinstance ( item , ( Movie , Show ) ) :
year_titles . append ( f " { item . title } ( { item . year } ) " )
2022-04-13 04:30:59 +00:00
for i , item in enumerate ( items , 1 ) :
logger . ghost ( f " Processing: { i } / { len ( items ) } { item . title } " )
2024-02-02 20:35:25 +00:00
map_key , attrs = self . library . get_locked_attributes ( item , titles , year_titles )
2022-04-13 04:30:59 +00:00
if map_key in special_names :
map_key = special_names [ map_key ]
2022-05-29 05:32:09 +00:00
og_dict = yaml . data [ " metadata " ] [ map_key ] if map_key in yaml . data [ " metadata " ] and yaml . data [ " metadata " ] [ map_key ] and isinstance ( yaml . data [ " metadata " ] [ map_key ] , dict ) else { }
2022-04-13 04:30:59 +00:00
if attrs or ( self . library . metadata_backup [ " add_blank_entries " ] and not og_dict ) :
2022-05-29 05:32:09 +00:00
2022-04-13 04:30:59 +00:00
def loop_dict ( looping , dest_dict ) :
if not looping :
return None
for lk , lv in looping . items ( ) :
2022-05-29 05:32:09 +00:00
if isinstance ( lv , dict ) and lk in dest_dict and dest_dict [ lk ] and isinstance ( dest_dict [ lk ] , dict ) :
dest_dict [ lk ] = loop_dict ( lv , dest_dict [ lk ] )
else :
dest_dict [ lk ] = lv
2022-04-13 04:30:59 +00:00
return dest_dict
2022-05-29 05:32:09 +00:00
yaml . data [ " metadata " ] [ map_key ] = loop_dict ( attrs , og_dict )
2022-04-13 04:30:59 +00:00
logger . exorcise ( )
2022-05-12 19:10:03 +00:00
yaml . save ( )
2022-10-05 14:49:39 +00:00
logger . info ( f " { len ( yaml . data [ ' metadata ' ] ) } { self . library . type } { ' s ' if len ( yaml . data [ ' metadata ' ] ) > 1 else ' ' } Backed Up " )
2022-05-05 13:21:17 +00:00
operation_run_time = str ( datetime . now ( ) - operation_start ) . split ( ' . ' ) [ 0 ]
logger . info ( " " )
logger . separator ( f " Finished { self . library . name } Library Operations \n Operations Run Time: { operation_run_time } " )
return operation_run_time
