Merge pull request #322 from JonnyWong16/feature/merge_multiepisodes

Add merge_multiepisodes.py

utility/merge_multiepisodes.py

@@ -0,0 +1,89 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+'''
+Description:  Automatically merge multi-episode files in Plex into a single entry.
+Author:       /u/SwiftPanda16
+Requires:     plexapi
+Notes:
+    * All episodes **MUST** be organized correctly according to Plex's "Multiple Episodes in a Single File".
+        https://support.plex.tv/articles/naming-and-organizing-your-tv-show-files/#toc-4
+
+    * Episode titles, summaries, and tags will be appended to the first episode of the group.
+
+    * To revert the changes and split the episodes again, the show must be removed and re-added to Plex (aka Plex Dance).
+
+Usage:
+    python merge_multiepisodes.py --library "TV Shows" --show "SpongeBob SquarePants"
+'''
+
+import argparse
+import os
+from collections import defaultdict
+from plexapi.server import PlexServer
+
+
+# ## EDIT SETTINGS ##
+
+PLEX_URL = ''
+PLEX_TOKEN = ''
+
+# Environmental Variables
+PLEX_URL = os.getenv('PLEX_URL', PLEX_URL)
+PLEX_TOKEN = os.getenv('PLEX_TOKEN', PLEX_TOKEN)
+
+
+def group_episodes(plex, library, show):
+    groups = defaultdict(list)
+
+    show = plex.library.section(library).get(show)
+    for episode in show.episodes():
+        groups[episode.locations[0]].append(episode)
+
+    for index, (first, *episodes) in enumerate(groups.values()):
+        if not episodes:
+            continue
+
+        title = first.title + ' / '
+        titleSort = first.titleSort + ' / '
+        summary = first.summary + '\n\n'
+        writers = []
+        directors = []
+
+        for episode in episodes:
+            title += episode.title + ' / '
+            titleSort += episode.titleSort + ' / '
+            summary += episode.summary + '\n\n'
+            writers.extend([writer.tag for writer in episode.writers])
+            directors.extend([director.tag for director in episode.directors])
+
+        merge(first, episodes)
+        first.edit(**{
+            'title.value': title[:-3],
+            'title.locked': 1,
+            'titleSort.value': titleSort[:-3],
+            'titleSort.locked': 1,
+            'summary.value': summary[:-2],
+            'summary.locked': 1,
+            'originallyAvailableAt.locked': 1,
+            'contentRating.locked': 1,
+            'index.value': index + 1,
+            'index.locked': 1
+        })
+        first.addWriter(writers, locked=True)
+        first.addDirector(directors, locked=True)
+
+
+def merge(first, episodes):
+    key = '%s/merge?ids=%s' % (first.key, ','.join([str(r.ratingKey) for r in episodes]))
+    first._server.query(key, method=first._server._session.put)
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--library', required=True)
+    parser.add_argument('--show', required=True)
+    opts = parser.parse_args()
+
+    plex = PlexServer(PLEX_URL, PLEX_TOKEN)
+    group_episodes(plex, **vars(opts))