Skip to content

Commit b1949a0

Browse files
committed
fix discover object methods
1 parent ac2aa1e commit b1949a0

File tree

1 file changed

+15
-6
lines changed

1 file changed

+15
-6
lines changed

tmdbapis/objs/reload.py

+15-6
Original file line numberDiff line numberDiff line change
@@ -440,12 +440,14 @@ class Keyword(TMDbReload):
440440
Attributes:
441441
id (int): Keyword ID.
442442
name (str): Keyword Name.
443-
movies (List[:class:`~tmdbapis.objs.reload.Movie`]): Keyword Movies.
444-
tv_shows (List[:class:`~tmdbapis.objs.reload.TVShow`]): Keyword TV Shows.
443+
movies (:class:`~tmdbapis.objs.pagination.DiscoverMovies`): Keyword Movies.
444+
tv_shows (:class:`~tmdbapis.objs.pagination.DiscoverTVShows`): Keyword TV Shows.
445445
"""
446446

447447
def _load(self, data):
448448
super()._load(data)
449+
self._movies = None
450+
self._tv_shows = None
449451
self.id = self._parse(attrs="id", value_type="int")
450452
self.name = self._parse(attrs="name")
451453
self._finish(self.name)
@@ -455,11 +457,15 @@ def _full_load(self):
455457

456458
@property
457459
def movies(self):
458-
return self._api.discover_movie_discover(with_keywords=self.id)
460+
if not self._movies:
461+
self._movies = self._tmdb.discover_movies(with_keywords=self.id)
462+
return self._movies
459463

460464
@property
461465
def tv_shows(self):
462-
return self._api.discover_tv_discover(with_keywords=self.id)
466+
if not self._tv_shows:
467+
self._tv_shows = self._tmdb.discover_tv_shows(with_keywords=self.id)
468+
return self._tv_shows
463469

464470

465471
class Movie(TMDbReload, Favorite, Rate, Watchlist):
@@ -614,11 +620,12 @@ class Network(TMDbReload):
614620
logo_url (str): Logo Full URL.
615621
logos (List[:class:`~tmdbapis.objs.image.Logo`]): List of other Logos for the Network.
616622
name (str): Network Name.
617-
tv_shows (List[:class:`~tmdbapis.objs.reload.TVShow`]): Network TV Shows.
623+
tv_shows (:class:`~tmdbapis.objs.pagination.DiscoverTVShows`): Network TV Shows.
618624
"""
619625

620626
def _load(self, data):
621627
super()._load(data)
628+
self._tv_shows = None
622629
self.alternative_names = self._parse(attrs=["alternative_names", "results"],
623630
value_type="alternative_name", is_list=True)
624631
self.country = self._parse(attrs="origin_country", value_type="country")
@@ -636,7 +643,9 @@ def _full_load(self):
636643

637644
@property
638645
def tv_shows(self):
639-
return self._api.discover_tv_discover(with_networks=self.id)
646+
if not self._tv_shows:
647+
self._tv_shows = self._tmdb.discover_tv_shows(with_networks=self.id)
648+
return self._tv_shows
640649

641650

642651
class Person(TMDbReload):

0 commit comments

Comments
 (0)