Merge pull request #33 from zeroquinc:trakt-url-fix

feature: support for TMDB url
zeroquinc · Oct 30, 2024 · e6880ee · e6880ee
2 parents 67e1fe3 + 528b962
commit e6880ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/plex/client.py b/api/plex/client.py
@@ -16,7 +16,7 @@ def extract_details(self):
         fields = [
             'media_type', 'year', 'title', 'summary', 'quality', 'air_date', 'genres',
             'release_date', 'season_num00', 'episode_num00', 'episode_count', 'poster_url',
-            'imdb_url', 'tvdb_url', 'trakt_url', 'plex_url', 'critic_rating', 'audience_rating',
+            'imdb_url', 'tvdb_url', 'trakt_url', 'plex_url', 'tmdb_url', 'tmdb_id', 'critic_rating', 'audience_rating',
             'rating', 'username', 'platform', 'player', 'product', 'video_decision',
             'remaining_time', 'duration_time', 'server_name', 'webhook_type'
         ]
@@ -99,7 +99,7 @@ def embed_for_newcontent(self, color):
         elif self.webhook_type == 'newcontent_season':
             embed.add_field(name="Episodes", value=f"{self.episode_count}", inline=False)
         elif self.webhook_type == 'newcontent_movie':
-            embed.add_field(name="Links", value=f"[IMDb]({self.imdb_url}) • [Trakt]({self.trakt_url})")
+            embed.add_field(name="Links", value=f"[IMDb]({self.imdb_url}) • [TMDB]({self.tmdb_url}) • [Trakt](https://trakt.tv/search/imdb?query={self.tmdb_id})")
             embed.set_footer(text=f"{self.genres} • {self.format_duration_time()}")
         embed.set_author(name=f"New {self.media_type.capitalize()} added to Plex", icon_url=PLEX_ICON)
         return embed