implement showing total view_count

1 files changed, 7 insertions, 11 deletions

diff --git a/webapi/holodex.py b/webapi/holodex.py
index 5f81892..2b92305 100644
--- a/webapi/holodex.py
+++ b/webapi/holodex.py
@@ -31,7 +31,9 @@ class HolodexAPI(WebAPI):
         for channel in data:
             print("DEBUG: ", channel["id"])
             try:
-                channel["description"] = self.get_channel_description(channel["id"])
+                additional_data = self.get_additional_data(channel["id"])
+                channel["view_count"] = additional_data[0]
+                channel["description"] = additional_data[1]
                 if channel["inactive"]:
                     self._inactive_channels.append(channel["id"])
                     continue
@@ -42,19 +44,13 @@ class HolodexAPI(WebAPI):
         self._channel_data = active_channels
         return active_channels
 
-    def get_view_count(self, channel_id: str) -> int:
-        """
-        Gets the view count for a particular channel
-        """
-        data = self._download_url(f"channels/{channel_id}")
-        return data["view_count"]
-
-    def get_channel_description(self, channel_id: str) -> str:
+    
+    def get_additional_data(self, channel_id: str) -> tuple:
         """
-        Gets the description for a particular channel
+        Gets additional data for a particular channel
         """
         data = self._download_url(f"channels/{channel_id}")
-        return data["description"]
+        return data["view_count"], data["description"]
     
     def set_organization(self, organization: str):
         """