From b5903938e19396d7d5228798d3c7922844f78d99 Mon Sep 17 00:00:00 2001 From: liuwei Date: Mon, 10 Mar 2025 10:11:32 +0800 Subject: [PATCH] =?UTF-8?q?=E8=82=8C=E8=82=89=E7=94=B7=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- group_video_man/bot_video_man.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/group_video_man/bot_video_man.py b/group_video_man/bot_video_man.py index 11239d5..38462a6 100644 --- a/group_video_man/bot_video_man.py +++ b/group_video_man/bot_video_man.py @@ -37,7 +37,7 @@ class BotVideoMan: try: file_abspath = self.download_video("https://api.guiguiya.com/api/video/fuji?type=json", - "group_video_man/down_load_dir/video.mp4") + "group_video_man/down_load_dir") self.wcf.send_file(file_abspath, (message.roomid if message.from_group() else message.sender)) return except requests.RequestException as e: @@ -45,7 +45,7 @@ class BotVideoMan: (message.roomid if message.from_group() else message.sender), message.sender) return - def download_video(self, api_url, save_path): + def download_video(api_url, save_dir): response = requests.get(api_url) if response.status_code == 200: data = response.json() @@ -53,10 +53,14 @@ class BotVideoMan: if video_url: video_response = requests.get(video_url, stream=True) 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: for chunk in video_response.iter_content(chunk_size=1024): file.write(chunk) - return os.path.abspath(save_path) + abs_path = os.path.abspath(save_path) + print(f"视频已下载至: {abs_path}") + return abs_path else: print("无法下载视频，HTTP 状态码:", video_response.status_code) else: