Cool_Tools/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-12 02:01:05 +01:00
Code Issues Packages Projects Releases Wiki Activity

[utils] Escape URLs in sanitized_Request, not sanitize_url

Browse Source 
d2558234cf added escaping of URLs while sanitizing. However, `sanitize_url` may not always receive an actual URL.
Eg: When using `yt-dlp "search query" --default-search ytsearch`, `search query` gets escaped to `search%20query` before being prefixed with `ytsearch:` which is not the intended behavior. So the escaping is moved to `sanitized_Request` instead.
This commit is contained in:
pukkandan 2021-06-01 18:05:41 +05:30
parent 6e6390321c
commit bc6b9bcd65
No known key found for this signature in database
GPG Key ID: 0F00D95A001F4698
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
test/test_utils.py
View File

@ -239,6 +239,7 @@ def test_sanitize_url(self):
self.assertEqual(sanitize_url('httpss://foo.bar'), 'https://foo.bar') self.assertEqual(sanitize_url('httpss://foo.bar'), 'https://foo.bar')
self.assertEqual(sanitize_url('rmtps://foo.bar'), 'rtmps://foo.bar') self.assertEqual(sanitize_url('rmtps://foo.bar'), 'rtmps://foo.bar')
self.assertEqual(sanitize_url('https://foo.bar'), 'https://foo.bar') self.assertEqual(sanitize_url('https://foo.bar'), 'https://foo.bar')
self.assertEqual(sanitize_url('foo bar'), 'foo bar')
def test_extract_basic_auth(self): def test_extract_basic_auth(self):
auth_header = lambda url: sanitized_Request(url).get_header('Authorization') auth_header = lambda url: sanitized_Request(url).get_header('Authorization')

4
yt_dlp/utils.py
View File

@ -2165,7 +2165,7 @@ def sanitize_url(url):
for mistake, fixup in COMMON_TYPOS: for mistake, fixup in COMMON_TYPOS:
if re.match(mistake, url): if re.match(mistake, url):
return re.sub(mistake, fixup, url) return re.sub(mistake, fixup, url)
return escape_url(url) return url
def extract_basic_auth(url): def extract_basic_auth(url):
@ -2181,7 +2181,7 @@ def extract_basic_auth(url):
def sanitized_Request(url, *args, **kwargs): def sanitized_Request(url, *args, **kwargs):
url, auth_header = extract_basic_auth(sanitize_url(url)) url, auth_header = extract_basic_auth(escape_url(sanitize_url(url)))
if auth_header is not None: if auth_header is not None:
headers = args[1] if len(args) >= 2 else kwargs.setdefault('headers', {}) headers = args[1] if len(args) >= 2 else kwargs.setdefault('headers', {})
headers['Authorization'] = auth_header headers['Authorization'] = auth_header