aboutsummaryrefslogtreecommitdiffstats
path: root/yt_dlp/__init__.py
diff options
context:
space:
mode:
authorJesús <heckyel@hyperbola.info>2022-03-05 00:59:20 +0800
committerJesús <heckyel@hyperbola.info>2022-03-05 00:59:20 +0800
commitf4f05a6915b8683655b073654d58f70a92f96330 (patch)
treef0fb8acb7acb580030f1b0b758ad03531d6cbb39 /yt_dlp/__init__.py
parent2064247b6faafff3ee304ea6ec58495b8d9ee023 (diff)
parent72e995f1226a2fce3cd7f5fda50ebe024fe2a57e (diff)
downloadhypervideo-pre-f4f05a6915b8683655b073654d58f70a92f96330.tar.lz
hypervideo-pre-f4f05a6915b8683655b073654d58f70a92f96330.tar.xz
hypervideo-pre-f4f05a6915b8683655b073654d58f70a92f96330.zip
updated from upstream | 05/03/2022 at 00:59
Diffstat (limited to 'yt_dlp/__init__.py')
-rw-r--r--yt_dlp/__init__.py15
1 files changed, 6 insertions, 9 deletions
diff --git a/yt_dlp/__init__.py b/yt_dlp/__init__.py
index f308f6a89..524130807 100644
--- a/yt_dlp/__init__.py
+++ b/yt_dlp/__init__.py
@@ -39,6 +39,7 @@ from .utils import (
SameFileError,
setproctitle,
std_headers,
+ traverse_obj,
write_string,
)
from .downloader import (
@@ -72,20 +73,15 @@ def _real_main(argv=None):
parser, opts, args = parseOpts(argv)
warnings, deprecation_warnings = [], []
- # Set user agent
if opts.user_agent is not None:
- std_headers['User-Agent'] = opts.user_agent
-
- # Set referer
+ opts.headers.setdefault('User-Agent', opts.user_agent)
if opts.referer is not None:
- std_headers['Referer'] = opts.referer
-
- # Custom HTTP headers
- std_headers.update(opts.headers)
+ opts.headers.setdefault('Referer', opts.referer)
# Dump user agent
if opts.dump_user_agent:
- write_string(std_headers['User-Agent'] + '\n', out=sys.stdout)
+ ua = traverse_obj(opts.headers, 'User-Agent', casesense=False, default=std_headers['User-Agent'])
+ write_string(f'{ua}\n', out=sys.stdout)
sys.exit(0)
# Batch file verification
@@ -764,6 +760,7 @@ def _real_main(argv=None):
'legacyserverconnect': opts.legacy_server_connect,
'nocheckcertificate': opts.no_check_certificate,
'prefer_insecure': opts.prefer_insecure,
+ 'http_headers': opts.headers,
'proxy': opts.proxy,
'socket_timeout': opts.socket_timeout,
'bidi_workaround': opts.bidi_workaround,