77 lines
2.9 KiB
Python
77 lines
2.9 KiB
Python
import logging
|
||
import tomllib
|
||
import os
|
||
import requests
|
||
from wcferry import WxMsg, Wcf
|
||
|
||
from robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager
|
||
|
||
|
||
class BotVideoMan:
|
||
def __init__(self, wcf: Wcf, gbm: GroupBotManager):
|
||
self.LOG = logging.getLogger(__name__)
|
||
self.wcf = wcf # 假设 wcf 对象在此类中初始化
|
||
self.gbm = gbm # 权限功能
|
||
with open("group_video_man/config.toml", "rb") as f:
|
||
plugin_config = tomllib.load(f)
|
||
|
||
config = plugin_config["VideoMan"]
|
||
|
||
self.enable = config["enable"]
|
||
self.command = config["command"]
|
||
self.LOG.info(f"[猛男视频] 组件初始化完成，指令： {self.command}")
|
||
|
||
def get_video(self, message: WxMsg):
|
||
if not self.enable:
|
||
return
|
||
|
||
content = str(message.content).strip()
|
||
command = content.split(" ")
|
||
|
||
if command[0] not in self.command:
|
||
return
|
||
|
||
# 如果触发了指令，但是没有权限，则返回权限不足
|
||
if self.gbm.get_group_permission(message.roomid, Feature.VIDEO_MAN) == PermissionStatus.DISABLED:
|
||
return
|
||
|
||
try:
|
||
file_abspath = self.download_video("https://api.guiguiya.com/api/video/fuji?type=json",
|
||
"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:
|
||
self.wcf.send_text(f"❌请求出错：{e}",
|
||
(message.roomid if message.from_group() else message.sender), message.sender)
|
||
return
|
||
|
||
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)
|
||
if response.status_code == 200:
|
||
data = response.json()
|
||
video_url = data.get("url")
|
||
if video_url:
|
||
video_response = requests.get(video_url, stream=True)
|
||
if video_response.status_code == 200:
|
||
with open(save_path, "wb") as file:
|
||
for chunk in video_response.iter_content(chunk_size=1024):
|
||
file.write(chunk)
|
||
abs_path = os.path.abspath(save_path)
|
||
print(f"视频已下载至: {abs_path}")
|
||
return abs_path
|
||
else:
|
||
print("无法下载视频，HTTP 状态码:", video_response.status_code)
|
||
else:
|
||
print("API 响应中没有找到视频 URL")
|
||
else:
|
||
print("API 请求失败，HTTP 状态码:", response.status_code)
|
||
return None
|