Make updates per comments

sigma67 · Jan 14, 2024 · 1efef81 · 1efef81
1 parent 91254ce
commit 1efef81
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/ytmusicapi/parsers/search.py b/ytmusicapi/parsers/search.py
@@ -35,13 +35,8 @@ def parse_top_result(data, search_result_types):
 
     if result_type in ["song", "video", "album"]:
         on_tap = data.get("onTap")
-        if on_tap is not None:
-            if "watchEndpoint" in on_tap:
-                if nav(on_tap, WATCH_VIDEO_ID) is not None:
-                    search_result["videoId"] = nav(on_tap, WATCH_VIDEO_ID)
-            if "navigationEndpoint" in on_tap:
-                if nav(on_tap, NAVIGATION_VIDEO_TYPE) is not None:
-                    search_result["videoType"] = nav(on_tap, NAVIGATION_VIDEO_TYPE)
+        search_result["videoId"] = nav(data, ["onTap"] + WATCH_VIDEO_ID, True)
+        search_result["videoType"] = nav(data, ["onTap"] + NAVIGATION_VIDEO_TYPE, True)
 
         search_result["title"] = nav(data, TITLE_TEXT)
         runs = nav(data, ["subtitle", "runs"])