Merge pull request #23 from Kyrluckechuck/main

Improve Large Playlist Handling, Handle Missing Songs

Author: cmathews393

spotiplex/main.py

@@ -63,15 +63,25 @@ def __init__(self):
         if self.lidarr_sync == "true":
             self.sync_lists = self.lidarr_api.get_lidarr_playlists()
         else:
+            # This should be an array of arrays to be run by multiple 'threads':
+            # For example: [["playlist1"],["playlist2"],["playlist3","playlist4"]]
             self.sync_lists = self.config.get("manual_playlists")
-        currentuser = self.plex_service.plex.myPlexAccount().username.lower()
-        if currentuser in self.user_list:
-            self.user_list.remove(currentuser)
+        print(f"Attempting to run for {self.sync_lists}")
+        self.default_user = self.plex_service.plex.myPlexAccount().username
+
+        # If the the user list provided is empty, add the default user from the token
+        if not self.user_list or len(self.user_list) == 0:
+            self.user_list.append(self.default_user)
 
     def process_for_user(self, user):
-        if user:
-            self.plex_service.plex = self.plex_service.plex.switchUser(user)
+        print(f"processing for user {user}")
+        if user == self.default_user:
+            self.plex_service.plex = self.plex_service.plex
             print(f"Processing playlists for user: {user}")
+            print("User matches credentials provided, defaulting.")
+        else:
+            print(f"Attempting to switch to user {user}")
+            self.plex_service.plex = self.plex_service.plex.switchUser(user)
 
         with ThreadPoolExecutor(max_workers=self.worker_count) as executor:
             futures = [

spotiplex/plex.py

@@ -1,7 +1,9 @@
 import requests
 import urllib3
 from plexapi.server import PlexServer
+from plexapi.exceptions import BadRequest, NotFound
 from confighandler import read_config
+import copy
 
 
 class PlexService:
@@ -19,25 +21,43 @@ def connect_plex(self):
         return PlexServer(self.server_url, self.server_token, session=session)
 
     def check_tracks_in_plex(self, spotify_tracks):
+        print("Checking tracks in plex")
         music_lib = self.plex.library.section("Music")
         plex_tracks = []
         orig_tracks = []
 
         for track_name, artist_name in spotify_tracks:
             artist_tracks_in_plex = music_lib.search(title=artist_name)
             if artist_tracks_in_plex:
-                try:
-                    for track in artist_tracks_in_plex:
+                for track in artist_tracks_in_plex:
+                    try:
                         plex_track = track.track(title=track_name)
+
+                        # Once we find a matching track, we want to break out of this iteration to get to the next song
                         if plex_track:
-                            plex_tracks.append(plex_track)
-                        else:
-                            orig_tracks.append([track_name, "Song Not in Plex"])
-                except Exception as plex_search_exception:
-                    print(plex_search_exception)
+                            break
+                    except NotFound:
+                        # This is not really an exception, just continue
+                        continue
+                    except (Exception, BadRequest) as plex_search_exception:
+                        print(f"Exception trying to search for title={artist_name}, title={track_name}")
+                        print(plex_search_exception)
+                        # While this is a fatal exception to the specific search, continue on since a subsequent match may succeed
+                        continue
+
+                if plex_track:
+                    plex_tracks.append(plex_track)
+                else:
+                    print("Song not in plex!")
+                    print(f"Found artists for '{artist_name}' ({len(artist_tracks_in_plex)})")
+                    print(f"Attempted to match song '{track_name}', but could not!")
+                    orig_tracks.append([track_name, "Song Not in Plex"])
             else:
+                print(f"No results found for artist: {artist_name}")
                 continue
 
+        print(f"Found {len(plex_tracks)} of possible {len(spotify_tracks)} (Failed to find {len(orig_tracks)})")
+
         return plex_tracks
 
     def create_or_update_playlist(
@@ -61,9 +81,18 @@ def find_playlist_by_name(self, playlist_name):
                 return playlist
         return None
 
+# Playlist is created in intervals of 300 since Plex's API will return a 414 URL TOO LONG inconsistently past that
     def create_playlist(self, playlist_name, playlist_id, tracks):
+        tracks_to_add = copy.deepcopy(tracks)
         try:
-            new_playlist = self.plex.createPlaylist(playlist_name, items=tracks)
+            iteration_tracks = tracks_to_add[:300]
+            del tracks_to_add[:300]
+            new_playlist = self.plex.createPlaylist(playlist_name, items=iteration_tracks)
+
+            while len(tracks_to_add) > 0:
+                iteration_tracks = tracks_to_add[:300]
+                del tracks_to_add[:300]
+                new_playlist.addItems(iteration_tracks)
             return new_playlist
         except Exception as e:
             print(f"Error creating playlist {playlist_name}: {e}")