Add headers with user_agent to save_video request

This commit is contained in:
Dreyfi 2024-09-30 15:48:54 +10:00 committed by GitHub
parent 9d3d99a595
commit 4ec19fd56a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -161,11 +161,15 @@ def save_video(video_url: str, save_dir: str = "") -> str:
logger.info(f"video already exists: {video_path}")
return video_path
headers = {
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/115.0.0.0 Safari/537.36"
}
# if video does not exist, download it
with open(video_path, "wb") as f:
f.write(
requests.get(
video_url, proxies=config.proxy, verify=False, timeout=(60, 240)
video_url, headers=headers, proxies=config.proxy, verify=False, timeout=(60, 240)
).content
)