Cleanup

couchpotato/core/media/show/_base/main.py

@@ -86,7 +86,7 @@ class ShowBase(MediaBase):
             # Can we make a base function to do this stuff?
 
             # Remove season info for later use (save separately)
-            season_info = info.get('seasons', [])
+            season_info = info.get('seasons', {})
 
             # Make sure we don't nest in_wanted data
             del info['identifiers']

couchpotato/core/media/show/providers/info/thetvdb.py

@@ -270,6 +270,22 @@ class TheTVDb(ShowProvider):
 
         show_data = dict((k, v) for k, v in show_data.iteritems() if v)
 
+        # Parse season and episode data
+        seasons = {}
+        episodes = get('episodes')
+        for episode in episodes:
+            episode_nr = episode.get('nr')
+            episode_season = episode.get('season')
+
+            # Create season
+            if seasons.get(episode_season):
+                seasons[episode_season] = {
+                    'episodes': {}
+                }
+
+            # Add episode information
+            seasons[episode_season]['episodes'][episode_nr] = self._parseEpisode(show, episode)
+
         # Add alternative titles
         # try:
         #     raw = self.tvdb.search(show['seriesname'])