[27] fix grouping

pull/858/head
meisnate12 3 years ago
parent 2cc4d0fa43
commit 6163a1f83b

@ -1 +1 @@
1.16.5-develop26
1.16.5-develop27

@ -18,10 +18,10 @@ class Overlays:
logger.info("")
logger.separator(f"{self.library.name} Library Overlays")
logger.info("")
key_to_item = {}
os.makedirs(self.library.overlay_backup, exist_ok=True)
key_to_item = {}
key_to_overlays = {}
overlay_attrs = {}
settings = {}
if self.library.remove_overlays:
logger.info("")
logger.separator(f"Removing Overlays for the {self.library.name} Library")
@ -35,8 +35,8 @@ class Overlays:
logger.separator(f"Gathering Items for {k} Overlay", space=False, border=False)
if builder.overlay not in overlay_attrs:
overlay_attrs[builder.overlay] = {
if builder.overlay not in settings:
settings[builder.overlay] = {
"keys": [], "suppress": [], "group": None,
"priority": None, "updated": False, "image": None
}
@ -59,27 +59,27 @@ class Overlays:
for item in builder.added_items:
key_to_item[item.ratingKey] = item
added_titles.append(item.title)
if item.ratingKey not in overlay_attrs[builder.overlay]["keys"]:
overlay_attrs[builder.overlay]["keys"].append(item.ratingKey)
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}")
if builder.suppress_overlays:
overlay_attrs[builder.overlay]["suppress"] = builder.suppress_overlays
settings[builder.overlay]["suppress"] = builder.suppress_overlays
except Failed as e:
logger.error(e)
overlay_groups = {}
for overlay_name, over_attrs in overlay_attrs.items():
if overlay_attrs["group"]:
if overlay_attrs["group"] not in overlay_groups:
overlay_groups[overlay_attrs["group"]] = {}
overlay_groups[overlay_attrs["group"]][overlay_name] = overlay_attrs["priority"]
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["priority"]
for rk in over_attrs["keys"]:
for suppress_name in over_attrs["suppress"]:
if suppress_name in overlay_attrs and rk in overlay_attrs[suppress_name]["keys"]:
overlay_attrs[suppress_name]["keys"].remove(rk)
if suppress_name in settings and rk in settings[suppress_name]["keys"]:
settings[suppress_name]["keys"].remove(rk)
if not overlay_name.startswith("blur"):
clean_name, _ = util.validate_filename(overlay_name)
image_compare = None
@ -87,12 +87,12 @@ class Overlays:
_, image_compare, _ = self.config.Cache.query_image_map(overlay_name, f"{self.library.image_table_name}_overlays")
overlay_file = os.path.join(self.library.overlay_folder, f"{clean_name}.png")
overlay_size = os.stat(overlay_file).st_size
overlay_attrs[overlay_name]["updated"] = not image_compare or str(overlay_size) != str(image_compare)
overlay_attrs[overlay_name]["image"] = Image.open(overlay_file).convert("RGBA")
settings[overlay_name]["updated"] = not image_compare or str(overlay_size) != str(image_compare)
settings[overlay_name]["image"] = Image.open(overlay_file).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 overlay_attrs.items():
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], [])
@ -192,7 +192,7 @@ class Overlays:
overlay_change = True
if not overlay_change:
for over_name in over_names:
if over_name not in overlay_compare or overlay_attrs[over_name]["updated"]:
if over_name not in overlay_compare or settings[over_name]["updated"]:
overlay_change = True
clean_name, _ = util.validate_filename(item.title)
@ -258,8 +258,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(overlay_attrs[over_name]["image"].size, Image.ANTIALIAS)
new_poster.paste(overlay_attrs[over_name]["image"], (0, 0), overlay_attrs[over_name]["image"])
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.save(temp, "PNG")
self.library.upload_poster(item, temp)
self.library.edit_tags("label", item, add_tags=["Overlay"], do_print=False)

Loading…
Cancel
Save