Adding more info to YT video outputs

Fix #948
medialab · Mar 28, 2024 · 741e064 · 741e064
1 parent 9f0e6cf
commit 741e064
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/ftest/youtube_api.py b/ftest/youtube_api.py
@@ -7,6 +7,6 @@
 
 client = YouTubeAPIClient(config["youtube"]["key"])
 console.print(
-    client.video("https://www.youtube.com/watch?v=uH7e9wumYWg", raw=True),
+    client.video("https://www.youtube.com/watch?v=auPMtIFJGsQ", raw=False),
     highlight=True,
 )
diff --git a/minet/youtube/types.py b/minet/youtube/types.py
@@ -90,6 +90,7 @@ class YouTubeVideo(YouTubeVideoSnippet):
     has_captions: bool
     category_id: str
     category: Optional[str]
+    topics: List[str]
     tags: List[str]
 
     @classmethod
@@ -98,6 +99,7 @@ def from_payload(cls, payload) -> "YouTubeVideo":
         stats = payload["statistics"]
         details = payload["contentDetails"]
         category_id = snippet["categoryId"]
+        topics = getpath(payload, ("topicDetails", "topicCategories"), [])
 
         return cls(
             video_id=payload["id"],
@@ -117,6 +119,7 @@ def from_payload(cls, payload) -> "YouTubeVideo":
             duration=details["duration"],
             has_captions=details["caption"] == "true",
             tags=snippet.get("tags", []),
+            topics=topics,
         )
 
 

diff --git a/minet/youtube/urls.py b/minet/youtube/urls.py
@@ -33,7 +33,10 @@ def playlist_videos(self, playlist_id: str, token: Optional[str]) -> str:
     def videos(self, ids: Iterable[str]) -> str:
         return self.format(
             path="videos",
-            args={"id": ids, "part": ["snippet", "statistics", "contentDetails"]},
+            args={
+                "id": ids,
+                "part": ["snippet", "statistics", "contentDetails", "topicDetails"],
+            },
         )
 
     def channels(self, ids: Iterable[str]) -> str: