Merge pull request #176 from fabi321/main

Made ChannelSearch more resilliant, Fixes #175
alexmercerind · Apr 30, 2022 · 9bbc3d2 · 9bbc3d2
2 parents 77518c5 + 090d9fb
commit 9bbc3d2
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 1 deletion.
diff --git a/tests/async/search.py b/tests/async/search.py
@@ -31,6 +31,10 @@ async def main():
     result = await search.next()
     print(result)
 
+    channel = ChannelSearch('The Beatles - Topic', 'UC2XdaAVUannpujzv32jcouQ')
+    result = await search.next()
+    print(result)
+
 
     search = VideosSearch('NoCopyrightSounds')
     index = 0

diff --git a/tests/sync/search.py b/tests/sync/search.py
@@ -39,3 +39,6 @@
 
 channel = ChannelSearch("Watermelon Sugar", "UCZFWPqqPkFlNwIxcpsLOwew")
 print(channel.result(mode=ResultMode.json))
+
+channel = ChannelSearch('The Beatles - Topic', 'UC2XdaAVUannpujzv32jcouQ')
+print(channel.result(mode=ResultMode.json))
diff --git a/youtubesearchpython/core/channelsearch.py b/youtubesearchpython/core/channelsearch.py
@@ -35,7 +35,15 @@ async def next(self):
 
     def _parseChannelSearchSource(self) -> None:
         try:
-            self.response = self.response["contents"]["twoColumnBrowseResultsRenderer"]["tabs"][-1]["expandableTabRenderer"]["content"]["sectionListRenderer"]["contents"]
+            last_tab = self.response["contents"]["twoColumnBrowseResultsRenderer"]["tabs"][-1]
+            if 'expandableTabRenderer' in last_tab:
+                self.response = last_tab["expandableTabRenderer"]["content"]["sectionListRenderer"]["contents"]
+            else:
+                tab_renderer = last_tab["tabRenderer"]
+                if 'content' in tab_renderer:
+                    self.response = tab_renderer["content"]["sectionListRenderer"]["contents"]
+                else:
+                    self.response = []
         except:
             raise Exception('ERROR: Could not parse YouTube response.')