Add stats to channel call (#58)

Author: joostlek

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "youtubeaio"
-version = "1.1.0"
+version = "1.1.1"
 description = "Asynchronous Python client for YouTube V3 API."
 authors = ["Joost Lekkerkerker <[email protected]>"]
 maintainers = ["Joost Lekkerkerker <[email protected]>"]

src/youtubeaio/models.py

@@ -102,6 +102,15 @@ class YouTubeChannelContentDetails(BaseModel):
     )
 
 
+class YouTubeChannelStatistics(BaseModel):
+    """Model representing statistics of a channel."""
+
+    view_count: int = Field(..., alias="viewCount")
+    subscriber_count: int = Field(..., alias="subscriberCount")
+    subscriber_count_hidden: bool = Field(..., alias="hiddenSubscriberCount")
+    video_count: int = Field(..., alias="videoCount")
+
+
 class YouTubeChannelSnippet(BaseModel):
     """Model representing channel snippet."""
 
@@ -121,6 +130,7 @@ class YouTubeChannel(BaseModel):
         None,
         alias="contentDetails",
     )
+    statistics: YouTubeChannelStatistics | None = None
 
     @property
     def upload_playlist_id(self) -> str:

src/youtubeaio/youtube.py

@@ -233,7 +233,7 @@ async def get_channels(
     ) -> AsyncGenerator[YouTubeChannel, None]:
         """Return list of channels."""
         param = {
-            "part": "snippet",
+            "part": "snippet,contentDetails,statistics",
             "id": channel_ids,
         }
         async for item in self._get_channels(param):

tests/fixtures/channel_response_snippet.json

@@ -1,14 +1,14 @@
 {
   "kind": "youtube#channelListResponse",
-  "etag": "sM5ffk5W79swi2V870w6CmMlPgs",
+  "etag": "9sIVzjSfOfk6gN357UcUDTWRTec",
   "pageInfo": {
     "totalResults": 1,
     "resultsPerPage": 5
   },
   "items": [
     {
       "kind": "youtube#channel",
-      "etag": "ieb4dig-VKDKW3FEut81IwUEiog",
+      "etag": "veRvdNDE9HmhyENPqeFNq-VH6NI",
       "id": "UC_x5XG1OV2P6uZZ5FSM9Ttw",
       "snippet": {
         "title": "Google for Developers",
@@ -37,6 +37,12 @@
           "description": "Subscribe to join a community of creative developers and learn the latest in Google technology — from AI and cloud, to mobile and web.\n\nExplore more at developers.google.com\n\n"
         },
         "country": "US"
+      },
+      "statistics": {
+        "viewCount": "234084903",
+        "subscriberCount": "2300000",
+        "hiddenSubscriberCount": false,
+        "videoCount": "5783"
       }
     }
   ]