Merge pull request #216 from felipeucelli/search-update

[Shorts] fixed KeyError: 'reelItemRenderer'
JuanBindez · Sep 8, 2024 · 2bd7b99 · 2bd7b99
2 parents ee39697 + a936d1d
commit 2bd7b99
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/pytubefix/contrib/search.py b/pytubefix/contrib/search.py
@@ -48,8 +48,8 @@ def __init__(
             then passed as a `po_token` query parameter to affected clients.
             If allow_oauth_cache is set to True, the user should only be prompted once.
         :param Callable po_token_verifier:
-            (Optional) Verified used to obtain the visitorData and po_tokenoken.
-            The verifier will return the visitorData and po_tokenoken respectively.
+            (Optional) Verified used to obtain the visitorData and po_token.
+            The verifier will return the visitorData and po_token respectively.
             (if passed, else default verifier will be used)
         """
         self.query = query
@@ -306,8 +306,13 @@ def fetch_and_parse(self, continuation=None):
                 # Get shorts results
                 if 'reelShelfRenderer' in video_details:
                     for items in video_details['reelShelfRenderer']['items']:
-                        shorts.append(YouTube(f"https://www.youtube.com/watch?v="
-                                              f"{items['reelItemRenderer']['videoId']}",
+                        if 'reelItemRenderer' in items:
+                            video_id = items['reelItemRenderer']['videoId']
+                        else:
+                            video_id = items['shortsLockupViewModel']['inlinePlayerData']['onVisible'][
+                                'innertubeCommand']['watchEndpoint']['videoId']
+
+                        shorts.append(YouTube(f"https://www.youtube.com/watch?v={video_id}",
                                               use_oauth=self.use_oauth,
                                               allow_oauth_cache=self.allow_oauth_cache,
                                               token_file=self.token_file,