Simon/Yamtrack
1
0
Fork 0
mirror of https://github.com/FuzzyGrim/Yamtrack.git synced 2026-06-28 06:45:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: include season number in enrich_items_with_user_data lookup key

Browse Source
This commit is contained in:
FuzzyGrim
2025-11-21 13:39:56 +01:00
parent 607c86f1f9
commit 181a46ae3e
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/app/helpers.py
View File

@@ -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,