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

Don't use brotli if import fails

Browse Source
This commit is contained in:
James Taylor 2020-08-11 21:57:40 -07:00
parent cc123d6761
commit bd255a9ab0
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
youtube/util.py
View File

@ -2,7 +2,11 @@ import settings
from youtube import yt_data_extract from youtube import yt_data_extract
import socks, sockshandler import socks, sockshandler
import gzip import gzip
import brotli try:
import brotli
have_brotli = True
except ImportError:
have_brotli = False
import urllib.parse import urllib.parse
import re import re
import time import time
@ -125,7 +129,10 @@ def fetch_url(url, headers=(), timeout=15, report_text=None, data=None, cookieja
and response cookies will be merged into it. and response cookies will be merged into it.
''' '''
headers = dict(headers) # Note: Calling dict() on a dict will make a copy headers = dict(headers) # Note: Calling dict() on a dict will make a copy
headers['Accept-Encoding'] = 'gzip, br' if have_brotli:
headers['Accept-Encoding'] = 'gzip, br'
else:
headers['Accept-Encoding'] = 'gzip'
# prevent python version being leaked by urllib if User-Agent isn't provided # prevent python version being leaked by urllib if User-Agent isn't provided
# (urllib will use ex. Python-urllib/3.6 otherwise) # (urllib will use ex. Python-urllib/3.6 otherwise)