hzhang/MoneyPrinterTurbo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add headers with user_agent to save_video request

Browse Source
This commit is contained in:
Dreyfi
2024-09-30 15:48:54 +10:00
committed by GitHub
parent 9d3d99a595
commit 4ec19fd56a
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app/services/material.py
View File

@@ -161,11 +161,15 @@ def save_video(video_url: str, save_dir: str = "") -> str:
logger.info(f"video already exists: {video_path}") logger.info(f"video already exists: {video_path}")
return 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 # if video does not exist, download it
with open(video_path, "wb") as f: with open(video_path, "wb") as f:
f.write( f.write(
requests.get( 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 ).content
) )