|
|
|
@ -19,153 +19,47 @@ class Overlays:
|
|
|
|
|
logger.separator(f"{self.library.name} Library Overlays")
|
|
|
|
|
logger.info("")
|
|
|
|
|
os.makedirs(self.library.overlay_backup, exist_ok=True)
|
|
|
|
|
key_to_item = {}
|
|
|
|
|
key_to_overlays = {}
|
|
|
|
|
settings = {}
|
|
|
|
|
if self.library.remove_overlays:
|
|
|
|
|
|
|
|
|
|
old_overlays = [l for l in self.library.Plex.listFilterChoices("label") if str(l.title).lower().endswith(" overlay")]
|
|
|
|
|
if old_overlays:
|
|
|
|
|
logger.info("")
|
|
|
|
|
logger.separator(f"Removing Overlays for the {self.library.name} Library")
|
|
|
|
|
logger.separator(f"Removing Old Overlays for the {self.library.name} Library")
|
|
|
|
|
logger.info("")
|
|
|
|
|
else:
|
|
|
|
|
for overlay_file in self.library.overlay_files:
|
|
|
|
|
for k, v in overlay_file.overlays.items():
|
|
|
|
|
try:
|
|
|
|
|
builder = CollectionBuilder(self.config, overlay_file, k, v, library=self.library, overlay=True)
|
|
|
|
|
for old_overlay in old_overlays:
|
|
|
|
|
label_items = self.get_overlay_items(label=old_overlay)
|
|
|
|
|
if label_items:
|
|
|
|
|
logger.info("")
|
|
|
|
|
|
|
|
|
|
logger.separator(f"Gathering Items for {k} Overlay", space=False, border=False)
|
|
|
|
|
|
|
|
|
|
if builder.overlay not in settings:
|
|
|
|
|
settings[builder.overlay] = {
|
|
|
|
|
"keys": [], "suppress": builder.suppress_overlays, "group": builder.overlay_group,
|
|
|
|
|
"weight": builder.overlay_weight, "path": builder.overlay_path, "updated": False, "image": None
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for method, value in builder.builders:
|
|
|
|
|
logger.debug("")
|
|
|
|
|
logger.debug(f"Builder: {method}: {value}")
|
|
|
|
|
logger.separator(f"Removing {old_overlay.title}")
|
|
|
|
|
logger.info("")
|
|
|
|
|
builder.filter_and_save_items(builder.gather_ids(method, value))
|
|
|
|
|
|
|
|
|
|
if builder.filters or builder.tmdb_filters:
|
|
|
|
|
logger.info("")
|
|
|
|
|
for filter_key, filter_value in builder.filters:
|
|
|
|
|
logger.info(f"Collection Filter {filter_key}: {filter_value}")
|
|
|
|
|
for filter_key, filter_value in builder.tmdb_filters:
|
|
|
|
|
logger.info(f"Collection Filter {filter_key}: {filter_value}")
|
|
|
|
|
|
|
|
|
|
added_titles = []
|
|
|
|
|
if builder.added_items:
|
|
|
|
|
for item in builder.added_items:
|
|
|
|
|
key_to_item[item.ratingKey] = item
|
|
|
|
|
added_titles.append(item.title)
|
|
|
|
|
if item.ratingKey not in settings[builder.overlay]["keys"]:
|
|
|
|
|
settings[builder.overlay]["keys"].append(item.ratingKey)
|
|
|
|
|
if added_titles:
|
|
|
|
|
logger.debug(f"{len(added_titles)} Titles Found: {added_titles}")
|
|
|
|
|
logger.info(f"{len(added_titles) if added_titles else 'No'} Items found for {builder.overlay}")
|
|
|
|
|
except Failed as e:
|
|
|
|
|
logger.error(e)
|
|
|
|
|
|
|
|
|
|
overlay_groups = {}
|
|
|
|
|
for overlay_name, over_attrs in settings.items():
|
|
|
|
|
if over_attrs["group"]:
|
|
|
|
|
if over_attrs["group"] not in overlay_groups:
|
|
|
|
|
overlay_groups[over_attrs["group"]] = {}
|
|
|
|
|
overlay_groups[over_attrs["group"]][overlay_name] = over_attrs["weight"]
|
|
|
|
|
for rk in over_attrs["keys"]:
|
|
|
|
|
for suppress_name in over_attrs["suppress"]:
|
|
|
|
|
if suppress_name in settings and rk in settings[suppress_name]["keys"]:
|
|
|
|
|
settings[suppress_name]["keys"].remove(rk)
|
|
|
|
|
if not overlay_name.startswith("blur"):
|
|
|
|
|
image_compare = None
|
|
|
|
|
if self.config.Cache:
|
|
|
|
|
_, image_compare, _ = self.config.Cache.query_image_map(overlay_name, f"{self.library.image_table_name}_overlays")
|
|
|
|
|
overlay_size = os.stat(settings[overlay_name]["path"]).st_size
|
|
|
|
|
settings[overlay_name]["updated"] = not image_compare or str(overlay_size) != str(image_compare)
|
|
|
|
|
settings[overlay_name]["image"] = Image.open(settings[overlay_name]["path"]).convert("RGBA")
|
|
|
|
|
if self.config.Cache:
|
|
|
|
|
self.config.Cache.update_image_map(overlay_name, f"{self.library.image_table_name}_overlays", overlay_name, overlay_size)
|
|
|
|
|
|
|
|
|
|
for overlay_name, over_attrs in settings.items():
|
|
|
|
|
for over_key in over_attrs["keys"]:
|
|
|
|
|
if over_key not in key_to_overlays:
|
|
|
|
|
key_to_overlays[over_key] = (key_to_item[over_key], [])
|
|
|
|
|
key_to_overlays[over_key][1].append(overlay_name)
|
|
|
|
|
|
|
|
|
|
for over_key, (item, over_names) in key_to_overlays.items():
|
|
|
|
|
group_status = {}
|
|
|
|
|
for over_name in over_names:
|
|
|
|
|
for overlay_group, group_names in overlay_groups.items():
|
|
|
|
|
if over_name in group_names:
|
|
|
|
|
if overlay_group not in group_status:
|
|
|
|
|
group_status[overlay_group] = []
|
|
|
|
|
group_status[overlay_group].append(over_name)
|
|
|
|
|
for gk, gv in group_status.items():
|
|
|
|
|
if len(gv) > 1:
|
|
|
|
|
final = None
|
|
|
|
|
for v in gv:
|
|
|
|
|
if final is None or overlay_groups[gk][v] > overlay_groups[gk][final]:
|
|
|
|
|
final = v
|
|
|
|
|
for v in gv:
|
|
|
|
|
if final != v:
|
|
|
|
|
key_to_overlays[over_key][1].remove(v)
|
|
|
|
|
for i, item in enumerate(label_items, 1):
|
|
|
|
|
logger.ghost(f"Restoring {old_overlay.title}: {i}/{len(label_items)} {item.title}")
|
|
|
|
|
self.remove_overlay(item, old_overlay.title, [
|
|
|
|
|
os.path.join(self.library.overlay_folder, old_overlay.title[:-8], f"{item.ratingKey}.png")
|
|
|
|
|
])
|
|
|
|
|
|
|
|
|
|
def find_poster_url(plex_item):
|
|
|
|
|
if isinstance(plex_item, Movie):
|
|
|
|
|
if plex_item.ratingKey in self.library.movie_rating_key_map:
|
|
|
|
|
return self.config.TMDb.get_movie(self.library.movie_rating_key_map[plex_item.ratingKey]).poster_url
|
|
|
|
|
elif isinstance(plex_item, (Show, Season, Episode)):
|
|
|
|
|
check_key = plex_item.ratingKey if isinstance(plex_item, Show) else plex_item.show().ratingKey
|
|
|
|
|
tmdb_id = self.config.Convert.tvdb_to_tmdb(self.library.show_rating_key_map[check_key])
|
|
|
|
|
if isinstance(plex_item, Show) and plex_item.ratingKey in self.library.show_rating_key_map:
|
|
|
|
|
return self.config.TMDb.get_show(tmdb_id).poster_url
|
|
|
|
|
elif isinstance(plex_item, Season):
|
|
|
|
|
return self.config.TMDb.get_season(tmdb_id, plex_item.seasonNumber).poster_url
|
|
|
|
|
elif isinstance(plex_item, Episode):
|
|
|
|
|
return self.config.TMDb.get_episode(tmdb_id, plex_item.seasonNumber, plex_item.episodeNumber).still_url
|
|
|
|
|
|
|
|
|
|
def get_overlay_items(libtype=None):
|
|
|
|
|
return [o for o in self.library.search(label="Overlay", libtype=libtype) if o.ratingKey not in key_to_overlays]
|
|
|
|
|
|
|
|
|
|
remove_overlays = get_overlay_items()
|
|
|
|
|
if self.library.remove_overlays:
|
|
|
|
|
remove_overlays = self.get_overlay_items()
|
|
|
|
|
if self.library.is_show:
|
|
|
|
|
remove_overlays.extend(get_overlay_items(libtype="episode"))
|
|
|
|
|
remove_overlays.extend(get_overlay_items(libtype="season"))
|
|
|
|
|
remove_overlays.extend(self.get_overlay_items(libtype="episode"))
|
|
|
|
|
remove_overlays.extend(self.get_overlay_items(libtype="season"))
|
|
|
|
|
elif self.library.is_music:
|
|
|
|
|
remove_overlays.extend(get_overlay_items(libtype="album"))
|
|
|
|
|
remove_overlays.extend(self.get_overlay_items(libtype="album"))
|
|
|
|
|
|
|
|
|
|
logger.info("")
|
|
|
|
|
if remove_overlays:
|
|
|
|
|
logger.separator(f"Removing Overlays for the {self.library.name} Library")
|
|
|
|
|
for i, item in enumerate(remove_overlays, 1):
|
|
|
|
|
logger.ghost(f"Restoring: {i}/{len(remove_overlays)} {item.title}")
|
|
|
|
|
clean_name, _ = util.validate_filename(item.title)
|
|
|
|
|
poster, _, item_dir = self.library.find_assets(
|
|
|
|
|
name="poster" if self.library.asset_folders else clean_name,
|
|
|
|
|
folder_name=clean_name if self.library.asset_folders else None,
|
|
|
|
|
prefix=f"{item.title}'s "
|
|
|
|
|
)
|
|
|
|
|
is_url = False
|
|
|
|
|
original = None
|
|
|
|
|
if poster:
|
|
|
|
|
poster_location = poster.location
|
|
|
|
|
elif os.path.exists(os.path.join(self.library.overlay_backup, f"{item.ratingKey}.png")):
|
|
|
|
|
original = os.path.join(self.library.overlay_backup, f"{item.ratingKey}.png")
|
|
|
|
|
poster_location = original
|
|
|
|
|
elif os.path.exists(os.path.join(self.library.overlay_backup, f"{item.ratingKey}.jpg")):
|
|
|
|
|
original = os.path.join(self.library.overlay_backup, f"{item.ratingKey}.jpg")
|
|
|
|
|
poster_location = original
|
|
|
|
|
self.remove_overlay(item, "Overlay", [
|
|
|
|
|
os.path.join(self.library.overlay_backup, f"{item.ratingKey}.png"),
|
|
|
|
|
os.path.join(self.library.overlay_backup, f"{item.ratingKey}.jpg")
|
|
|
|
|
])
|
|
|
|
|
logger.exorcise()
|
|
|
|
|
else:
|
|
|
|
|
is_url = True
|
|
|
|
|
poster_location = find_poster_url(item)
|
|
|
|
|
if poster_location:
|
|
|
|
|
self.library.upload_poster(item, poster_location, url=is_url)
|
|
|
|
|
self.library.edit_tags("label", item, remove_tags=["Overlay"], do_print=False)
|
|
|
|
|
if original:
|
|
|
|
|
os.remove(original)
|
|
|
|
|
logger.separator(f"No Overlays to Remove for the {self.library.name} Library")
|
|
|
|
|
logger.info("")
|
|
|
|
|
else:
|
|
|
|
|
logger.error(f"No Poster found to restore for {item.title}")
|
|
|
|
|
logger.exorcise()
|
|
|
|
|
|
|
|
|
|
if key_to_overlays:
|
|
|
|
|
key_to_overlays, properties = self.compile_overlays()
|
|
|
|
|
logger.info("")
|
|
|
|
|
logger.separator(f"Applying Overlays for the {self.library.name} Library")
|
|
|
|
|
logger.info("")
|
|
|
|
@ -187,15 +81,10 @@ class Overlays:
|
|
|
|
|
overlay_change = True
|
|
|
|
|
if not overlay_change:
|
|
|
|
|
for over_name in over_names:
|
|
|
|
|
if over_name not in overlay_compare or settings[over_name]["updated"]:
|
|
|
|
|
if over_name not in overlay_compare or properties[over_name]["updated"]:
|
|
|
|
|
overlay_change = True
|
|
|
|
|
|
|
|
|
|
clean_name, _ = util.validate_filename(item.title)
|
|
|
|
|
poster, _, item_dir = self.library.find_assets(
|
|
|
|
|
name="poster" if self.library.asset_folders else clean_name,
|
|
|
|
|
folder_name=clean_name if self.library.asset_folders else None,
|
|
|
|
|
prefix=f"{item.title}'s "
|
|
|
|
|
)
|
|
|
|
|
poster, _, item_dir = self.find_asset(item)
|
|
|
|
|
|
|
|
|
|
has_original = None
|
|
|
|
|
changed_image = False
|
|
|
|
@ -209,7 +98,7 @@ class Overlays:
|
|
|
|
|
elif os.path.exists(os.path.join(self.library.overlay_backup, f"{item.ratingKey}.jpg")):
|
|
|
|
|
has_original = os.path.join(self.library.overlay_backup, f"{item.ratingKey}.jpg")
|
|
|
|
|
else:
|
|
|
|
|
new_backup = find_poster_url(item)
|
|
|
|
|
new_backup = self.find_poster_url(item)
|
|
|
|
|
if new_backup is None:
|
|
|
|
|
new_backup = item.posterUrl
|
|
|
|
|
else:
|
|
|
|
@ -244,8 +133,8 @@ class Overlays:
|
|
|
|
|
new_poster = new_poster.filter(ImageFilter.GaussianBlur(blur_num))
|
|
|
|
|
for over_name in over_names:
|
|
|
|
|
if not over_name.startswith("blur"):
|
|
|
|
|
new_poster = new_poster.resize(settings[over_name]["image"].size, Image.ANTIALIAS)
|
|
|
|
|
new_poster.paste(settings[over_name]["image"], (0, 0), settings[over_name]["image"])
|
|
|
|
|
new_poster = new_poster.resize(properties[over_name]["image"].size, Image.ANTIALIAS)
|
|
|
|
|
new_poster.paste(properties[over_name]["image"], (0, 0), properties[over_name]["image"])
|
|
|
|
|
new_poster.save(temp, "PNG")
|
|
|
|
|
self.library.upload_poster(item, temp)
|
|
|
|
|
self.library.edit_tags("label", item, add_tags=["Overlay"], do_print=False)
|
|
|
|
@ -262,3 +151,140 @@ class Overlays:
|
|
|
|
|
except Failed as e:
|
|
|
|
|
logger.error(e)
|
|
|
|
|
logger.exorcise()
|
|
|
|
|
|
|
|
|
|
def compile_overlays(self):
|
|
|
|
|
key_to_item = {}
|
|
|
|
|
properties = {}
|
|
|
|
|
overlay_groups = {}
|
|
|
|
|
key_to_overlays = {}
|
|
|
|
|
|
|
|
|
|
for overlay_file in self.library.overlay_files:
|
|
|
|
|
for k, v in overlay_file.overlays.items():
|
|
|
|
|
try:
|
|
|
|
|
builder = CollectionBuilder(self.config, overlay_file, k, v, library=self.library, overlay=True)
|
|
|
|
|
logger.info("")
|
|
|
|
|
|
|
|
|
|
logger.separator(f"Gathering Items for {k} Overlay", space=False, border=False)
|
|
|
|
|
|
|
|
|
|
if builder.overlay not in properties:
|
|
|
|
|
properties[builder.overlay] = {
|
|
|
|
|
"keys": [], "suppress": builder.suppress_overlays, "group": builder.overlay_group,
|
|
|
|
|
"weight": builder.overlay_weight, "path": builder.overlay_path, "updated": False,
|
|
|
|
|
"image": None
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for method, value in builder.builders:
|
|
|
|
|
logger.debug("")
|
|
|
|
|
logger.debug(f"Builder: {method}: {value}")
|
|
|
|
|
logger.info("")
|
|
|
|
|
builder.filter_and_save_items(builder.gather_ids(method, value))
|
|
|
|
|
|
|
|
|
|
if builder.filters or builder.tmdb_filters:
|
|
|
|
|
logger.info("")
|
|
|
|
|
for filter_key, filter_value in builder.filters:
|
|
|
|
|
logger.info(f"Collection Filter {filter_key}: {filter_value}")
|
|
|
|
|
for filter_key, filter_value in builder.tmdb_filters:
|
|
|
|
|
logger.info(f"Collection Filter {filter_key}: {filter_value}")
|
|
|
|
|
|
|
|
|
|
added_titles = []
|
|
|
|
|
if builder.added_items:
|
|
|
|
|
for item in builder.added_items:
|
|
|
|
|
key_to_item[item.ratingKey] = item
|
|
|
|
|
added_titles.append(item.title)
|
|
|
|
|
if item.ratingKey not in properties[builder.overlay]["keys"]:
|
|
|
|
|
properties[builder.overlay]["keys"].append(item.ratingKey)
|
|
|
|
|
if added_titles:
|
|
|
|
|
logger.debug(f"{len(added_titles)} Titles Found: {added_titles}")
|
|
|
|
|
logger.info(f"{len(added_titles) if added_titles else 'No'} Items found for {builder.overlay}")
|
|
|
|
|
except Failed as e:
|
|
|
|
|
logger.error(e)
|
|
|
|
|
|
|
|
|
|
for overlay_name, over_attrs in properties.items():
|
|
|
|
|
if over_attrs["group"]:
|
|
|
|
|
if over_attrs["group"] not in overlay_groups:
|
|
|
|
|
overlay_groups[over_attrs["group"]] = {}
|
|
|
|
|
overlay_groups[over_attrs["group"]][overlay_name] = over_attrs["weight"]
|
|
|
|
|
for rk in over_attrs["keys"]:
|
|
|
|
|
for suppress_name in over_attrs["suppress"]:
|
|
|
|
|
if suppress_name in properties and rk in properties[suppress_name]["keys"]:
|
|
|
|
|
properties[suppress_name]["keys"].remove(rk)
|
|
|
|
|
if not overlay_name.startswith("blur"):
|
|
|
|
|
image_compare = None
|
|
|
|
|
if self.config.Cache:
|
|
|
|
|
_, image_compare, _ = self.config.Cache.query_image_map(overlay_name, f"{self.library.image_table_name}_overlays")
|
|
|
|
|
overlay_size = os.stat(properties[overlay_name]["path"]).st_size
|
|
|
|
|
properties[overlay_name]["updated"] = not image_compare or str(overlay_size) != str(image_compare)
|
|
|
|
|
properties[overlay_name]["image"] = Image.open(properties[overlay_name]["path"]).convert("RGBA")
|
|
|
|
|
if self.config.Cache:
|
|
|
|
|
self.config.Cache.update_image_map(overlay_name, f"{self.library.image_table_name}_overlays", overlay_name, overlay_size)
|
|
|
|
|
|
|
|
|
|
for overlay_name, over_attrs in properties.items():
|
|
|
|
|
for over_key in over_attrs["keys"]:
|
|
|
|
|
if over_key not in key_to_overlays:
|
|
|
|
|
key_to_overlays[over_key] = (key_to_item[over_key], [])
|
|
|
|
|
key_to_overlays[over_key][1].append(overlay_name)
|
|
|
|
|
|
|
|
|
|
for over_key, (item, over_names) in key_to_overlays.items():
|
|
|
|
|
group_status = {}
|
|
|
|
|
for over_name in over_names:
|
|
|
|
|
for overlay_group, group_names in overlay_groups.items():
|
|
|
|
|
if over_name in group_names:
|
|
|
|
|
if overlay_group not in group_status:
|
|
|
|
|
group_status[overlay_group] = []
|
|
|
|
|
group_status[overlay_group].append(over_name)
|
|
|
|
|
for gk, gv in group_status.items():
|
|
|
|
|
if len(gv) > 1:
|
|
|
|
|
final = None
|
|
|
|
|
for v in gv:
|
|
|
|
|
if final is None or overlay_groups[gk][v] > overlay_groups[gk][final]:
|
|
|
|
|
final = v
|
|
|
|
|
for v in gv:
|
|
|
|
|
if final != v:
|
|
|
|
|
key_to_overlays[over_key][1].remove(v)
|
|
|
|
|
return key_to_overlays, properties
|
|
|
|
|
|
|
|
|
|
def find_asset(self, item):
|
|
|
|
|
clean_asset_name, _ = util.validate_filename(item.title)
|
|
|
|
|
return self.library.find_assets(
|
|
|
|
|
name="poster" if self.library.asset_folders else clean_asset_name,
|
|
|
|
|
folder_name=clean_asset_name if self.library.asset_folders else None,
|
|
|
|
|
prefix=f"{item.title}'s "
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
def find_poster_url(self, item):
|
|
|
|
|
if isinstance(item, Movie):
|
|
|
|
|
if item.ratingKey in self.library.movie_rating_key_map:
|
|
|
|
|
return self.config.TMDb.get_movie(self.library.movie_rating_key_map[item.ratingKey]).poster_url
|
|
|
|
|
elif isinstance(item, (Show, Season, Episode)):
|
|
|
|
|
check_key = item.ratingKey if isinstance(item, Show) else item.show().ratingKey
|
|
|
|
|
tmdb_id = self.config.Convert.tvdb_to_tmdb(self.library.show_rating_key_map[check_key])
|
|
|
|
|
if isinstance(item, Show) and item.ratingKey in self.library.show_rating_key_map:
|
|
|
|
|
return self.config.TMDb.get_show(tmdb_id).poster_url
|
|
|
|
|
elif isinstance(item, Season):
|
|
|
|
|
return self.config.TMDb.get_season(tmdb_id, item.seasonNumber).poster_url
|
|
|
|
|
elif isinstance(item, Episode):
|
|
|
|
|
return self.config.TMDb.get_episode(tmdb_id, item.seasonNumber, item.episodeNumber).still_url
|
|
|
|
|
|
|
|
|
|
def get_overlay_items(self, label="Overlay", libtype=None, ignore=None):
|
|
|
|
|
items = self.library.search(label=label, libtype=libtype)
|
|
|
|
|
return items if not ignore else [o for o in items if o.ratingKey not in ignore]
|
|
|
|
|
|
|
|
|
|
def remove_overlay(self, item, label, locations):
|
|
|
|
|
poster, _, item_dir = self.find_asset(item)
|
|
|
|
|
is_url = False
|
|
|
|
|
original = None
|
|
|
|
|
if poster:
|
|
|
|
|
poster_location = poster.location
|
|
|
|
|
elif any([os.path.exists(loc) for loc in locations]):
|
|
|
|
|
original = next((loc for loc in locations if os.path.exists(loc)))
|
|
|
|
|
poster_location = original
|
|
|
|
|
else:
|
|
|
|
|
is_url = True
|
|
|
|
|
poster_location = self.find_poster_url(item)
|
|
|
|
|
if poster_location:
|
|
|
|
|
self.library.upload_poster(item, poster_location, url=is_url)
|
|
|
|
|
self.library.edit_tags("label", item, remove_tags=[label], do_print=False)
|
|
|
|
|
if original:
|
|
|
|
|
os.remove(original)
|
|
|
|
|
else:
|
|
|
|
|
logger.error(f"No Poster found to restore for {item.title}")
|