Add Playlist tab to user page

This commit is contained in:
JonnyWong16 2020-10-02 11:57:13 -07:00
commit 501f08dd5e
No known key found for this signature in database
GPG key ID: B1F1F9807184697A
6 changed files with 175 additions and 70 deletions

View file

@ -142,7 +142,7 @@ def has_library_type(section_type):
return bool(result)
def get_collections(section_id):
def get_collections(section_id=None):
plex = Plex(plexpy.CONFIG.PMS_URL, session.get_session_user_token())
library = plex.get_library(section_id)
@ -196,7 +196,7 @@ def get_collections_list(section_id=None, **kwargs):
'error': 'Unable to get collections: missing section_id.'}
return default_return
collections = get_collections(section_id)
collections = get_collections(section_id=section_id)
# Get datatables JSON data
json_data = helpers.process_json_kwargs(json_kwargs=kwargs['json_data'])
@ -230,20 +230,21 @@ def get_collections_list(section_id=None, **kwargs):
return data
def get_playlists(section_id):
plex = Plex(plexpy.CONFIG.PMS_URL, session.get_session_user_token())
library = Libraries().get_details(section_id=section_id)
if library['section_type'] == 'artist':
playlist_type = 'audio'
elif library['section_type'] == 'photo':
playlist_type = 'photo'
def get_playlists(section_id=None, user_id=None):
if user_id and not session.get_session_user_id():
import users
user_tokens = users.Users().get_tokens(user_id=user_id)
plex_token = user_tokens['server_token']
else:
playlist_type = 'video'
plex_token = session.get_session_user_token()
playlists = plex.plex.fetchItems(
'/playlists?type=15&playlistType={}&sectionID={}'.format(playlist_type, section_id))
plex = Plex(plexpy.CONFIG.PMS_URL, plex_token)
if user_id:
playlists = plex.plex.playlists()
else:
library = plex.get_library(section_id)
playlists = library.playlist()
playlists_list = []
for playlist in playlists:
@ -254,22 +255,22 @@ def get_playlists(section_id):
'guid': playlist.guid,
'leafCount': playlist.leafCount,
'librarySectionID': section_id,
'librarySectionTitle': library['section_name'],
'playlistType': playlist.playlistType,
'ratingKey': playlist.ratingKey,
'smart': playlist.smart,
'summary': playlist.summary,
'title': playlist.title,
'type': playlist.type,
'updatedAt': helpers.datetime_to_iso(playlist.updatedAt)
'updatedAt': helpers.datetime_to_iso(playlist.updatedAt),
'userID': user_id
}
playlists_list.append(playlist_dict)
return playlists_list
def get_playlists_list(section_id=None, **kwargs):
if not section_id:
def get_playlists_list(section_id=None, user_id=None, **kwargs):
if not section_id and not user_id:
default_return = {'recordsFiltered': 0,
'recordsTotal': 0,
'draw': 0,
@ -277,7 +278,7 @@ def get_playlists_list(section_id=None, **kwargs):
'error': 'Unable to get playlists: missing section_id.'}
return default_return
playlists = get_playlists(section_id)
playlists = get_playlists(section_id=section_id, user_id=user_id)
# Get datatables JSON data
json_data = helpers.process_json_kwargs(json_kwargs=kwargs['json_data'])

View file

@ -852,7 +852,7 @@ class WebInterface(object):
```
Required parameters:
section_id (str): The id of the Plex library section, OR
section_id (str): The id of the Plex library section
Optional parameters:
None
@ -884,12 +884,13 @@ class WebInterface(object):
@cherrypy.tools.json_out()
@requireAuth()
@addtoapi("get_playlists_table")
def get_playlists_list(self, section_id=None, **kwargs):
def get_playlists_list(self, section_id=None, user_id=None, **kwargs):
""" Get the data on the Tautulli playlists tables.
```
Required parameters:
section_id (str): The id of the Plex library section, OR
section_id (str): The section id of the Plex library, OR
user_id (str): The user id of the Plex user
Optional parameters:
None
@ -912,7 +913,9 @@ class WebInterface(object):
("duration", True, True)]
kwargs['json_data'] = build_datatables_json(kwargs, dt_columns, "title")
result = libraries.get_playlists_list(section_id=section_id, **kwargs)
result = libraries.get_playlists_list(section_id=section_id,
user_id=user_id,
**kwargs)
return result
@ -4372,7 +4375,7 @@ class WebInterface(object):
@cherrypy.expose
@requireAuth()
def info(self, rating_key=None, guid=None, source=None, section_id=None, **kwargs):
def info(self, rating_key=None, guid=None, source=None, section_id=None, user_id=None, **kwargs):
if rating_key and not str(rating_key).isdigit():
raise cherrypy.HTTPRedirect(plexpy.HTTP_ROOT)
@ -4383,6 +4386,12 @@ class WebInterface(object):
"pms_web_url": plexpy.CONFIG.PMS_WEB_URL
}
if user_id:
user_data = users.Users()
user_info = user_data.get_details(user_id=user_id)
else:
user_info = {}
# Try to get metadata from the Plex server first
if rating_key:
pms_connect = pmsconnect.PmsConnect()
@ -4405,7 +4414,7 @@ class WebInterface(object):
raise cherrypy.HTTPRedirect(plexpy.HTTP_ROOT)
return serve_template(templatename="info.html", metadata=metadata, title="Info",
config=config, source=source)
config=config, source=source, user_info=user_info)
else:
if get_session_user_id():
raise cherrypy.HTTPRedirect(plexpy.HTTP_ROOT)