github/plexpy
1
0
Fork 0
mirror of https://github.com/Tautulli/Tautulli.git synced 2025-07-16 02:02:58 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Exporter check rating key first

Browse source
This commit is contained in:
JonnyWong16 2020-08-04 10:45:46 -07:00
parent 14bb377794
commit 28c6163a31
No known key found for this signature in database
GPG key ID: B1F1F9807184697A
3 changed files with 21 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

28
plexpy/exporter.py
View file

@ -853,29 +853,19 @@ def export(section_id=None, rating_key=None, file_format='json'):
if not section_id and not rating_key:
logger.error("Tautulli Exporter :: Export called but no section_id or rating_key provided.")
return
elif section_id and not str(section_id).isdigit():
logger.error("Tautulli Exporter :: Export called with invalid section_id '%s'.", section_id)
return
elif rating_key and not str(rating_key).isdigit():
logger.error("Tautulli Exporter :: Export called with invalid rating_key '%s'.", rating_key)
return
elif section_id and not str(section_id).isdigit():
logger.error("Tautulli Exporter :: Export called with invalid section_id '%s'.", section_id)
return
elif file_format not in ('json', 'csv'):
logger.error("Tautulli Exporter :: Export called but invalid file_format '%s' provided.", file_format)
return
plex = Plex(plexpy.CONFIG.PMS_URL, plexpy.CONFIG.PMS_TOKEN)
if section_id:
logger.debug("Tautulli Exporter :: Export called with section_id %s", section_id)
library = plex.get_library(section_id)
media_type = library.type
library_title = library.title
filename = 'Library - {} [{}].{}.{}'.format(
library_title, section_id, helpers.timestamp_to_YMDHMS(timestamp), file_format)
items = library.all()
elif rating_key:
if rating_key:
logger.debug("Tautulli Exporter :: Export called with rating_key %s", rating_key)
item = plex.get_item(helpers.cast_to_int(rating_key))
@ -897,6 +887,16 @@ def export(section_id=None, rating_key=None, file_format='json'):
items = [item]
elif section_id:
logger.debug("Tautulli Exporter :: Export called with section_id %s", section_id)
library = plex.get_library(section_id)
media_type = library.type
library_title = library.title
filename = 'Library - {} [{}].{}.{}'.format(
library_title, section_id, helpers.timestamp_to_YMDHMS(timestamp), file_format)
items = library.all()
else:
return