Mirrors/Plex-Meta-Manager
2
0
Fork 0
mirror of https://github.com/meisnate12/Plex-Meta-Manager synced 2024-09-20 06:22:07 +00:00
Code Issues Projects Releases Packages Wiki Activity
Plex-Meta-Manager/modules/operations.py

719 lines
45 KiB
Python
Raw Normal View History

[20] move operations
2022-04-13 04:30:59 +00:00
import os, re
[65] #738 Even more expanded summary
2022-05-05 13:21:17 +00:00
from datetime import datetime
[61] #738 Expand Summary
2022-05-04 16:27:17 +00:00
from modules import plex, util
[125] small wiki fixes
2022-10-17 16:06:32 +00:00
from modules.util import Failed, LimitReached, YAML
[20] move operations
2022-04-13 04:30:59 +00:00
logger = util.logger
[2] remove mass_trakt_rating_update in favor of trakt_user option for other mass_rating operations
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",
[158] add mass_poster_update and mass_background_update library operations
2022-10-26 06:57:39 +00:00
"mass_genre_update", "mass_content_rating_update", "mass_originally_available_update", "mass_original_title_update",
"mass_poster_update", "mass_background_update"
[2] remove mass_trakt_rating_update in favor of trakt_user option for other mass_rating operations
2022-06-28 20:35:29 +00:00
]
[20] move operations
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):
[65] #738 Even more expanded summary
2022-05-05 13:21:17 +00:00
operation_start = datetime.now()
[20] move operations
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}")
[191] added delete_collections operation
2022-11-08 20:59:23 +00:00
logger.debug(f"Delete Collections: {self.library.delete_collections}")
[20] add smart url debug
2022-08-04 13:34:02 +00:00
logger.debug(f"Show Unmanaged Collections: {self.library.show_unmanaged}")
[194] add monitor to defaults
2022-11-09 02:21:46 +00:00
logger.debug(f"Show Unconfigured Collections: {self.library.show_unconfigured}")
[20] move operations
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}")
[30] add mass episode ratings
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}")
[20] move operations
2022-04-13 04:30:59 +00:00
logger.debug(f"Mass Content Rating Update: {self.library.mass_content_rating_update}")
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
logger.debug(f"Mass Original Title Update: {self.library.mass_original_title_update}")
[20] move operations
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}")
[158] add mass_poster_update and mass_background_update library operations
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}")
[20] move operations
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:
[10] builder_level
2022-07-26 19:58:53 +00:00
tracks = self.library.get_all(builder_level="track")
[20] move operations
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:
[14] #831 utilize cache more
2022-04-20 21:30:31 +00:00
items = self.library.get_all()
[20] move operations
2022-04-13 04:30:59 +00:00
radarr_adds = []
sonarr_adds = []
[2] remove mass_trakt_rating_update in favor of trakt_user option for other mass_rating operations
2022-06-28 20:35:29 +00:00
trakt_ratings = self.config.Trakt.user_ratings(self.library.is_movie) if any([o == "trakt_user" for o in self.library.meta_operations]) else []
[20] move operations
2022-04-13 04:30:59 +00:00
reverse_anidb = {}
[45] add mal as an option to operations
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
[20] move operations
2022-04-13 04:30:59 +00:00
[99] has_coordinates fix
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")
[20] move operations
2022-04-13 04:30:59 +00:00
for i, item in enumerate(items, 1):
[28] remove timers
2022-04-22 02:57:26 +00:00
try:
[29] add nightly check and cached is reloaded
2022-04-22 15:24:44 +00:00
item = self.library.reload(item)
[28] remove timers
2022-04-22 02:57:26 +00:00
except Failed as e:
logger.error(e)
continue
[13] fix tmdb episode metadata
2022-07-06 18:38:26 +00:00
logger.info("")
logger.info(f"Processing: {i}/{len(items)} {item.title}")
[25] more minor fixes
2022-06-07 20:42:17 +00:00
current_labels = [la.tag for la in self.library.item_labels(item)] if self.library.assets_for_all or self.library.mass_imdb_parental_labels else []
[31] update overlays to remove the old overlays automatically
2022-04-23 19:57:33 +00:00
[39] season asset fix
2022-06-15 14:45:30 +00:00
if self.library.assets_for_all and self.library.asset_directory:
[38] asset fixes
2022-06-14 15:46:50 +00:00
self.library.find_and_upload_assets(item, current_labels)
[31] update overlays to remove the old overlays automatically
2022-04-23 19:57:33 +00:00
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
locked_fields = [f.name for f in item.fields if f.locked]
[20] move operations
2022-04-13 04:30:59 +00:00
tmdb_id, tvdb_id, imdb_id = self.library.get_ids(item)
item.batchEdits()
[34] fix plex_collectionless
2022-06-10 13:53:16 +00:00
batch_display = ""
[20] move operations
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)
batch_display += f"\n{item.title[:25]:<25} | Title | {new_title}"
if self.library.mass_imdb_parental_labels:
try:
parental_guide = self.config.IMDb.parental_guide(imdb_id)
[31] update overlays to remove the old overlays automatically
2022-04-23 19:57:33 +00:00
parental_labels = [f"{k.capitalize()}:{v}" for k, v in parental_guide.items() if self.library.mass_imdb_parental_labels == "with_none" or v != "None"]
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]
if add_labels or remove_labels:
[80] preserve capitalization for tags
2022-05-10 13:56:29 +00:00
batch_display += f"\n{self.library.edit_tags('label', item, add_tags=add_labels, remove_tags=remove_labels, do_print=False)}"
[20] move operations
2022-04-13 04:30:59 +00:00
except Failed:
pass
[10] library level template variables
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))
[20] move operations
2022-04-13 04:30:59 +00:00
tmdb_item = None
if any([o == "tmdb" for o in self.library.meta_operations]):
tmdb_item = self.config.TMDb.get_item(item, tmdb_id, tvdb_id, imdb_id, is_movie=self.library.is_movie)
omdb_item = None
if any([o == "omdb" for o in self.library.meta_operations]):
[30] add mass episode ratings
2022-06-09 14:20:43 +00:00
if self.config.OMDb.limit is not False:
logger.error("Daily OMDb Limit Reached")
elif not imdb_id:
[13] fix tmdb episode metadata
2022-07-06 18:38:26 +00:00
logger.info(f" No IMDb ID for Guid: {item.guid}")
[30] add mass episode ratings
2022-06-09 14:20:43 +00:00
else:
try:
omdb_item = self.config.OMDb.get_omdb(imdb_id)
except Failed as e:
logger.error(str(e))
except Exception:
logger.error(f"IMDb ID: {imdb_id}")
raise
[20] move operations
2022-04-13 04:30:59 +00:00
tvdb_item = None
if any([o == "tvdb" for o in self.library.meta_operations]):
if tvdb_id:
try:
[68] Cache TVDb Objects
2022-05-05 22:05:16 +00:00
tvdb_item = self.config.TVDb.get_tvdb_obj(tvdb_id, is_movie=self.library.is_movie)
[20] move operations
2022-04-13 04:30:59 +00:00
except Failed as e:
logger.error(str(e))
else:
[13] fix tmdb episode metadata
2022-07-06 18:38:26 +00:00
logger.info(f"No TVDb ID for Guid: {item.guid}")
[20] move operations
2022-04-13 04:30:59 +00:00
anidb_item = None
[45] add mal as an option to operations
2022-09-22 20:24:58 +00:00
mal_item = None
if any([o.startswith("anidb") or o.startswith("mal") for o in self.library.meta_operations]):
[20] move operations
2022-04-13 04:30:59 +00:00
if item.ratingKey in reverse_anidb:
anidb_id = reverse_anidb[item.ratingKey]
elif tvdb_id in self.config.Convert._tvdb_to_anidb:
anidb_id = self.config.Convert._tvdb_to_anidb[tvdb_id]
elif imdb_id in self.config.Convert._imdb_to_anidb:
anidb_id = self.config.Convert._imdb_to_anidb[imdb_id]
else:
anidb_id = None
[6] fix No Anime Found errors
2022-11-10 19:24:38 +00:00
if any([o.startswith("anidb") for o in self.library.meta_operations]):
if anidb_id:
try:
anidb_item = self.config.AniDB.get_anime(anidb_id)
except Failed as e:
logger.error(str(e))
else:
logger.warning(f"No AniDB ID for Guid: {item.guid}")
[45] add mal as an option to operations
2022-09-22 20:24:58 +00:00
if any([o.startswith("mal") for o in self.library.meta_operations]):
if item.ratingKey in reverse_mal:
mal_id = reverse_mal[item.ratingKey]
elif not anidb_id or anidb_id not in self.config.Convert._anidb_to_mal:
[6] fix No Anime Found errors
2022-11-10 19:24:38 +00:00
logger.warning(f"No AniDB ID to Convert to MyAnimeList ID for Guid: {item.guid}")
[45] add mal as an option to operations
2022-09-22 20:24:58 +00:00
mal_id = None
else:
mal_id = self.config.Convert._anidb_to_mal[anidb_id]
if mal_id:
try:
mal_item = self.config.MyAnimeList.get_anime(mal_id)
except Failed as e:
logger.error(str(e))
[20] move operations
2022-04-13 04:30:59 +00:00
mdb_item = None
if any([o and o.startswith("mdb") for o in self.library.meta_operations]):
if self.config.Mdblist.limit is False:
[125] small wiki fixes
2022-10-17 16:06:32 +00:00
try:
if self.library.is_show and tvdb_id and mdb_item is None:
try:
mdb_item = self.config.Mdblist.get_series(tvdb_id)
except Failed as e:
logger.trace(str(e))
except Exception:
logger.trace(f"TVDb ID: {tvdb_id}")
raise
if tmdb_id and mdb_item is None:
try:
mdb_item = self.config.Mdblist.get_movie(tmdb_id)
except LimitReached as e:
logger.debug(e)
except Failed as e:
logger.trace(str(e))
except Exception:
logger.trace(f"TMDb ID: {tmdb_id}")
raise
if imdb_id and mdb_item is None:
try:
mdb_item = self.config.Mdblist.get_imdb(imdb_id)
except LimitReached as e:
logger.debug(e)
except Failed as e:
logger.trace(str(e))
except Exception:
logger.trace(f"IMDb ID: {imdb_id}")
raise
if mdb_item is None:
logger.warning(f"No MdbItem for Guid: {item.guid}")
except LimitReached as e:
logger.debug(e)
[20] move operations
2022-04-13 04:30:59 +00:00
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
def update_rating(attribute, item_attr, display):
current = getattr(item, item_attr)
if attribute in ["remove", "reset"] and current:
item.editField(item_attr, None, locked=attribute == "remove")
return f"\n{display} | None"
elif attribute in ["unlock", "reset"] and item_attr in locked_fields:
self.library.edit_query(item, {f"{item_attr}.locked": 0})
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
return f"\n{display} | Unlocked"
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
elif attribute in ["lock", "remove"] and item_attr not in locked_fields:
self.library.edit_query(item, {f"{item_attr}.locked": 1})
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
return f"\n{display} | Locked"
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
elif attribute not in ["lock", "unlock", "remove", "reset"]:
if tmdb_item and attribute == "tmdb":
found_rating = tmdb_item.vote_average
elif imdb_id and attribute == "imdb":
found_rating = self.config.IMDb.get_rating(imdb_id)
elif attribute == "trakt_user" and self.library.is_movie and tmdb_id in trakt_ratings:
found_rating = trakt_ratings[tmdb_id]
elif attribute == "trakt_user" and self.library.is_show and tvdb_id in trakt_ratings:
found_rating = trakt_ratings[tvdb_id]
elif omdb_item and attribute == "omdb":
found_rating = omdb_item.imdb_rating
elif mdb_item and attribute == "mdb":
found_rating = mdb_item.score / 10 if mdb_item.score else None
elif mdb_item and attribute == "mdb_imdb":
found_rating = mdb_item.imdb_rating if mdb_item.imdb_rating else None
elif mdb_item and attribute == "mdb_metacritic":
found_rating = mdb_item.metacritic_rating / 10 if mdb_item.metacritic_rating else None
elif mdb_item and attribute == "mdb_metacriticuser":
found_rating = mdb_item.metacriticuser_rating if mdb_item.metacriticuser_rating else None
elif mdb_item and attribute == "mdb_trakt":
found_rating = mdb_item.trakt_rating / 10 if mdb_item.trakt_rating else None
elif mdb_item and attribute == "mdb_tomatoes":
found_rating = mdb_item.tomatoes_rating / 10 if mdb_item.tomatoes_rating else None
elif mdb_item and attribute == "mdb_tomatoesaudience":
found_rating = mdb_item.tomatoesaudience_rating / 10 if mdb_item.tomatoesaudience_rating else None
elif mdb_item and attribute == "mdb_tmdb":
found_rating = mdb_item.tmdb_rating / 10 if mdb_item.tmdb_rating else None
elif mdb_item and attribute == "mdb_letterboxd":
found_rating = mdb_item.letterboxd_rating * 2 if mdb_item.letterboxd_rating else None
elif mdb_item and attribute == "mdb_myanimelist":
found_rating = mdb_item.myanimelist_rating if mdb_item.myanimelist_rating else None
elif anidb_item and attribute == "anidb_rating":
found_rating = anidb_item.rating
elif anidb_item and attribute == "anidb_average":
found_rating = anidb_item.average
elif anidb_item and attribute == "anidb_score":
found_rating = anidb_item.score
[45] add mal as an option to operations
2022-09-22 20:24:58 +00:00
elif mal_item and attribute == "mal":
found_rating = mal_item.score
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
else:
found_rating = None
if found_rating is None:
logger.info(f"No {display} Found")
elif str(current) != str(found_rating):
item.editField(item_attr, found_rating)
return f"\n{display} | {found_rating}"
return ""
if self.library.mass_audience_rating_update:
batch_display += update_rating(self.library.mass_audience_rating_update, "audienceRating", "Audience Rating")
if self.library.mass_critic_rating_update:
batch_display += update_rating(self.library.mass_critic_rating_update, "rating", "Critic Rating")
if self.library.mass_user_rating_update:
batch_display += update_rating(self.library.mass_user_rating_update, "userRating", "User Rating")
[20] move operations
2022-04-13 04:30:59 +00:00
if self.library.mass_genre_update or self.library.genre_mapper:
try:
new_genres = []
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
if self.library.mass_genre_update and self.library.mass_genre_update not in ["lock", "unlock", "remove", "reset"]:
[20] move operations
2022-04-13 04:30:59 +00:00
if tmdb_item and self.library.mass_genre_update == "tmdb":
new_genres = tmdb_item.genres
[64] small fixes and new rating style
2022-06-27 06:15:54 +00:00
elif imdb_id and self.library.mass_genre_update == "imdb" and imdb_id in self.config.IMDb.genres:
new_genres = self.config.IMDb.genres[imdb_id]
[20] move operations
2022-04-13 04:30:59 +00:00
elif omdb_item and self.library.mass_genre_update == "omdb":
new_genres = omdb_item.genres
elif tvdb_item and self.library.mass_genre_update == "tvdb":
new_genres = tvdb_item.genres
elif anidb_item and self.library.mass_genre_update == "anidb":
[32] add mass_original_title_update library operation
2022-09-20 05:21:56 +00:00
new_genres = [str(t).title() for t in anidb_item.tags]
[45] add mal as an option to operations
2022-09-22 20:24:58 +00:00
elif mal_item and self.library.mass_genre_update == "mal":
new_genres = mal_item.genres
[20] move operations
2022-04-13 04:30:59 +00:00
else:
raise Failed
if not new_genres:
[13] fix tmdb episode metadata
2022-07-06 18:38:26 +00:00
logger.info(f"No Genres Found")
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
if self.library.genre_mapper or self.library.mass_genre_update in ["lock", "unlock"]:
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
if not new_genres and self.library.mass_genre_update not in ["remove", "reset"]:
[20] move operations
2022-04-13 04:30:59 +00:00
new_genres = [g.tag for g in item.genres]
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
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
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
temp_display = self.library.edit_tags("genre", item, sync_tags=new_genres, do_print=False,
locked=False if self.library.mass_genre_update in ["unlock", "reset"] else True,
is_locked="genre" in locked_fields)
[24] fix anime IDs
2022-07-13 15:18:35 +00:00
if temp_display:
batch_display += f"\n{temp_display}"
[20] move operations
2022-04-13 04:30:59 +00:00
except Failed:
pass
if self.library.mass_content_rating_update or self.library.content_rating_mapper:
try:
new_rating = None
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
if self.library.mass_content_rating_update and self.library.mass_content_rating_update not in ["lock", "unlock", "remove", "reset"]:
[20] move operations
2022-04-13 04:30:59 +00:00
if omdb_item and self.library.mass_content_rating_update == "omdb":
new_rating = omdb_item.content_rating
elif mdb_item and self.library.mass_content_rating_update == "mdb":
new_rating = mdb_item.content_rating if mdb_item.content_rating else None
elif mdb_item and self.library.mass_content_rating_update == "mdb_commonsense":
new_rating = mdb_item.commonsense if mdb_item.commonsense else None
[49] add mdb_commonsense0 to mass content rating update
2022-09-23 17:06:56 +00:00
elif mdb_item and self.library.mass_content_rating_update == "mdb_commonsense0":
[50] fix mdb_commonsense0
2022-09-23 17:28:26 +00:00
new_rating = str(mdb_item.commonsense).rjust(2, "0") if mdb_item.commonsense else None
[45] add mal as an option to operations
2022-09-22 20:24:58 +00:00
elif mal_item and self.library.mass_content_rating_update == "mal":
new_rating = mal_item.rating
[20] move operations
2022-04-13 04:30:59 +00:00
else:
raise Failed
if self.library.content_rating_mapper:
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
if new_rating is None and self.library.mass_content_rating_update not in ["remove", "reset"]:
[20] move operations
2022-04-13 04:30:59 +00:00
new_rating = item.contentRating
if new_rating in self.library.content_rating_mapper:
new_rating = self.library.content_rating_mapper[new_rating]
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
if self.library.mass_content_rating_update in ["remove", "reset"] and item.contentRating:
item.editField("contentRating", None, locked=self.library.mass_content_rating_update == "remove")
batch_display += f"\nContent Rating | None"
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
elif not new_rating and self.library.mass_content_rating_update not in ["lock", "unlock", "remove", "reset"]:
[13] fix tmdb episode metadata
2022-07-06 18:38:26 +00:00
logger.info(f"No Content Rating Found")
[70] #941 Fix NoneType Error
2022-06-28 02:12:00 +00:00
elif str(item.contentRating) != str(new_rating):
[20] move operations
2022-04-13 04:30:59 +00:00
item.editContentRating(new_rating)
[13] fix tmdb episode metadata
2022-07-06 18:38:26 +00:00
batch_display += f"\nContent Rating | {new_rating}"
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
elif self.library.mass_content_rating_update in ["unlock", "reset"] and "contentRating" in locked_fields:
self.library.edit_query(item, {"contentRating.locked": 0})
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
batch_display += f"\nContent Rating | Unlocked"
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
elif self.library.mass_content_rating_update in ["lock", "remove"] and "contentRating" not in locked_fields:
self.library.edit_query(item, {"contentRating.locked": 1})
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
batch_display += f"\nContent Rating | Locked"
[20] move operations
2022-04-13 04:30:59 +00:00
except Failed:
pass
[32] add mass_original_title_update library operation
2022-09-20 05:21:56 +00:00
if self.library.mass_original_title_update:
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
if self.library.mass_original_title_update in ["remove", "reset"] and item.originalTitle:
item.editField("originalTitle", None, locked=self.library.mass_original_title_update == "remove")
batch_display += f"\nOriginal Title | None"
elif self.library.mass_original_title_update in ["unlock", "reset"] and "originalTitle" in locked_fields:
self.library.edit_query(item, {"originalTitle.locked": 0})
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
batch_display += f"\nOriginal Title | Unlocked"
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
elif self.library.mass_original_title_update in ["lock", "remove"] and "originalTitle" not in locked_fields:
self.library.edit_query(item, {"originalTitle.locked": 1})
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
batch_display += f"\nOriginal Title | Locked"
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
elif self.library.mass_original_title_update not in ["lock", "unlock", "remove", "reset"]:
try:
if anidb_item and self.library.mass_original_title_update == "anidb":
new_original_title = anidb_item.main_title
elif anidb_item and self.library.mass_original_title_update == "anidb_official":
new_original_title = anidb_item.official_title
[45] add mal as an option to operations
2022-09-22 20:24:58 +00:00
elif mal_item and self.library.mass_original_title_update == "mal":
new_original_title = mal_item.title
elif mal_item and self.library.mass_original_title_update == "mal_english":
new_original_title = mal_item.title_english
elif mal_item and self.library.mass_original_title_update == "mal_japanese":
new_original_title = mal_item.title_japanese
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
else:
raise Failed
if not new_original_title:
logger.info(f"No Original Title Found")
elif str(item.originalTitle) != str(new_original_title):
item.editOriginalTitle(new_original_title)
batch_display += f"\nOriginal Title | {new_original_title}"
except Failed:
pass
[32] add mass_original_title_update library operation
2022-09-20 05:21:56 +00:00
[20] move operations
2022-04-13 04:30:59 +00:00
if self.library.mass_originally_available_update:
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
if self.library.mass_originally_available_update in ["remove", "reset"] and item.originallyAvailableAt:
item.editField("originallyAvailableAt", None, locked=self.library.mass_originally_available_update == "remove")
batch_display += f"\nOriginally Available Date | None"
elif self.library.mass_originally_available_update in ["unlock", "reset"] and "originallyAvailableAt" in locked_fields:
self.library.edit_query(item, {"originallyAvailableAt.locked": 0})
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
batch_display += f"\nOriginally Available Date | Unlocked"
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
elif self.library.mass_originally_available_update in ["lock", "remove"] and "originallyAvailableAt" not in locked_fields:
self.library.edit_query(item, {"originallyAvailableAt.locked": 1})
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
batch_display += f"\nOriginally Available Date | Locked"
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
elif self.library.mass_originally_available_update not in ["lock", "unlock", "remove", "reset"]:
try:
if omdb_item and self.library.mass_originally_available_update == "omdb":
new_date = omdb_item.released
elif mdb_item and self.library.mass_originally_available_update == "mdb":
new_date = mdb_item.released
elif tvdb_item and self.library.mass_originally_available_update == "tvdb":
new_date = tvdb_item.release_date
elif tmdb_item and self.library.mass_originally_available_update == "tmdb":
new_date = tmdb_item.release_date if self.library.is_movie else tmdb_item.first_air_date
elif anidb_item and self.library.mass_originally_available_update == "anidb":
new_date = anidb_item.released
[45] add mal as an option to operations
2022-09-22 20:24:58 +00:00
elif mal_item and self.library.mass_originally_available_update == "mal":
new_date = mal_item.aired
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
else:
raise Failed
if not new_date:
logger.info(f"No Originally Available Date Found")
elif str(item.originallyAvailableAt) != str(new_date):
item.editOriginallyAvailable(new_date)
batch_display += f"\nOriginally Available Date | {new_date.strftime('%Y-%m-%d')}"
except Failed:
pass
[20] move operations
2022-04-13 04:30:59 +00:00
[43] fixes for bull
2022-06-17 15:00:07 +00:00
if len(batch_display) > 0:
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
item.saveEdits()
[34] fix plex_collectionless
2022-06-10 13:53:16 +00:00
logger.info(f"Batch Edits{batch_display}")
[30] add mass episode ratings
2022-06-09 14:20:43 +00:00
[158] add mass_poster_update and mass_background_update library operations
2022-10-26 06:57:39 +00:00
if self.library.mass_poster_update or self.library.mass_background_update:
try:
new_poster, new_background, _, _ = self.library.find_item_assets(item)
except Failed:
new_poster = None
new_background = None
if self.library.mass_poster_update:
if self.library.mass_poster_update == "lock":
self.library.query(item.lockPoster)
logger.infd(f"Poster | Locked")
elif self.library.mass_poster_update == "unlock":
self.library.query(item.unlockPoster)
logger.infd(f"Poster | Unlocked")
else:
poster_location = "the Assets Directory" if new_poster else ""
poster_url = False if new_poster else True
new_poster = new_poster.location if new_poster else None
if not new_poster:
if self.library.mass_poster_update == "tmdb" and tmdb_item:
new_poster = tmdb_item.poster_url
poster_location = "TMDb"
if not new_poster:
poster = next((p for p in item.posters() if p.provider == "local"), None)
if poster:
new_poster = f"{self.library.url}{poster.key}&X-Plex-Token={self.library.token}"
poster_location = "Plex"
if new_poster:
self.library.upload_poster(item, new_poster, url=poster_url)
logger.infd(f"Poster | Reset from {poster_location}")
else:
logger.infd(f"Poster | No Reset Image Found")
if self.library.mass_background_update:
if self.library.mass_background_update == "lock":
self.library.query(item.lockArt)
logger.infd(f"Background | Locked")
elif self.library.mass_background_update == "unlock":
self.library.query(item.unlockArt)
logger.infd(f"Background | Unlocked")
else:
background_location = "the Assets Directory" if new_background else ""
background_url = False if new_background else True
new_background = new_background.location if new_background else None
if not new_background:
if self.library.mass_background_update == "tmdb" and tmdb_item:
new_background = tmdb_item.backdrop_url
background_location = "TMDb"
if not new_background:
background = next((p for p in item.arts() if p.provider == "local"), None)
if background:
new_background = f"{self.library.url}{background.key}&X-Plex-Token={self.library.token}"
background_location = "Plex"
if new_background:
self.library.upload_background(item, new_background, url=background_url)
logger.infd(f"Background | Reset from {background_location}")
else:
logger.infd(f"Background | No Reset Image Found")
[30] add mass episode ratings
2022-06-09 14:20:43 +00:00
episode_ops = [self.library.mass_episode_audience_rating_update, self.library.mass_episode_critic_rating_update, self.library.mass_episode_user_rating_update]
if any([x is not None for x in episode_ops]):
if any([x == "imdb" for x in episode_ops]) and not imdb_id:
[13] fix tmdb episode metadata
2022-07-06 18:38:26 +00:00
logger.info(f"No IMDb ID for Guid: {item.guid}")
[30] add mass episode ratings
2022-06-09 14:20:43 +00:00
for ep in item.episodes():
ep.batchEdits()
[8] fix episode rating update
2022-07-05 06:11:21 +00:00
batch_display = ""
[30] add mass episode ratings
2022-06-09 14:20:43 +00:00
item_title = self.library.get_item_sort_title(ep, atr="title")
[13] fix tmdb episode metadata
2022-07-06 18:38:26 +00:00
logger.info("")
logger.info(f"Processing {item_title}")
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
episode_locked_fields = [f.name for f in ep.fields if f.locked]
def update_episode_rating(attribute, item_attr, display):
current = getattr(ep, item_attr)
if attribute in ["remove", "reset"] and current:
ep.editField(item_attr, None, locked=attribute == "remove")
return f"\n{display} | None"
elif attribute in ["unlock", "reset"] and item_attr in episode_locked_fields:
self.library.edit_query(ep, {f"{item_attr}.locked": 0})
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
return f"\n{display} | Unlocked"
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
elif attribute in ["lock", "remove"] and item_attr not in episode_locked_fields:
self.library.edit_query(ep, {f"{item_attr}.locked": 1})
[48] fix mass_* operations
2022-09-23 15:01:38 +00:00
return f"\n{display} | Locked"
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
elif attribute not in ["lock", "unlock", "remove", "reset"]:
found_rating = None
if tmdb_item and attribute == "tmdb":
try:
found_rating = self.config.TMDb.get_episode(tmdb_item.tmdb_id, ep.seasonNumber, ep.episodeNumber).vote_average
except Failed as er:
logger.error(er)
elif imdb_id and attribute == "imdb":
found_rating = self.config.IMDb.get_episode_rating(imdb_id, ep.seasonNumber, ep.episodeNumber)
if found_rating is None:
logger.info(f"No {display} Found")
elif str(current) != str(found_rating):
ep.editField(item_attr, found_rating)
return f"\n{display} | {found_rating}"
return ""
[30] add mass episode ratings
2022-06-09 14:20:43 +00:00
if self.library.mass_episode_audience_rating_update:
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
batch_display += update_episode_rating(self.library.mass_episode_audience_rating_update, "audienceRating", "Audience Rating")
[30] add mass episode ratings
2022-06-09 14:20:43 +00:00
if self.library.mass_episode_critic_rating_update:
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
batch_display += update_episode_rating(self.library.mass_episode_critic_rating_update, "rating", "Critic Rating")
[30] add mass episode ratings
2022-06-09 14:20:43 +00:00
if self.library.mass_episode_user_rating_update:
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
batch_display += update_episode_rating(self.library.mass_episode_user_rating_update, "userRating", "User Rating")
[8] fix episode rating update
2022-07-05 06:11:21 +00:00
if len(batch_display) > 0:
[40] add lock, unlock, remove, reset options to mass_* operations
2022-09-21 19:17:11 +00:00
ep.saveEdits()
[13] fix tmdb episode metadata
2022-07-06 18:38:26 +00:00
logger.info(f"Batch Edits:{batch_display}")
[8] fix episode rating update
2022-07-05 06:11:21 +00:00
[20] move operations
2022-04-13 04:30:59 +00:00
if self.library.Radarr and self.library.radarr_add_all_existing:
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:
try:
self.library.Sonarr.add_tvdb(sonarr_adds)
except Failed as e:
logger.error(e)
if self.library.radarr_remove_by_tag:
self.library.Radarr.remove_all_with_tags(self.library.radarr_remove_by_tag)
if self.library.sonarr_remove_by_tag:
self.library.Sonarr.remove_all_with_tags(self.library.sonarr_remove_by_tag)
[191] added delete_collections operation
2022-11-08 20:59:23 +00:00
if self.library.delete_collections:
[20] move operations
2022-04-13 04:30:59 +00:00
logger.info("")
[191] added delete_collections operation
2022-11-08 20:59:23 +00:00
logger.separator(f"Deleting All Collections", space=False, border=False)
[20] move operations
2022-04-13 04:30:59 +00:00
logger.info("")
[191] added delete_collections operation
2022-11-08 20:59:23 +00:00
[194] add monitor to defaults
2022-11-09 02:21:46 +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 False
unmanaged = self.library.delete_collections["unmanaged"] if self.library.delete_collections else False
configured = self.library.delete_collections["configured"] if self.library.delete_collections else False
unconfigured = self.library.delete_collections["unconfigured"] if self.library.delete_collections else False
[20] move operations
2022-04-13 04:30:59 +00:00
unmanaged_collections = []
[191] added delete_collections operation
2022-11-08 20:59:23 +00:00
unconfigured_collections = []
[21] better plex_search identification
2022-08-04 20:47:54 +00:00
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}")
[27] label pmm managed collections
2022-06-08 13:29:56 +00:00
labels = [la.tag for la in self.library.item_labels(col)]
[191] added delete_collections operation
2022-11-08 20:59:23 +00:00
if (less is not None or unmanaged or managed or unconfigured or configured) \
[194] add monitor to defaults
2022-11-09 02:21:46 +00:00
and (less is None or col.childCount < less) \
and (unmanaged is False or "PMM" not in labels) \
and (managed is False or "PMM" in labels) \
and (unconfigured is False or col.title not in self.library.collections) \
and (configured is False or col.title in self.library.collections):
[20] move operations
2022-04-13 04:30:59 +00:00
self.library.query(col.delete)
logger.info(f"{col.title} Deleted")
[191] added delete_collections operation
2022-11-08 20:59:23 +00:00
else:
if "PMM" not in labels:
unmanaged_collections.append(col)
if col.title not in self.library.collections:
unconfigured_collections.append(col)
[20] move operations
2022-04-13 04:30:59 +00:00
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("")
[191] added delete_collections operation
2022-11-08 20:59:23 +00:00
if self.library.show_unconfigured and len(unconfigured_collections) > 0:
[20] move operations
2022-04-13 04:30:59 +00:00
logger.info("")
[191] added delete_collections operation
2022-11-08 20:59:23 +00:00
logger.separator(f"Unconfigured Collections in {self.library.name} Library", space=False, border=False)
[20] move operations
2022-04-13 04:30:59 +00:00
logger.info("")
[191] added delete_collections operation
2022-11-08 20:59:23 +00:00
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("")
if self.library.assets_for_all and len(unconfigured_collections) > 0:
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:
[35] catch asset errors
2022-04-25 22:55:59 +00:00
try:
poster, background, item_dir, name = self.library.find_item_assets(col)
[34] assets fix
2022-04-25 20:18:04 +00:00
if poster or background:
self.library.upload_images(col, poster=poster, background=background)
[35] catch asset errors
2022-04-25 22:55:59 +00:00
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)
[34] assets fix
2022-04-25 20:18:04 +00:00
[14] #831 utilize cache more
2022-04-20 21:30:31 +00:00
if self.library.mass_collection_mode:
logger.info("")
[191] added delete_collections operation
2022-11-08 20:59:23 +00:00
logger.separator(f"Unconfigured Mass Collection Mode to {self.library.mass_collection_mode} for {self.library.name} Library", space=False, border=False)
[14] #831 utilize cache more
2022-04-20 21:30:31 +00:00
logger.info("")
[191] added delete_collections operation
2022-11-08 20:59:23 +00:00
for col in unconfigured_collections:
[61] #738 Expand Summary
2022-05-04 16:27:17 +00:00
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")
[20] move operations
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("")
[92] update YAML read/writes
2022-05-12 19:10:03 +00:00
yaml = None
[20] move operations
2022-04-13 04:30:59 +00:00
if os.path.exists(self.library.metadata_backup["path"]):
try:
[92] update YAML read/writes
2022-05-12 19:10:03 +00:00
yaml = YAML(path=self.library.metadata_backup["path"])
except Failed as e:
logger.error(e)
[20] move operations
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}")
[92] update YAML read/writes
2022-05-12 19:10:03 +00:00
if not yaml:
yaml = YAML(path=self.library.metadata_backup["path"], create=True)
if "metadata" not in yaml.data:
yaml.data["metadata"] = {}
[20] move operations
2022-04-13 04:30:59 +00:00
special_names = {}
[92] update YAML read/writes
2022-05-12 19:10:03 +00:00
for mk, mv in yaml.data["metadata"].items():
[108] metadata_backup fix
2022-05-19 19:53:22 +00:00
if mv and "title" in mv:
[20] move operations
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)
titles = [i.title for i in items]
for i, item in enumerate(items, 1):
logger.ghost(f"Processing: {i}/{len(items)} {item.title}")
map_key, attrs = self.library.get_locked_attributes(item, titles)
if map_key in special_names:
map_key = special_names[map_key]
[4] metadata backup fix
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 {}
[20] move operations
2022-04-13 04:30:59 +00:00
if attrs or (self.library.metadata_backup["add_blank_entries"] and not og_dict):
[4] metadata backup fix
2022-05-29 05:32:09 +00:00
[20] move operations
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():
[4] metadata backup fix
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
[20] move operations
2022-04-13 04:30:59 +00:00
return dest_dict
[4] metadata backup fix
2022-05-29 05:32:09 +00:00
yaml.data["metadata"][map_key] = loop_dict(attrs, og_dict)
[20] move operations
2022-04-13 04:30:59 +00:00
logger.exorcise()
[92] update YAML read/writes
2022-05-12 19:10:03 +00:00
yaml.save()
[89] add variables to translations
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")
[65] #738 Even more expanded summary
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\nOperations Run Time: {operation_run_time}")
return operation_run_time
Reference in a new issue Copy permalink