Fix error getting exit node ip if format urls are None

2020-05-27 11:14:52 -07:00 · 2020-05-27 11:14:52 -07:00 · f1f77c4d77
commit f1f77c4d77
parent d5e661dfd3
1 changed files with 1 additions and 1 deletions
--- a/youtube/yt_data_extract/watch_extraction.py
+++ b/youtube/yt_data_extract/watch_extraction.py
@ -364,7 +364,7 @@ def _extract_formats(info, player_response):

    # get ip address
    if info['formats']:
-        query_string = info['formats'][0].get('url', '?').split('?')[1]
+        query_string = (info['formats'][0].get('url') or '?').split('?')[1]
        info['ip_address'] = deep_get(
            urllib.parse.parse_qs(query_string), 'ip', 0)
    else: