Data parity for seasons and episodes

plexapi/mixins/__init__.py

@@ -29,6 +29,7 @@
 from .split_merge import SplitMergeMixin
 from .unmatch_match import UnmatchMatchMixin
 from .watchlist import WatchlistMixin
+from .tv_parent_child import TvParentChildMixin
 
 
 class MovieEditMixins(
@@ -140,23 +141,26 @@ class ShowMixins(
     AdvancedSettingsMixin, SplitMergeMixin, UnmatchMatchMixin, ExtrasMixin, HubsMixin, RatingMixin,
     ArtMixin, LogoMixin, PosterMixin, SquareArtMixin, ThemeMixin,
     ShowEditMixins,
-    WatchlistMixin
+    WatchlistMixin,
+    TvParentChildMixin
 ):
     pass
 
 
 class SeasonMixins(
     AdvancedSettingsMixin, ExtrasMixin, RatingMixin,
     ArtMixin, LogoMixin, PosterMixin, SquareArtMixin, ThemeUrlMixin,
-    SeasonEditMixins
+    SeasonEditMixins,
+    TvParentChildMixin
 ):
     pass
 
 
 class EpisodeMixins(
     ExtrasMixin, RatingMixin,
     ArtMixin, LogoMixin, PosterMixin, SquareArtMixin, ThemeUrlMixin,
-    EpisodeEditMixins
+    EpisodeEditMixins,
+    TvParentChildMixin
 ):
     pass
 

plexapi/mixins/tv_parent_child.py

@@ -0,0 +1,17 @@
+from plexapi import utils
+
+
+class TvParentChildMixin:
+    """ Mixin for Plex objects that have parent/child relationships (episode/season/show). """
+
+    def _buildRelationKey(self, key, **kwargs):
+        """ Returns a key suitable for fetching parent/child TV items """
+        args = {}
+
+        args['includeGuids'] = int(bool(kwargs.pop('includeGuids', True)))
+        for name, value in list(kwargs.items()):
+            args[name] = value
+
+        params = utils.joinArgs(args).lstrip('?')
+
+        return f"{key}?{params}"

plexapi/video.py

@@ -710,7 +710,7 @@ def season(self, title=None, season=None):
             Raises:
                 :exc:`~plexapi.exceptions.BadRequest`: If title or season parameter is missing.
         """
-        key = f'{self.key}/children?excludeAllLeaves=1'
+        key = self._buildRelationKey(f'{self.key}/children', excludeAllLeaves=1)
         if title is not None and not isinstance(title, int):
             return self.fetchItem(key, Season, title__iexact=title)
         elif season is not None or isinstance(title, int):
@@ -723,7 +723,7 @@ def season(self, title=None, season=None):
 
     def seasons(self, **kwargs):
         """ Returns a list of :class:`~plexapi.video.Season` objects in the show. """
-        key = f'{self.key}/children?excludeAllLeaves=1'
+        key = self._buildRelationKey(f'{self.key}/children', excludeAllLeaves=1)
         return self.fetchItems(key, Season, container_size=self.childCount, **kwargs)
 
     def episode(self, title=None, season=None, episode=None):
@@ -737,7 +737,7 @@ def episode(self, title=None, season=None, episode=None):
             Raises:
                 :exc:`~plexapi.exceptions.BadRequest`: If title or season and episode parameters are missing.
         """
-        key = f'{self.key}/allLeaves'
+        key = self._buildRelationKey(f'{self.key}/allLeaves')
         if title is not None:
             return self.fetchItem(key, Episode, title__iexact=title)
         elif season is not None and episode is not None:
@@ -746,7 +746,7 @@ def episode(self, title=None, season=None, episode=None):
 
     def episodes(self, **kwargs):
         """ Returns a list of :class:`~plexapi.video.Episode` objects in the show. """
-        key = f'{self.key}/allLeaves'
+        key = self._buildRelationKey(f'{self.key}/allLeaves')
         return self.fetchItems(key, Episode, **kwargs)
 
     def get(self, title=None, season=None, episode=None):
@@ -906,7 +906,7 @@ def episode(self, title=None, episode=None):
             Raises:
                 :exc:`~plexapi.exceptions.BadRequest`: If title or episode parameter is missing.
         """
-        key = f'{self.key}/children'
+        key = self._buildRelationKey(f'{self.key}/children')
         if title is not None and not isinstance(title, int):
             return self.fetchItem(key, Episode, title__iexact=title)
         elif episode is not None or isinstance(title, int):
@@ -919,7 +919,7 @@ def episode(self, title=None, episode=None):
 
     def episodes(self, **kwargs):
         """ Returns a list of :class:`~plexapi.video.Episode` objects in the season. """
-        key = f'{self.key}/children'
+        key = self._buildRelationKey(f'{self.key}/children')
         return self.fetchItems(key, Episode, **kwargs)
 
     def get(self, title=None, episode=None):
@@ -928,7 +928,7 @@ def get(self, title=None, episode=None):
 
     def show(self):
         """ Return the season's :class:`~plexapi.video.Show`. """
-        return self.fetchItem(self.parentKey)
+        return self.fetchItem(self._buildRelationKey(self.parentKey))
 
     def watched(self):
         """ Returns list of watched :class:`~plexapi.video.Episode` objects. """
@@ -1136,7 +1136,12 @@ def parentThumb(self):
     def _season(self):
         """ Returns the :class:`~plexapi.video.Season` object by querying for the show's children. """
         if self.grandparentKey and self.parentIndex is not None:
-            return self.fetchItem(f'{self.grandparentKey}/children?excludeAllLeaves=1&index={self.parentIndex}')
+            key = self._buildRelationKey(
+                f'{self.grandparentKey}/children',
+                excludeAllLeaves=1,
+                index=self.parentIndex
+            )
+            return self.fetchItem(key)
         return None
 
     def __repr__(self):
@@ -1213,11 +1218,11 @@ def hasPreviewThumbnails(self):
 
     def season(self):
         """" Return the episode's :class:`~plexapi.video.Season`. """
-        return self.fetchItem(self.parentKey)
+        return self.fetchItem(self._buildRelationKey(self.parentKey))
 
     def show(self):
         """" Return the episode's :class:`~plexapi.video.Show`. """
-        return self.fetchItem(self.grandparentKey)
+        return self.fetchItem(self._buildRelationKey(self.grandparentKey))
 
     def _defaultSyncTitle(self):
         """ Returns str, default title for a new syncItem. """

tests/test_video.py

@@ -4,6 +4,7 @@
 from urllib.parse import quote_plus
 
 import pytest
+import plexapi.base
 import plexapi.utils as plexutils
 from plexapi.exceptions import BadRequest, NotFound
 from plexapi.utils import setDatetimeTimezone
@@ -990,6 +991,30 @@ def test_video_Show_isPlayed(show):
     assert not show.isPlayed
 
 
+def test_video_Show_season_guids(show):
+    plexapi.base.USER_DONT_RELOAD_FOR_KEYS.add('guids')
+    try:
+        season = show.season("Season 1")
+        assert season.guids
+        seasons = show.seasons()
+        assert len(seasons) > 0
+        assert seasons[0].guids
+    finally:
+        plexapi.base.USER_DONT_RELOAD_FOR_KEYS.remove('guids')
+
+
+def test_video_Show_episode_guids(show):
+    plexapi.base.USER_DONT_RELOAD_FOR_KEYS.add('guids')
+    try:
+        episode = show.episode("Winter Is Coming")
+        assert episode.guids
+        episodes = show.episodes()
+        assert len(episodes) > 0
+        assert episodes[0].guids
+    finally:
+        plexapi.base.USER_DONT_RELOAD_FOR_KEYS.remove('guids')
+
+
 def test_video_Show_section(show):
     section = show.section()
     assert section.title == "TV Shows"
@@ -1175,6 +1200,7 @@ def test_video_Season_show(show):
     season_by_name = show.season("Season 1")
     assert show.ratingKey == season.parentRatingKey and season_by_name.parentRatingKey
     assert season.ratingKey == season_by_name.ratingKey
+    assert season.guids
 
 
 def test_video_Season_watched(show):
@@ -1213,6 +1239,29 @@ def test_video_Season_episodes(show):
     assert len(episodes) >= 1
 
 
+def test_video_Season_episode_guids(show):
+    plexapi.base.USER_DONT_RELOAD_FOR_KEYS.add('guids')
+    try:
+        season = show.season("Season 1")
+        episode = season.episode("Winter Is Coming")
+        assert episode.guids
+        episodes = season.episodes()
+        assert len(episodes) > 0
+        assert episodes[0].guids
+    finally:
+        plexapi.base.USER_DONT_RELOAD_FOR_KEYS.remove('guids')
+
+
+def test_video_Season_show_guids(show):
+    plexapi.base.USER_DONT_RELOAD_FOR_KEYS.add('guids')
+    try:
+        a_show = show.season("Season 1").show()
+        assert a_show
+        assert 'tmdb://1399' in [i.id for i in a_show.guids]
+    finally:
+        plexapi.base.USER_DONT_RELOAD_FOR_KEYS.remove('guids')
+
+
 @pytest.mark.xfail(reason="Changing images fails randomly")
 def test_video_Season_mixins_images(show):
     season = show.season(season=1)
@@ -1283,6 +1332,30 @@ def test_video_Episode(show):
         show.episode(season=1337, episode=1337)
 
 
+def test_video_Episode_parent_guids(show):
+    plexapi.base.USER_DONT_RELOAD_FOR_KEYS.add('guids')
+    try:
+        episodes = show.episodes()
+        assert episodes
+        episode = episodes[0]
+        assert episode
+        assert episode.isPartialObject()
+        season = episode._season
+        assert season
+        assert season.isPartialObject()
+        assert season.guids
+        season = episode.season()
+        assert season
+        assert season.isPartialObject()
+        assert season.guids
+        show = episode.show()
+        assert show
+        assert show.isPartialObject()
+        assert show.guids
+    finally:
+        plexapi.base.USER_DONT_RELOAD_FOR_KEYS.remove('guids')
+
+
 def test_video_Episode_hidden_season(episode):
     assert episode.skipParent is False
     assert episode.parentRatingKey