better db model

src/core/SerieScanner.py

@@ -540,7 +540,7 @@ class SerieScanner:
         Save or update a series in the database.
         
         Creates a new record if the series doesn't exist, or updates
-        the episode_dict if it has changed.
+        the episodes if they have changed.
         
         Args:
             serie: Serie instance to save
@@ -549,26 +549,53 @@ class SerieScanner:
         Returns:
             Created or updated AnimeSeries instance, or None if unchanged
         """
-        from src.server.database.service import AnimeSeriesService
+        from src.server.database.service import AnimeSeriesService, EpisodeService
 
         # Check if series already exists
         existing = await AnimeSeriesService.get_by_key(db, serie.key)
         
         if existing:
-            # Update episode_dict if changed
-            if existing.episode_dict != serie.episodeDict:
-                updated = await AnimeSeriesService.update(
-                    db,
-                    existing.id,
-                    episode_dict=serie.episodeDict,
-                    folder=serie.folder
-                )
+            # Build existing episode dict from episodes for comparison
+            existing_episodes = await EpisodeService.get_by_series(
+                db, existing.id
+            )
+            existing_dict: dict[int, list[int]] = {}
+            for ep in existing_episodes:
+                if ep.season not in existing_dict:
+                    existing_dict[ep.season] = []
+                existing_dict[ep.season].append(ep.episode_number)
+            for season in existing_dict:
+                existing_dict[season].sort()
+            
+            # Update episodes if changed
+            if existing_dict != serie.episodeDict:
+                # Add new episodes
+                new_dict = serie.episodeDict or {}
+                for season, episode_numbers in new_dict.items():
+                    existing_eps = set(existing_dict.get(season, []))
+                    for ep_num in episode_numbers:
+                        if ep_num not in existing_eps:
+                            await EpisodeService.create(
+                                db=db,
+                                series_id=existing.id,
+                                season=season,
+                                episode_number=ep_num,
+                            )
+                
+                # Update folder if changed
+                if existing.folder != serie.folder:
+                    await AnimeSeriesService.update(
+                        db,
+                        existing.id,
+                        folder=serie.folder
+                    )
+                
                 logger.info(
                     "Updated series in database: %s (key=%s)",
                     serie.name,
                     serie.key
                 )
-                return updated
+                return existing
             else:
                 logger.debug(
                     "Series unchanged in database: %s (key=%s)",
@@ -584,8 +611,19 @@ class SerieScanner:
                 name=serie.name,
                 site=serie.site,
                 folder=serie.folder,
-                episode_dict=serie.episodeDict,
             )
+            
+            # Create Episode records
+            if serie.episodeDict:
+                for season, episode_numbers in serie.episodeDict.items():
+                    for ep_num in episode_numbers:
+                        await EpisodeService.create(
+                            db=db,
+                            series_id=anime_series.id,
+                            season=season,
+                            episode_number=ep_num,
+                        )
+            
             logger.info(
                 "Created series in database: %s (key=%s)",
                 serie.name,
@@ -608,7 +646,7 @@ class SerieScanner:
         Returns:
             Updated AnimeSeries instance, or None if not found
         """
-        from src.server.database.service import AnimeSeriesService
+        from src.server.database.service import AnimeSeriesService, EpisodeService
         
         existing = await AnimeSeriesService.get_by_key(db, serie.key)
         if not existing:
@@ -619,20 +657,43 @@ class SerieScanner:
             )
             return None
         
-        updated = await AnimeSeriesService.update(
+        # Update basic fields
+        await AnimeSeriesService.update(
             db,
             existing.id,
             name=serie.name,
             site=serie.site,
             folder=serie.folder,
-            episode_dict=serie.episodeDict,
         )
+        
+        # Update episodes - add any new ones
+        if serie.episodeDict:
+            existing_episodes = await EpisodeService.get_by_series(
+                db, existing.id
+            )
+            existing_dict: dict[int, set[int]] = {}
+            for ep in existing_episodes:
+                if ep.season not in existing_dict:
+                    existing_dict[ep.season] = set()
+                existing_dict[ep.season].add(ep.episode_number)
+            
+            for season, episode_numbers in serie.episodeDict.items():
+                existing_eps = existing_dict.get(season, set())
+                for ep_num in episode_numbers:
+                    if ep_num not in existing_eps:
+                        await EpisodeService.create(
+                            db=db,
+                            series_id=existing.id,
+                            season=season,
+                            episode_number=ep_num,
+                        )
+        
         logger.info(
             "Updated series in database: %s (key=%s)",
             serie.name,
             serie.key
         )
-        return updated
+        return existing
 
     def __find_mp4_files(self) -> Iterator[tuple[str, list[str]]]:
         """Find all .mp4 files in the directory structure."""