github/plexpy
1
0
Fork 0
mirror of https://github.com/Tautulli/Tautulli.git synced 2025-08-19 12:59:42 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Refactor export get all metadata attrs

Browse source
This commit is contained in:
JonnyWong16 2020-09-20 21:21:37 -07:00
commit d7c96d46e0
No known key found for this signature in database
GPG key ID: B1F1F9807184697A
1 changed files with 30 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

54
plexpy/exporter.py
View file

@ -53,6 +53,31 @@ class Export(object):
'playlist' 'playlist'
) )
def __init__(self, section_id=None, rating_key=None, file_format='json',
metadata_level=1, media_info_level=1, include_images=False):
self.section_id = helpers.cast_to_int(section_id)
self.rating_key = helpers.cast_to_int(rating_key)
self.file_format = file_format
self.metadata_level = helpers.cast_to_int(metadata_level)
self.media_info_level = helpers.cast_to_int(media_info_level)
self.include_images = include_images
self.timestamp = helpers.timestamp()
self.media_type = None
self.items = []
self.filename = None
self.filename_ext = None
self.export_id = None
self.file_size = None
self.success = False
def _get_all_metadata_attr(self, media_type):
exclude_attrs = ('media', 'artFile', 'thumbFile')
all_attrs = self.return_attrs(media_type)
return [attr for attr in all_attrs if attr not in exclude_attrs]
def return_attrs(self, media_type): def return_attrs(self, media_type):
def movie_attrs(): def movie_attrs():
_movie_attrs = { _movie_attrs = {
@ -935,7 +960,7 @@ class Export(object):
1: [ 1: [
'ratingKey', 'title', 'titleSort', 'originalTitle', 'originallyAvailableAt', 'year', 'addedAt', 'ratingKey', 'title', 'titleSort', 'originalTitle', 'originallyAvailableAt', 'year', 'addedAt',
'rating', 'ratingImage', 'audienceRating', 'audienceRatingImage', 'userRating', 'contentRating', 'rating', 'ratingImage', 'audienceRating', 'audienceRatingImage', 'userRating', 'contentRating',
'studio', 'tagline', 'summary', 'guid', 'duration', 'durationHuman', 'type', 'artFile', 'thumbFile' 'studio', 'tagline', 'summary', 'guid', 'duration', 'durationHuman', 'type'
], ],
2: [ 2: [
'directors.tag', 'writers.tag', 'producers.tag', 'roles.tag', 'roles.role', 'directors.tag', 'writers.tag', 'producers.tag', 'roles.tag', 'roles.role',
@ -946,7 +971,7 @@ class Export(object):
'chapters.tag', 'chapters.index', 'chapters.start', 'chapters.end', 'chapters.thumb', 'chapters.tag', 'chapters.index', 'chapters.start', 'chapters.end', 'chapters.thumb',
'updatedAt', 'lastViewedAt', 'viewCount' 'updatedAt', 'lastViewedAt', 'viewCount'
], ],
9: [k for k in self.return_attrs('movie') if k != 'media'] 9: self._get_all_metadata_attr('movie')
}, },
{ {
1: [ 1: [
@ -1007,7 +1032,7 @@ class Export(object):
'art', 'thumb', 'banner', 'theme', 'key', 'art', 'thumb', 'banner', 'theme', 'key',
'updatedAt', 'lastViewedAt', 'viewCount' 'updatedAt', 'lastViewedAt', 'viewCount'
] + ['seasons.' + attr for attr in self.return_levels('season')[0][3]], ] + ['seasons.' + attr for attr in self.return_levels('season')[0][3]],
9: [k for k in self.return_attrs('show') if k != 'media'] 9: self._get_all_metadata_attr('show')
}, },
{ {
1: ['seasons.' + attr for attr in self.return_levels('season')[1][1]], 1: ['seasons.' + attr for attr in self.return_levels('season')[1][1]],
@ -1035,7 +1060,7 @@ class Export(object):
'updatedAt', 'lastViewedAt', 'viewCount', 'updatedAt', 'lastViewedAt', 'viewCount',
'parentKey', 'parentTheme', 'parentThumb' 'parentKey', 'parentTheme', 'parentThumb'
] + ['episodes.' + attr for attr in self.return_levels('episode')[0][3]], ] + ['episodes.' + attr for attr in self.return_levels('episode')[0][3]],
9: [k for k in self.return_attrs('season') if k != 'media'] 9: self._get_all_metadata_attr('season')
}, },
{ {
1: ['episodes.' + attr for attr in self.return_levels('episode')[1][1]], 1: ['episodes.' + attr for attr in self.return_levels('episode')[1][1]],
@ -1066,7 +1091,7 @@ class Export(object):
'parentThumb', 'parentKey', 'parentThumb', 'parentKey',
'grandparentArt', 'grandparentThumb', 'grandparentTheme', 'grandparentKey' 'grandparentArt', 'grandparentThumb', 'grandparentTheme', 'grandparentKey'
], ],
9: [k for k in self.return_attrs('episode') if k != 'media'] 9: self._get_all_metadata_attr('episode')
}, },
{ {
1: [ 1: [
@ -1155,25 +1180,6 @@ class Export(object):
return _media_types[media_type]() return _media_types[media_type]()
def __init__(self, section_id=None, rating_key=None, file_format='json',
metadata_level=1, media_info_level=1):
self.section_id = helpers.cast_to_int(section_id)
self.rating_key = helpers.cast_to_int(rating_key)
self.file_format = file_format
self.metadata_level = helpers.cast_to_int(metadata_level)
self.media_info_level = helpers.cast_to_int(media_info_level)
self.timestamp = helpers.timestamp()
self.media_type = None
self.items = []
self.filename = None
self.filename_ext = None
self.export_id = None
self.file_size = None
self.success = False
def export(self): def export(self):
if not self.section_id and not self.rating_key: if not self.section_id and not self.rating_key:
logger.error("Tautulli Exporter :: Export called but no section_id or rating_key provided.") logger.error("Tautulli Exporter :: Export called but no section_id or rating_key provided.")