肌肉男功能

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: