github/plexpy
1
0
Fork 0
mirror of https://github.com/Tautulli/Tautulli.git synced 2025-07-30 03:28:31 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add custom export level 0

Browse source
This commit is contained in:
JonnyWong16 2020-09-27 15:10:08 -07:00
parent 13ff8f3a84
commit 0a60d5f2b2
No known key found for this signature in database
GPG key ID: B1F1F9807184697A
1 changed files with 61 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

107
plexpy/exporter.py
View file

@ -53,16 +53,23 @@ class Export(object):
'collection', 'collection',
'playlist' 'playlist'
) )
METADATA_LEVELS = (1, 2, 3, 9) PLURAL_MEDIA_TYPES = {
MEDIA_INFO_LEVELS = (1, 2, 3, 9) 'movie': 'movies',
CUSTOM_FIELD_KEYS = ( 'show': 'shows',
'movies', 'season': 'seasons',
'shows', 'seasons', 'episodes', 'episode': 'episodes',
'artists', 'albums', 'tracks', 'artist': 'artists',
'photoalbums', 'photos', 'album': 'albums',
'collections', 'children', 'track': 'tracks',
'playlists', 'items' 'phtoalbum': 'photoalbums',
) 'photo': 'photos',
'collection': 'collections',
'children': 'children',
'playlist': 'playlists',
'item': 'items'
}
METADATA_LEVELS = (0, 1, 2, 3, 9)
MEDIA_INFO_LEVELS = (0, 1, 2, 3, 9)
def __init__(self, section_id=None, rating_key=None, file_format='json', def __init__(self, section_id=None, rating_key=None, file_format='json',
metadata_level=1, media_info_level=1, include_images=False, metadata_level=1, media_info_level=1, include_images=False,
@ -982,8 +989,6 @@ class Export(object):
return _media_types[media_type]() return _media_types[media_type]()
def return_levels(self, media_type): def return_levels(self, media_type):
empty_media_info_levels = {level: [] for level in self.MEDIA_INFO_LEVELS}
def movie_levels(): def movie_levels():
_media_type = 'movie' _media_type = 'movie'
_metadata_levels = { _metadata_levels = {
@ -1049,11 +1054,13 @@ class Export(object):
def show_levels(): def show_levels():
_media_type = 'show' _media_type = 'show'
_metadata_levels = { _metadata_levels = {
0: [
'seasons'
],
1: [ 1: [
'ratingKey', 'title', 'titleSort', 'originallyAvailableAt', 'year', 'addedAt', 'ratingKey', 'title', 'titleSort', 'originallyAvailableAt', 'year', 'addedAt',
'rating', 'userRating', 'contentRating', 'rating', 'userRating', 'contentRating',
'studio', 'summary', 'guid', 'duration', 'durationHuman', 'type', 'childCount', 'studio', 'summary', 'guid', 'duration', 'durationHuman', 'type', 'childCount'
'seasons'
], ],
2: [ 2: [
'roles.tag', 'roles.role', 'roles.tag', 'roles.role',
@ -1066,18 +1073,20 @@ class Export(object):
], ],
9: self._get_all_metadata_attr(_media_type) 9: self._get_all_metadata_attr(_media_type)
} }
_media_info_levels = empty_media_info_levels _media_info_levels = {}
return _metadata_levels, _media_info_levels return _metadata_levels, _media_info_levels
def season_levels(): def season_levels():
_media_type = 'season' _media_type = 'season'
_metadata_levels = { _metadata_levels = {
0: [
'episodes'
],
1: [ 1: [
'ratingKey', 'title', 'titleSort', 'addedAt', 'ratingKey', 'title', 'titleSort', 'addedAt',
'userRating', 'userRating',
'summary', 'guid', 'type', 'index', 'summary', 'guid', 'type', 'index',
'parentTitle', 'parentRatingKey', 'parentGuid', 'parentTitle', 'parentRatingKey', 'parentGuid'
'episodes'
], ],
2: [ 2: [
'fields.name', 'fields.locked' 'fields.name', 'fields.locked'
@ -1089,7 +1098,7 @@ class Export(object):
], ],
9: self._get_all_metadata_attr(_media_type) 9: self._get_all_metadata_attr(_media_type)
} }
_media_info_levels = empty_media_info_levels _media_info_levels = {}
return _metadata_levels, _media_info_levels return _metadata_levels, _media_info_levels
def episode_levels(): def episode_levels():
@ -1159,11 +1168,13 @@ class Export(object):
def artist_levels(): def artist_levels():
_media_type = 'artist' _media_type = 'artist'
_metadata_levels = { _metadata_levels = {
0: [
'albums'
],
1: [ 1: [
'ratingKey', 'title', 'titleSort', 'addedAt', 'ratingKey', 'title', 'titleSort', 'addedAt',
'rating', 'userRating', 'rating', 'userRating',
'summary', 'guid', 'type', 'summary', 'guid', 'type'
'albums'
], ],
2: [ 2: [
'collections.tag', 'genres.tag', 'countries.tag', 'moods.tag', 'styles.tag', 'collections.tag', 'genres.tag', 'countries.tag', 'moods.tag', 'styles.tag',
@ -1175,18 +1186,20 @@ class Export(object):
], ],
9: self._get_all_metadata_attr(_media_type) 9: self._get_all_metadata_attr(_media_type)
} }
_media_info_levels = empty_media_info_levels _media_info_levels = {}
return _metadata_levels, _media_info_levels return _metadata_levels, _media_info_levels
def album_levels(): def album_levels():
_media_type = 'album' _media_type = 'album'
_metadata_levels = { _metadata_levels = {
0: [
'tracks'
],
1: [ 1: [
'ratingKey', 'title', 'titleSort', 'originallyAvailableAt', 'addedAt', 'ratingKey', 'title', 'titleSort', 'originallyAvailableAt', 'addedAt',
'rating', 'userRating', 'rating', 'userRating',
'summary', 'guid', 'type', 'index', 'summary', 'guid', 'type', 'index',
'parentTitle', 'parentRatingKey', 'parentGuid', 'parentTitle', 'parentRatingKey', 'parentGuid'
'tracks'
], ],
2: [ 2: [
'collections.tag', 'genres.tag', 'labels.tag', 'moods.tag', 'styles.tag', 'collections.tag', 'genres.tag', 'labels.tag', 'moods.tag', 'styles.tag',
@ -1199,7 +1212,7 @@ class Export(object):
], ],
9: self._get_all_metadata_attr(_media_type) 9: self._get_all_metadata_attr(_media_type)
} }
_media_info_levels = empty_media_info_levels _media_info_levels = {}
return _metadata_levels, _media_info_levels return _metadata_levels, _media_info_levels
def track_levels(): def track_levels():
@ -1263,6 +1276,9 @@ class Export(object):
def photo_album_levels(): def photo_album_levels():
_media_type = 'photoalbum' _media_type = 'photoalbum'
_metadata_levels = { _metadata_levels = {
0: [
'photos'
],
1: [ 1: [
'ratingKey', 'title', 'titleSort', 'addedAt', 'ratingKey', 'title', 'titleSort', 'addedAt',
'summary', 'guid', 'type', 'index', 'summary', 'guid', 'type', 'index',
@ -1276,7 +1292,7 @@ class Export(object):
], ],
9: self._get_all_metadata_attr(_media_type) 9: self._get_all_metadata_attr(_media_type)
} }
_media_info_levels = empty_media_info_levels _media_info_levels = {}
return _metadata_levels, _media_info_levels return _metadata_levels, _media_info_levels
def photo_levels(): def photo_levels():
@ -1319,12 +1335,14 @@ class Export(object):
def collection_levels(): def collection_levels():
_media_type = 'collection' _media_type = 'collection'
_metadata_levels = { _metadata_levels = {
0: [
'children'
],
1: [ 1: [
'ratingKey', 'title', 'titleSort', 'minYear', 'maxYear', 'addedAt', 'ratingKey', 'title', 'titleSort', 'minYear', 'maxYear', 'addedAt',
'contentRating', 'contentRating',
'summary', 'guid', 'type', 'subtype', 'childCount', 'summary', 'guid', 'type', 'subtype', 'childCount',
'collectionMode', 'collectionSort', 'collectionMode', 'collectionSort'
'children'
], ],
2: [ 2: [
'labels.tag', 'labels.tag',
@ -1336,17 +1354,19 @@ class Export(object):
], ],
9: self._get_all_metadata_attr(_media_type) 9: self._get_all_metadata_attr(_media_type)
} }
_media_info_levels = empty_media_info_levels _media_info_levels = {}
return _metadata_levels, _media_info_levels return _metadata_levels, _media_info_levels
def playlist_levels(): def playlist_levels():
_media_type = 'playlist' _media_type = 'playlist'
_metadata_levels = { _metadata_levels = {
0: [
'items'
],
1: [ 1: [
'ratingKey', 'title', 'addedAt', 'ratingKey', 'title', 'addedAt',
'summary', 'guid', 'type', 'duration', 'durationHuman', 'summary', 'guid', 'type', 'duration', 'durationHuman',
'playlistType', 'smart', 'playlistType', 'smart'
'items'
], ],
2: [ 2: [
], ],
@ -1356,7 +1376,7 @@ class Export(object):
], ],
9: self._get_all_metadata_attr(_media_type) 9: self._get_all_metadata_attr(_media_type)
} }
_media_info_levels = empty_media_info_levels _media_info_levels = {}
return _metadata_levels, _media_info_levels return _metadata_levels, _media_info_levels
_media_types = { _media_types = {
@ -1385,21 +1405,15 @@ class Export(object):
elif self.section_id and not str(self.section_id).isdigit(): elif self.section_id and not str(self.section_id).isdigit():
logger.error("Tautulli Exporter :: Export called with invalid section_id '%s'.", self.section_id) logger.error("Tautulli Exporter :: Export called with invalid section_id '%s'.", self.section_id)
return return
elif not self.metadata_level:
logger.error("Tautulli Exporter :: Export called with invalid metadata_level '%s'.", self.metadata_level)
return
elif not self.media_info_level:
logger.error("Tautulli Exporter :: Export called with invalid media_info_level '%s'.", self.media_info_level)
return
elif self.file_format not in ('json', 'csv'):
logger.error("Tautulli Exporter :: Export called but invalid file_format '%s' provided.", self.file_format)
return
elif self.metadata_level not in self.METADATA_LEVELS: elif self.metadata_level not in self.METADATA_LEVELS:
logger.error("Tautulli Exporter :: Export called with invalid metadata_level '%s'.", self.metadata_level) logger.error("Tautulli Exporter :: Export called with invalid metadata_level '%s'.", self.metadata_level)
return return
elif self.media_info_level not in self.MEDIA_INFO_LEVELS: elif self.media_info_level not in self.MEDIA_INFO_LEVELS:
logger.error("Tautulli Exporter :: Export called with invalid media_info_level '%s'.", self.media_info_level) logger.error("Tautulli Exporter :: Export called with invalid media_info_level '%s'.", self.media_info_level)
return return
elif self.file_format not in ('json', 'csv'):
logger.error("Tautulli Exporter :: Export called but invalid file_format '%s' provided.", self.file_format)
return
plex = Plex(plexpy.CONFIG.PMS_URL, plexpy.CONFIG.PMS_TOKEN) plex = Plex(plexpy.CONFIG.PMS_URL, plexpy.CONFIG.PMS_TOKEN)
@ -1480,11 +1494,10 @@ class Export(object):
continue continue
media_type = self.media_type media_type = self.media_type
for key in self.CUSTOM_FIELD_KEYS: for key in self.PLURAL_MEDIA_TYPES.values():
if field.startswith(key + '.'): if field.startswith(key + '.'):
media_type, field = field.split('.', maxsplit=1) _, field = field.split('.', maxsplit=1)
if key != 'children': media_type = key
media_type = media_type[:-1]
if media_type in self._custom_fields: if media_type in self._custom_fields:
self._custom_fields[media_type].add(field) self._custom_fields[media_type].add(field)
@ -1600,7 +1613,9 @@ class Export(object):
if image_attr in media_attrs: if image_attr in media_attrs:
export_attrs_set.add(image_attr) export_attrs_set.add(image_attr)
export_attrs_set.update(self._custom_fields.get(media_type, [])) plural_media_type = self.PLURAL_MEDIA_TYPES.get(media_type)
if plural_media_type in self._custom_fields:
export_attrs_set.update(self._custom_fields[plural_media_type])
if self.media_type == 'collection' and 'children' in self._custom_fields: if self.media_type == 'collection' and 'children' in self._custom_fields:
export_attrs_set.update(self._custom_fields['children']) export_attrs_set.update(self._custom_fields['children'])
elif self.media_type == 'playlist' and 'items' in self._custom_fields: elif self.media_type == 'playlist' and 'items' in self._custom_fields:
@ -1615,7 +1630,7 @@ class Export(object):
export_attrs_list.append(value) export_attrs_list.append(value)
return reduce(helpers.dict_merge, export_attrs_list) return reduce(helpers.dict_merge, export_attrs_list, {})
def get_any_hdr(self, item, media_type): def get_any_hdr(self, item, media_type):
root = self.return_attrs(media_type)['media'] root = self.return_attrs(media_type)['media']