Added Plex library refresh calls for series and movies after subtitle deletions or modifications if Plex integration is enabled.

Author: morpheus65535

bazarr/api/subtitles/subtitles.py

@@ -2,11 +2,10 @@
 
 import os
 import sys
-import gc
 
 from flask_restx import Resource, Namespace, reqparse, fields, marshal
 
-from app.database import TableEpisodes, TableMovies, database, select
+from app.database import TableShows, TableEpisodes, TableMovies, database, select
 from languages.get_languages import alpha3_from_alpha2
 from utilities.path_mappings import path_mappings
 from utilities.video_analyzer import subtitles_sync_references
@@ -18,6 +17,8 @@
 from subtitles.sync import sync_subtitles
 from app.config import settings, empty_values, get_array_from
 from app.event_handler import event_stream
+from plex.operations import plex_refresh_item
+
 
 from ..utils import authenticate
 
@@ -122,7 +123,9 @@ def patch(self):
 
         if media_type == 'episode':
             metadata = database.execute(
-                select(TableEpisodes.path, TableEpisodes.sonarrSeriesId, TableEpisodes.subtitles)
+                select(TableEpisodes.path, TableEpisodes.sonarrSeriesId, TableEpisodes.subtitles, TableEpisodes.season,
+                       TableEpisodes.episode, TableShows.imdbId)
+                .join(TableShows)
                 .where(TableEpisodes.sonarrEpisodeId == id)) \
                 .first()
 
@@ -132,7 +135,7 @@ def patch(self):
             video_path = path_mappings.path_replace(metadata.path)
         else:
             metadata = database.execute(
-                select(TableMovies.path, TableMovies.subtitles)
+                select(TableMovies.path, TableMovies.subtitles, TableMovies.imdbId)
                 .where(TableMovies.radarrId == id))\
                 .first()
 
@@ -190,7 +193,6 @@ def patch(self):
                 except OSError:
                     return 'Unable to edit subtitles file. Check logs.', 409
         else:
-            use_original_format = True if args.get('original_format') == 'true' else False
             try:
                 subtitles_apply_mods(language=language, subtitle_path=subtitles_path, mods=[action],
                                      video_path=video_path)
@@ -207,10 +209,17 @@ def patch(self):
             store_subtitles(path_mappings.path_replace_reverse(video_path), video_path)
             event_stream(type='series', payload=metadata.sonarrSeriesId)
             event_stream(type='episode', payload=id)
+
+            if settings.general.use_plex and settings.plex.update_series_library:
+                plex_refresh_item(metadata.imdbId, is_movie=False, season=metadata.season,
+                                  episode=metadata.episode)
         else:
             store_subtitles_movie(path_mappings.path_replace_reverse_movie(video_path), video_path)
             event_stream(type='movie', payload=id)
 
+            if settings.general.use_plex and settings.plex.update_movie_library:
+                plex_refresh_item(metadata.imdbId, is_movie=True)
+
         return '', 204
 
 

bazarr/subtitles/tools/delete.py

@@ -6,6 +6,8 @@
 from subliminal.subtitle import SUBTITLE_EXTENSIONS
 
 from app.event_handler import event_stream
+from app.config import settings
+from app.database import database, TableShows, TableEpisodes, TableMovies, select
 from languages.get_languages import language_from_alpha2
 from utilities.path_mappings import path_mappings
 from utilities.autopulse_webhook import call_external_webhook
@@ -16,6 +18,7 @@
 from radarr.history import history_log_movie
 from sonarr.notify import notify_sonarr
 from radarr.notify import notify_radarr
+from plex.operations import plex_refresh_item
 
 
 def delete_subtitles(media_type, language, forced, hi, media_path, subtitles_path, sonarr_series_id=None,
@@ -40,10 +43,19 @@ def delete_subtitles(media_type, language, forced, hi, media_path, subtitles_pat
     if media_type == 'series':
         pr = path_mappings.path_replace
         prr = path_mappings.path_replace_reverse
+
+        metadata = database.execute(
+            select(TableEpisodes.season, TableEpisodes.episode, TableShows.imdbId)
+            .join(TableShows)
+            .where(TableEpisodes.sonarrEpisodeId == sonarr_episode_id)).first()
     else:
         pr = path_mappings.path_replace_movie
         prr = path_mappings.path_replace_reverse_movie
 
+        metadata = database.execute(
+            select(TableMovies.imdbId)
+            .where(TableMovies.radarrId == radarr_id)).first()
+
     result = ProcessSubtitlesResult(message=f"{language_string} subtitles deleted from disk.",
                                     reversed_path=prr(media_path),
                                     downloaded_language_code2=language_log,
@@ -67,6 +79,10 @@ def delete_subtitles(media_type, language, forced, hi, media_path, subtitles_pat
             notify_sonarr(sonarr_series_id)
             event_stream(type='series', action='update', payload=sonarr_series_id)
             event_stream(type='episode-wanted', action='update', payload=sonarr_episode_id)
+
+            if settings.general.use_plex and settings.plex.update_series_library:
+                plex_refresh_item(metadata.imdbId, is_movie=False, season=metadata.season,
+                                  episode=metadata.episode)
 
             # Call external webhook after all processing is complete
             call_external_webhook(
@@ -89,7 +105,10 @@ def delete_subtitles(media_type, language, forced, hi, media_path, subtitles_pat
             store_subtitles_movie(prr(media_path), media_path)
             notify_radarr(radarr_id)
             event_stream(type='movie-wanted', action='update', payload=radarr_id)
-            
+
+            if settings.general.use_plex and settings.plex.update_movie_library:
+                plex_refresh_item(metadata.imdbId, is_movie=True)
+
             # Call external webhook after all processing is complete
             call_external_webhook(
                 subtitle_path=subtitles_path,