Cutlery/Kyoo
1
0
Fork 0
mirror of https://github.com/zoriya/Kyoo.git synced 2025-05-31 20:24:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix weird absolute filling

Browse Source
This commit is contained in:
Zoe Roux 2024-05-03 10:42:29 +02:00
parent 3725ed89db
commit b07f164ace
No known key found for this signature in database
1 changed files with 21 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
scanner/providers/implementations/themoviedatabase.py
View File

@ -579,9 +579,17 @@ class TheMovieDatabase(Provider):
return None return None
group = await self.get(f"tv/episode_group/{group_id}") group = await self.get(f"tv/episode_group/{group_id}")
absgrp = [ep for grp in group["groups"] for ep in grp["episodes"]] absgrp = [ep for grp in group["groups"] for ep in grp["episodes"]]
logger.warn( season_starts = [
f"Incomplete absolute group for show {show_id}. Filling missing values by assuming season/episode order is ascending" next(
) (
x["episode_number"]
for x in absgrp
if x["season_number"] == s.season_number
),
1,
)
for s in show.seasons
]
complete_abs = absgrp + [ complete_abs = absgrp + [
{"season_number": s.season_number, "episode_number": e} {"season_number": s.season_number, "episode_number": e}
for s in show.seasons for s in show.seasons
@ -589,10 +597,19 @@ class TheMovieDatabase(Provider):
if s.season_number > 0 if s.season_number > 0
for e in range(1, s.episodes_count) for e in range(1, s.episodes_count)
if not any( if not any(
x["season_number"] == s.season_number and x["episode_number"] == e x["season_number"] == s.season_number
and (
x["episode_number"] == e
# take into account weird absolute (for example one piece, episodes are not reset to 1 when the season starts)
or x["episode_number"] == season_starts[s.season_number - 1] + e
)
for x in absgrp for x in absgrp
) )
] ]
if len(complete_abs) != len(absgrp):
logger.warn(
f"Incomplete absolute group for show {show_id}. Filling missing values by assuming season/episode order is ascending"
)
return complete_abs return complete_abs
except Exception as e: except Exception as e:
logger.exception( logger.exception(