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

Export images based on level

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

40
plexpy/exporter.py
View file

@ -1780,29 +1780,43 @@ class Export(object):
return any(vs.get('hdr') for p in media.get('parts', []) for vs in p.get('videoStreams', [])) return any(vs.get('hdr') for p in media.get('parts', []) for vs in p.get('videoStreams', []))
def get_image(self, item, image): def get_image(self, item, image):
export_image = False
if self.thumb_level == 1 or self.art_level == 1:
posters = item.arts() if image == 'art' else item.posters()
export_image = any(poster.selected and poster.ratingKey.startswith('upload://')
for poster in posters)
elif self.thumb_level == 2 or self.art_level == 2:
export_image = any(field.locked and field.name == image
for field in item.fields)
elif self.thumb_level >= 9 or self.art_level >= 9:
export_image = True
if not export_image:
return
image_url = None
if image == 'thumb':
image_url = item.thumbUrl
elif image == 'art':
image_url = item.artUrl
if not image_url:
return
media_type = item.type media_type = item.type
rating_key = item.ratingKey rating_key = item.ratingKey
if media_type in ('season', 'episode', 'album', 'track'): if media_type in ('season', 'episode', 'album', 'track'):
item_title = item._defaultSyncTitle() item_title = item._defaultSyncTitle()
else: else:
item_title = item.title item_title = item.title
folder = get_export_filepath(self.filename, images=True) folder = get_export_filepath(self.filename, images=True)
filename = helpers.clean_filename('{} [{}].{}.jpg'.format(item_title, rating_key, image)) filename = helpers.clean_filename('{} [{}].{}.jpg'.format(item_title, rating_key, image))
filepath = os.path.join(folder, filename) filepath = os.path.join(folder, filename)
os.makedirs(folder, exist_ok=True) os.makedirs(folder, exist_ok=True)
image_url = None
if image == 'art':
image_url = item.artUrl
elif image == 'thumb':
image_url = item.thumbUrl
if not image_url:
return
r = requests.get(image_url, stream=True) r = requests.get(image_url, stream=True)
if r.status_code == 200: if r.status_code == 200:
with open(filepath, 'wb') as outfile: with open(filepath, 'wb') as outfile: