From 181a46ae3eafc65806ca4d5dec67f06134eb4e23 Mon Sep 17 00:00:00 2001 From: FuzzyGrim Date: Fri, 21 Nov 2025 13:39:56 +0100 Subject: [PATCH] fix: include season number in enrich_items_with_user_data lookup key --- src/app/helpers.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/src/app/helpers.py b/src/app/helpers.py index e2083fc3..c3384f34 100644 --- a/src/app/helpers.py +++ b/src/app/helpers.py @@ -103,14 +103,20 @@ def enrich_items_with_user_data(request, items): # Create a lookup dictionary for fast matching media_lookup = {} for media in media_queryset: - key = (media.item.media_id, media.item.source) + if media_type == MediaTypes.SEASON.value: + key = (media.item.media_id, media.item.source, media.item.season_number) + else: + key = (media.item.media_id, media.item.source) media_lookup[key] = media # Enrich items with matched media enriched_items = [] for item in items: - key = (str(item["media_id"]), item["source"]) + if media_type == MediaTypes.SEASON.value: + key = (str(item["media_id"]), item["source"], item.get("season_number")) + else: + key = (str(item["media_id"]), item["source"]) enriched_item = { "item": item,