heckyel/yt-local
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

common: support for POST data in fetch_url

Browse Source
This commit is contained in:
James Taylor 2018-12-23 02:16:15 -08:00
parent 6217e6f558
commit b1fff10065
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
youtube/common.py
View File

@ -148,20 +148,19 @@ def decode_content(content, encoding_header):
content = gzip.decompress(content) content = gzip.decompress(content)
return content return content
def fetch_url(url, headers=(), timeout=15, report_text=None): def fetch_url(url, headers=(), timeout=15, report_text=None, data=None):
if isinstance(headers, list): headers = dict(headers) # Note: Calling dict() on a dict will make a copy
headers += [('Accept-Encoding', 'gzip, br')] headers['Accept-Encoding'] = 'gzip, br'
headers = dict(headers)
elif isinstance(headers, tuple):
headers += (('Accept-Encoding', 'gzip, br'),)
headers = dict(headers)
else:
headers = headers.copy()
headers['Accept-Encoding'] = 'gzip, br'
if data is not None:
if isinstance(data, str):
data = data.encode('ascii')
elif not isinstance(data, bytes):
data = urllib.parse.urlencode(data).encode('ascii')
start_time = time.time() start_time = time.time()
req = urllib.request.Request(url, headers=headers) req = urllib.request.Request(url, data=data, headers=headers)
response = urllib.request.urlopen(req, timeout=timeout) response = urllib.request.urlopen(req, timeout=timeout)
response_time = time.time() response_time = time.time()