PickleRick/jellyplist
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Activity

feat: add batch processing for adding and removing songs in playlists

Browse Source
This commit is contained in:
Kamil
2025-02-11 09:15:32 +00:00
parent 8a883edf07
commit 9e675f8cf4
1 changed files with 44 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
jellyfin/client.py
View File

@@ -257,7 +257,8 @@ class JellyfinClient:
'IncludeItemTypes': 'Audio', # Search only for audio items 'IncludeItemTypes': 'Audio', # Search only for audio items
'Recursive': 'true', # Search within all folders 'Recursive': 'true', # Search within all folders
'Fields': 'Name,Id,Album,Artists,Path' # Retrieve the name and ID of the song 'Fields': 'Name,Id,Album,Artists,Path', # Retrieve the name and ID of the song
'Limit': 100
} }
self.logger.debug(f"Url={url}") self.logger.debug(f"Url={url}")
@@ -272,30 +273,38 @@ class JellyfinClient:
def add_songs_to_playlist(self, session_token: str, user_id: str, playlist_id: str, song_ids: list[str]): def add_songs_to_playlist(self, session_token: str, user_id: str, playlist_id: str, song_ids: list[str]):
""" """
Add songs to an existing playlist. Add songs to an existing playlist in batches to prevent URL length issues.
:param playlist_id: The ID of the playlist to update. :param playlist_id: The ID of the playlist to update.
:param song_ids: A list of song IDs to add. :param song_ids: A list of song IDs to add.
:return: A success message. :return: A success message.
""" """
# Construct the API URL with query parameters # Construct the API URL without query parameters
url = f'{self.base_url}/Playlists/{playlist_id}/Items' url = f'{self.base_url}/Playlists/{playlist_id}/Items'
batch_size = 50
total_songs = len(song_ids)
self.logger.debug(f"Total songs to add: {total_songs}")
for i in range(0, total_songs, batch_size):
batch = song_ids[i:i + batch_size]
params = { params = {
'ids': ','.join(song_ids), # Comma-separated song IDs 'ids': ','.join(batch), # Comma-separated song IDs
'userId': user_id 'userId': user_id
} }
self.logger.debug(f"Url={url} - Adding batch: {batch}")
self.logger.debug(f"Url={url}") response = requests.post(
url,
# Send the request to Jellyfin API with query parameters headers=self._get_headers(session_token=session_token),
response = requests.post(url, headers=self._get_headers(session_token=session_token), params=params, timeout = self.timeout) params=params,
timeout=self.timeout
)
self.logger.debug(f"Response = {response.status_code}") self.logger.debug(f"Response = {response.status_code}")
# Check for success if response.status_code != 204: # 204 No Content indicates success
if response.status_code == 204: # 204 No Content indicates success
return {"status": "success", "message": "Songs added to playlist successfully"}
else:
raise Exception(f"Failed to add songs to playlist: {response.status_code} - {response.content}") raise Exception(f"Failed to add songs to playlist: {response.status_code} - {response.content}")
return {"status": "success", "message": "Songs added to playlist successfully"}
def remove_songs_from_playlist(self, session_token: str, playlist_id: str, song_ids): def remove_songs_from_playlist(self, session_token: str, playlist_id: str, song_ids):
""" """
Remove songs from an existing playlist. Remove songs from an existing playlist.
@@ -303,20 +312,26 @@ class JellyfinClient:
:param song_ids: A list of song IDs to remove. :param song_ids: A list of song IDs to remove.
:return: A success message. :return: A success message.
""" """
batch_size = 50
total_songs = len(song_ids)
self.logger.debug(f"Total songs to remove: {total_songs}")
for i in range(0, total_songs, batch_size):
batch = song_ids[i:i + batch_size]
url = f'{self.base_url}/Playlists/{playlist_id}/Items' url = f'{self.base_url}/Playlists/{playlist_id}/Items'
params = { params = {
'EntryIds': ','.join(song_ids) # Join song IDs with commas 'EntryIds': ','.join(batch) # Join song IDs with commas
} }
self.logger.debug(f"Url={url}") self.logger.debug(f"Url={url} - Removing batch: {batch}")
response = requests.delete(url, headers=self._get_headers(session_token=session_token), params=params, timeout=self.timeout) response = requests.delete(url, headers=self._get_headers(session_token=session_token), params=params, timeout=self.timeout)
self.logger.debug(f"Response = {response.status_code}") self.logger.debug(f"Response = {response.status_code}")
if response.status_code == 204: # 204 No Content indicates success for updating if response.status_code != 204: # 204 No Content indicates success for updating
return {"status": "success", "message": "Songs removed from playlist successfully"}
else:
raise Exception(f"Failed to remove songs from playlist: {response.content}") raise Exception(f"Failed to remove songs from playlist: {response.content}")
return {"status": "success", "message": "Songs removed from playlist successfully"}
def remove_item(self, session_token: str, playlist_id: str): def remove_item(self, session_token: str, playlist_id: str):
""" """
Remove an existing playlist by its ID. Remove an existing playlist by its ID.