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:11:32 +08:00
parent e070eff659
commit b5903938e1
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
group_video_man/bot_video_man.py
View File

@@ -37,7 +37,7 @@ class BotVideoMan:
try: try:
file_abspath = self.download_video("https://api.guiguiya.com/api/video/fuji?type=json", 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)) self.wcf.send_file(file_abspath, (message.roomid if message.from_group() else message.sender))
return return
except requests.RequestException as e: except requests.RequestException as e:
@@ -45,7 +45,7 @@ class BotVideoMan:
(message.roomid if message.from_group() else message.sender), message.sender) (message.roomid if message.from_group() else message.sender), message.sender)
return return
def download_video(self, api_url, save_path): def download_video(api_url, save_dir):
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,10 +53,14 @@ 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)
return os.path.abspath(save_path) abs_path = os.path.abspath(save_path)
print(f"视频已下载至: {abs_path}")
return abs_path
else: else:
print("无法下载视频，HTTP 状态码:", video_response.status_code) print("无法下载视频，HTTP 状态码:", video_response.status_code)
else: else: