Itmens/lib.itmens
Itmens/lib.itmens
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add missing season_number when creating episode

Browse source
This commit is contained in:
Your Name 2024-02-03 22:02:22 -05:00 • committed by Henri Dickson
parent 0ee03cf61d
commit ac202a8e7e
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
catalog/sites/imdb.py
View file

@ -176,12 +176,13 @@ class IMDB(AbstractSite):
if res and res.item:
episode = res.item
episode.set_parent_item(season)
episode.season_number = season.season_number
episode.save()
else:
_logger.warning(f"season {season} has no episodes fetched, creating dummy")
cnt = int(season.episode_count or 0)
if cnt > 20:
cnt = 20
if cnt > 50:
cnt = 50
for i in range(1, cnt + 1):
episode = TVEpisode.objects.filter(
season=season, episode_number=i
@ -190,5 +191,6 @@ class IMDB(AbstractSite):
TVEpisode.objects.create(
title=f"S{season.season_number or '0'}E{i}",
season=season,
season_number=season.season_number,
episode_number=i,
)