liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

肌肉男功能

Browse Source
This commit is contained in:
liuwei
2025-03-10 10:14:33 +08:00
parent b5b6e44ad3
commit 8e4aafe9af
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
group_video_man/bot_video_man.py
View File

@@ -46,6 +46,14 @@ class BotVideoMan:
return return
def download_video(self, api_url, save_dir): def download_video(self, api_url, save_dir):
# 确保 save_dir 是一个目录
if os.path.isfile(save_dir):
print(f"错误: {save_dir} 是一个文件，不能作为目录使用。")
return None
os.makedirs(save_dir, exist_ok=True)
save_path = os.path.join(save_dir, "video.mp4")
response = requests.get(api_url) response = requests.get(api_url)
if response.status_code == 200: if response.status_code == 200:
data = response.json() data = response.json()
@@ -53,8 +61,6 @@ class BotVideoMan:
if video_url: if video_url:
video_response = requests.get(video_url, stream=True) video_response = requests.get(video_url, stream=True)
if video_response.status_code == 200: if video_response.status_code == 200:
os.makedirs(save_dir, exist_ok=True)
save_path = os.path.join(save_dir, "video.mp4")
with open(save_path, "wb") as file: with open(save_path, "wb") as file:
for chunk in video_response.iter_content(chunk_size=1024): for chunk in video_response.iter_content(chunk_size=1024):
file.write(chunk) file.write(chunk)