Skip to content

Commit 8d29185

Browse files
committed
web: allow overriding referer and user-agent
1 parent 46f46fd commit 8d29185

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

library/utils/arggroups.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1685,6 +1685,10 @@ def requests(parent_parser):
16851685
parser.add_argument(
16861686
"--http-retries", "--http-max-retries", "--retries", type=int, default=8, help="Use N retries for requests"
16871687
)
1688+
parser.add_argument(
1689+
"--referrer", "--http-referrer", "--referer", "--http-referer", help="Use a custom Referer header"
1690+
)
1691+
parser.add_argument("--user-agent", "--http-user-agent", help="Use a custom User-Agent header")
16881692
parser.add_argument(
16891693
"--http-max-redirects",
16901694
"--max-redirects",

library/utils/web.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -89,10 +89,14 @@ def parse_cookies_from_browser(input_str):
8989
def requests_session(args=argparse.Namespace()):
9090
global session
9191

92-
from yt_dlp.utils.networking import std_headers
93-
9492
if session is None:
9593
import requests
94+
from yt_dlp.utils.networking import std_headers
95+
96+
if getattr(args, "referrer"):
97+
std_headers |= {"Referer": args.referrer}
98+
if getattr(args, "user_agent"):
99+
std_headers |= {"User-Agent": args.user_agent}
96100

97101
cookie_file = getattr(args, "cookies", None)
98102
cookies_from_browser = getattr(args, "cookies_from_browser", None)

0 commit comments

Comments
 (0)