Skip to content

Commit

Permalink
Overwrite image provider export field for uploaded assets
Browse files Browse the repository at this point in the history
  • Loading branch information
@JonnyWong16
JonnyWong16 committed May 14, 2024
1 parent ee0b4c0 commit 68bf1c7
Showing 1 changed file with 23 additions and 21 deletions.
44 changes: 23 additions & 21 deletions plexpy/exporter.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -148,7 +148,7 @@ def movie_attrs():
'art': None,
'artBlurHash': None,
'artFile': lambda o: self.get_image(o, 'art'),
'artProvider': lambda o: self.get_image_attr(o, 'art', 'provider'),
'artProvider': lambda o: self.get_image_provider(o, 'art'),
'audienceRating': None,
'audienceRatingImage': None,
'chapters': {
Expand Down Expand Up @@ -385,7 +385,7 @@ def movie_attrs():
'thumb': None,
'thumbBlurHash': None,
'thumbFile': lambda o: self.get_image(o, 'thumb'),
'thumbProvider': lambda o: self.get_image_attr(o, 'thumb', 'provider'),
'thumbProvider': lambda o: self.get_image_provider(o, 'thumb'),
'title': None,
'titleSort': None,
'type': None,
Expand All @@ -408,7 +408,7 @@ def show_attrs():
'art': None,
'artBlurHash': None,
'artFile': lambda o: self.get_image(o, 'art'),
'artProvider': lambda o: self.get_image_attr(o, 'art', 'provider'),
'artProvider': lambda o: self.get_image_provider(o, 'art'),
'audienceRating': None,
'audienceRatingImage': None,
'audioLanguage': None,
Expand Down Expand Up @@ -475,7 +475,7 @@ def show_attrs():
'thumb': None,
'thumbBlurHash': None,
'thumbFile': lambda o: self.get_image(o, 'thumb'),
'thumbProvider': lambda o: self.get_image_attr(o, 'thumb', 'provider'),
'thumbProvider': lambda o: self.get_image_provider(o, 'thumb'),
'title': None,
'titleSort': None,
'type': None,
Expand All @@ -495,7 +495,7 @@ def season_attrs():
'art': None,
'artBlurHash': None,
'artFile': lambda o: self.get_image(o, 'art'),
'artProvider': lambda o: self.get_image_attr(o, 'art', 'provider'),
'artProvider': lambda o: self.get_image_provider(o, 'art'),
'audioLanguage': None,
'collections': {
'id': None,
Expand Down Expand Up @@ -539,7 +539,7 @@ def season_attrs():
'thumb': None,
'thumbBlurHash': None,
'thumbFile': lambda o: self.get_image(o, 'thumb'),
'thumbProvider': lambda o: self.get_image_attr(o, 'thumb', 'provider'),
'thumbProvider': lambda o: self.get_image_provider(o, 'thumb'),
'title': None,
'titleSort': None,
'type': None,
Expand All @@ -558,7 +558,7 @@ def episode_attrs():
'art': None,
'artBlurHash': None,
'artFile': lambda o: self.get_image(o, 'art'),
'artProvider': lambda o: self.get_image_attr(o, 'art', 'provider'),
'artProvider': lambda o: self.get_image_provider(o, 'art'),
'audienceRating': None,
'audienceRatingImage': None,
'chapters': {
Expand Down Expand Up @@ -799,7 +799,7 @@ def episode_attrs():
'thumb': None,
'thumbBlurHash': None,
'thumbFile': lambda o: self.get_image(o, 'thumb'),
'thumbProvider': lambda o: self.get_image_attr(o, 'thumb', 'provider'),
'thumbProvider': lambda o: self.get_image_provider(o, 'thumb'),
'title': None,
'titleSort': None,
'type': None,
Expand All @@ -822,7 +822,7 @@ def artist_attrs():
'art': None,
'artBlurHash': None,
'artFile': lambda o: self.get_image(o, 'art'),
'artProvider': lambda o: self.get_image_attr(o, 'art', 'provider'),
'artProvider': lambda o: self.get_image_provider(o, 'art'),
'collections': {
'id': None,
'tag': None
Expand Down Expand Up @@ -875,7 +875,7 @@ def artist_attrs():
'thumb': None,
'thumbBlurHash': None,
'thumbFile': lambda o: self.get_image(o, 'thumb'),
'thumbProvider': lambda o: self.get_image_attr(o, 'thumb', 'provider'),
'thumbProvider': lambda o: self.get_image_provider(o, 'thumb'),
'title': None,
'titleSort': None,
'type': None,
Expand All @@ -892,7 +892,7 @@ def album_attrs():
'art': None,
'artBlurHash': None,
'artFile': lambda o: self.get_image(o, 'art'),
'artProvider': lambda o: self.get_image_attr(o, 'art', 'provider'),
'artProvider': lambda o: self.get_image_provider(o, 'art'),
'collections': {
'id': None,
'tag': None
Expand Down Expand Up @@ -955,7 +955,7 @@ def album_attrs():
'thumb': None,
'thumbBlurHash': None,
'thumbFile': lambda o: self.get_image(o, 'thumb'),
'thumbProvider': lambda o: self.get_image_attr(o, 'thumb', 'provider'),
'thumbProvider': lambda o: self.get_image_provider(o, 'thumb'),
'title': None,
'titleSort': None,
'type': None,
Expand Down Expand Up @@ -1235,7 +1235,7 @@ def collection_attrs():
'art': None,
'artBlurHash': None,
'artFile': lambda o: self.get_image(o, 'art'),
'artProvider': lambda o: self.get_image_attr(o, 'art', 'provider'),
'artProvider': lambda o: self.get_image_provider(o, 'art'),
'childCount': None,
'collectionFilterBasedOnUser': None,
'collectionMode': None,
Expand Down Expand Up @@ -1266,7 +1266,7 @@ def collection_attrs():
'thumb': None,
'thumbBlurHash': None,
'thumbFile': lambda o: self.get_image(o, 'thumb'),
'thumbProvider': lambda o: self.get_image_attr(o, 'thumb', 'provider'),
'thumbProvider': lambda o: self.get_image_provider(o, 'poster'),
'title': None,
'titleSort': None,
'type': None,
Expand Down Expand Up @@ -2262,16 +2262,19 @@ def _get_cached_images(self, item, image):
if not hasattr(item, '_posters'):
item._posters = item.posters()
return getattr(item, '_posters', [])

def _get_selected_image(self, item, image):
images = self._get_cached_images(item, image)
return next((im for im in images if im.selected), None)

def get_image(self, item, image):
media_type = item.type
rating_key = item.ratingKey

export_image = True
if self.thumb_level == 1 or self.art_level == 1:
images = self._get_cached_images(item, image)
export_image = any(im.selected and im.ratingKey.startswith('upload://')
for im in images)
selected = self._get_selected_image(item, image)
export_image = selected and selected.ratingKey.startswith('upload://')
elif self.thumb_level == 2 or self.art_level == 2:
export_image = any(field.locked and field.name == image
for field in item.fields)
Expand Down Expand Up @@ -2317,11 +2320,10 @@ def get_image(self, item, image):

return os.path.join(os.path.basename(dirpath), filename)

def get_image_attr(self, item, image, attr):
images = self._get_cached_images(item, image)
selected = next((im for im in images if im.selected), None)
def get_image_provider(self, item, image):
selected = self._get_selected_image(item, image)
if selected:
return getattr(selected, attr)
return 'upload' if selected.ratingKey.startswith('upload://') else selected.provider


class ExportObject(Export):
Expand Down

0 comments on commit 68bf1c7

Please sign in to comment.