ChandimaPrabath commited on
Commit
d09d652
·
1 Parent(s): 46da173
Files changed (1) hide show
  1. tvdbApiClient.py +16 -14
tvdbApiClient.py CHANGED
@@ -58,27 +58,29 @@ async def fetch_and_cache_seasons(series_id):
58
  logging.error("Series info could not be fetched.")
59
  return
60
 
 
 
 
 
 
 
 
 
61
  all_seasons = {}
62
- for season in series_info.get('seasons', []):
63
- print(season)
64
- season_id = season['id']
65
- season_number = season.get('number')
66
 
67
  if season_number == 0:
68
  season_key = "Specials"
69
  else:
70
  season_key = f"Season {season_number}"
71
 
72
- all_seasons[season_key] = []
73
- try:
74
- season_info = tvdb.get_series_episodes(series_id, lang="eng")
75
- for episode in season_info.get('episodes', []):
76
- filtered_data = filter_episode_data(episode)
77
- all_seasons[season_key].append(filtered_data)
78
- logging.info(f"Fetched episodes for {season_key}.")
79
- except Exception as e:
80
- logging.error(f"Error fetching season info for {season_key}: {e}")
81
-
82
  # Create folder for the series
83
  series_folder = Path(SAVE_DIR) / str(series_id)
84
  series_folder.mkdir(parents=True, exist_ok=True)
 
58
  logging.error("Series info could not be fetched.")
59
  return
60
 
61
+ # Fetch all episodes for the series
62
+ try:
63
+ all_episodes = tvdb.get_series_episodes(series_id, lang="eng").get('episodes', [])
64
+ except Exception as e:
65
+ logging.error(f"Error fetching episodes for series ID {series_id}: {e}")
66
+ return
67
+
68
+ # Organize episodes by season
69
  all_seasons = {}
70
+ for episode in all_episodes:
71
+ season_number = episode.get('seasonNumber')
 
 
72
 
73
  if season_number == 0:
74
  season_key = "Specials"
75
  else:
76
  season_key = f"Season {season_number}"
77
 
78
+ if season_key not in all_seasons:
79
+ all_seasons[season_key] = []
80
+
81
+ filtered_data = filter_episode_data(episode)
82
+ all_seasons[season_key].append(filtered_data)
83
+
 
 
 
 
84
  # Create folder for the series
85
  series_folder = Path(SAVE_DIR) / str(series_id)
86
  series_folder.mkdir(parents=True, exist_ok=True)