342 lines
14 KiB
Python
342 lines
14 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
Bilibili 工具基础模块
|
||
提供 B 站视频信息获取和视频分析功能
|
||
"""
|
||
|
||
import asyncio
|
||
import re
|
||
from typing import Any
|
||
|
||
import aiohttp
|
||
|
||
from src.chat.utils.utils_video import get_video_analyzer
|
||
from src.common.logger import get_logger
|
||
|
||
logger = get_logger("bilibili_tool")
|
||
|
||
|
||
class BilibiliVideoAnalyzer:
|
||
"""哔哩哔哩视频分析器，集成视频下载和AI分析功能"""
|
||
|
||
def __init__(self):
|
||
self.video_analyzer = get_video_analyzer()
|
||
self.headers = {
|
||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
|
||
"Referer": "https://www.bilibili.com/",
|
||
}
|
||
|
||
def extract_bilibili_url(self, text: str) -> str | None:
|
||
"""从文本中提取哔哩哔哩视频链接"""
|
||
# 哔哩哔哩短链接模式
|
||
short_pattern = re.compile(r"https?://b23\.tv/[\w]+", re.IGNORECASE)
|
||
# 哔哩哔哩完整链接模式
|
||
full_pattern = re.compile(r"https?://(?:www\.)?bilibili\.com/video/(?:BV[\w]+|av\d+)", re.IGNORECASE)
|
||
|
||
# 先匹配短链接
|
||
short_match = short_pattern.search(text)
|
||
if short_match:
|
||
return short_match.group(0)
|
||
|
||
# 再匹配完整链接
|
||
full_match = full_pattern.search(text)
|
||
if full_match:
|
||
return full_match.group(0)
|
||
|
||
return None
|
||
|
||
async def get_video_info(self, url: str) -> dict[str, Any] | None:
|
||
"""获取哔哩哔哩视频基本信息"""
|
||
try:
|
||
logger.info(f"🔍 解析视频URL: {url}")
|
||
|
||
# 如果是短链接，先解析为完整链接
|
||
if "b23.tv" in url:
|
||
logger.info("🔗 检测到短链接，正在解析...")
|
||
timeout = aiohttp.ClientTimeout(total=30)
|
||
async with aiohttp.ClientSession(timeout=timeout) as session:
|
||
async with session.get(url, headers=self.headers, allow_redirects=True) as response:
|
||
url = str(response.url)
|
||
logger.info(f"✅ 短链接解析完成: {url}")
|
||
|
||
# 提取BV号或AV号
|
||
bv_match = re.search(r"BV([\w]+)", url)
|
||
av_match = re.search(r"av(\d+)", url)
|
||
|
||
if bv_match:
|
||
bvid = f"BV{bv_match.group(1)}"
|
||
api_url = f"https://api.bilibili.com/x/web-interface/view?bvid={bvid}"
|
||
logger.info(f"📺 提取到BV号: {bvid}")
|
||
elif av_match:
|
||
aid = av_match.group(1)
|
||
api_url = f"https://api.bilibili.com/x/web-interface/view?aid={aid}"
|
||
logger.info(f"📺 提取到AV号: av{aid}")
|
||
else:
|
||
logger.error("❌ 无法从URL中提取视频ID")
|
||
return None
|
||
|
||
# 获取视频信息
|
||
logger.info("📡 正在获取视频信息...")
|
||
timeout = aiohttp.ClientTimeout(total=30)
|
||
async with aiohttp.ClientSession(timeout=timeout) as session:
|
||
async with session.get(api_url, headers=self.headers) as response:
|
||
if response.status != 200:
|
||
logger.error(f"❌ API请求失败，状态码: {response.status}")
|
||
return None
|
||
data = await response.json()
|
||
|
||
if data.get("code") != 0:
|
||
error_msg = data.get("message", "未知错误")
|
||
logger.error(f"❌ B站API返回错误: {error_msg} (code: {data.get('code')})")
|
||
return None
|
||
|
||
video_data = data["data"]
|
||
|
||
# 验证必要字段
|
||
if not video_data.get("title"):
|
||
logger.error("❌ 视频数据不完整，缺少标题")
|
||
return None
|
||
|
||
result = {
|
||
"title": video_data.get("title", ""),
|
||
"desc": video_data.get("desc", ""),
|
||
"duration": video_data.get("duration", 0),
|
||
"view": video_data.get("stat", {}).get("view", 0),
|
||
"like": video_data.get("stat", {}).get("like", 0),
|
||
"coin": video_data.get("stat", {}).get("coin", 0),
|
||
"favorite": video_data.get("stat", {}).get("favorite", 0),
|
||
"share": video_data.get("stat", {}).get("share", 0),
|
||
"owner": video_data.get("owner", {}).get("name", ""),
|
||
"pubdate": video_data.get("pubdate", 0),
|
||
"aid": video_data.get("aid"),
|
||
"bvid": video_data.get("bvid"),
|
||
"cid": video_data.get("cid")
|
||
or (video_data.get("pages", [{}])[0].get("cid") if video_data.get("pages") else None),
|
||
}
|
||
|
||
logger.info(f"✅ 视频信息获取成功: {result['title']}")
|
||
return result
|
||
|
||
except asyncio.TimeoutError:
|
||
logger.error("❌ 获取视频信息超时")
|
||
return None
|
||
except aiohttp.ClientError as e:
|
||
logger.error(f"❌ 网络请求失败: {e}")
|
||
return None
|
||
except Exception as e:
|
||
logger.error(f"❌ 获取哔哩哔哩视频信息时发生未知错误: {e}")
|
||
logger.exception("详细错误信息:")
|
||
return None
|
||
|
||
async def get_video_stream_url(self, aid: int, cid: int) -> str | None:
|
||
"""获取视频流URL"""
|
||
try:
|
||
logger.info(f"🎥 获取视频流URL: aid={aid}, cid={cid}")
|
||
|
||
# 构建播放信息API请求
|
||
api_url = f"https://api.bilibili.com/x/player/playurl?avid={aid}&cid={cid}&qn=80&type=&otype=json&fourk=1&fnver=0&fnval=4048&session="
|
||
|
||
timeout = aiohttp.ClientTimeout(total=30)
|
||
async with aiohttp.ClientSession(timeout=timeout) as session:
|
||
async with session.get(api_url, headers=self.headers) as response:
|
||
if response.status != 200:
|
||
logger.error(f"❌ 播放信息API请求失败，状态码: {response.status}")
|
||
return None
|
||
data = await response.json()
|
||
|
||
if data.get("code") != 0:
|
||
error_msg = data.get("message", "未知错误")
|
||
logger.error(f"❌ 获取播放信息失败: {error_msg} (code: {data.get('code')})")
|
||
return None
|
||
|
||
play_data = data["data"]
|
||
|
||
# 尝试获取DASH格式的视频流
|
||
if "dash" in play_data and play_data["dash"].get("video"):
|
||
videos = play_data["dash"]["video"]
|
||
logger.info(f"🎬 找到 {len(videos)} 个DASH视频流")
|
||
|
||
# 选择最高质量的视频流
|
||
video_stream = max(videos, key=lambda x: x.get("bandwidth", 0))
|
||
stream_url = video_stream.get("baseUrl") or video_stream.get("base_url")
|
||
|
||
if stream_url:
|
||
logger.info(f"✅ 获取到DASH视频流URL (带宽: {video_stream.get('bandwidth', 0)})")
|
||
return stream_url
|
||
|
||
# 降级到FLV格式
|
||
if play_data.get("durl"):
|
||
logger.info("📹 使用FLV格式视频流")
|
||
stream_url = play_data["durl"][0].get("url")
|
||
if stream_url:
|
||
logger.info("✅ 获取到FLV视频流URL")
|
||
return stream_url
|
||
|
||
logger.error("❌ 未找到可用的视频流")
|
||
return None
|
||
|
||
except asyncio.TimeoutError:
|
||
logger.error("❌ 获取视频流URL超时")
|
||
return None
|
||
except aiohttp.ClientError as e:
|
||
logger.error(f"❌ 网络请求失败: {e}")
|
||
return None
|
||
except Exception as e:
|
||
logger.error(f"❌ 获取视频流URL时发生未知错误: {e}")
|
||
logger.exception("详细错误信息:")
|
||
return None
|
||
|
||
async def download_video_bytes(self, stream_url: str, max_size_mb: int = 100) -> bytes | None:
|
||
"""下载视频字节数据
|
||
|
||
Args:
|
||
stream_url: 视频流URL
|
||
max_size_mb: 最大下载大小限制（MB），默认100MB
|
||
|
||
Returns:
|
||
视频字节数据或None
|
||
"""
|
||
try:
|
||
logger.info(f"📥 开始下载视频: {stream_url[:50]}...")
|
||
|
||
# 设置超时和大小限制
|
||
timeout = aiohttp.ClientTimeout(total=300, connect=30) # 5分钟总超时，30秒连接超时
|
||
|
||
async with aiohttp.ClientSession(timeout=timeout) as session:
|
||
async with session.get(stream_url, headers=self.headers) as response:
|
||
if response.status != 200:
|
||
logger.error(f"❌ 下载失败，HTTP状态码: {response.status}")
|
||
return None
|
||
|
||
# 检查内容长度
|
||
content_length = response.headers.get("content-length")
|
||
if content_length:
|
||
size_mb = int(content_length) / 1024 / 1024
|
||
if size_mb > max_size_mb:
|
||
logger.error(f"❌ 视频文件过大: {size_mb:.1f}MB > {max_size_mb}MB")
|
||
return None
|
||
logger.info(f"📊 预计下载大小: {size_mb:.1f}MB")
|
||
|
||
# 分块下载并监控大小
|
||
video_bytes = bytearray()
|
||
downloaded_mb = 0
|
||
|
||
async for chunk in response.content.iter_chunked(8192): # 8KB块
|
||
video_bytes.extend(chunk)
|
||
downloaded_mb = len(video_bytes) / 1024 / 1024
|
||
|
||
# 检查大小限制
|
||
if downloaded_mb > max_size_mb:
|
||
logger.error(f"❌ 下载中止，文件过大: {downloaded_mb:.1f}MB > {max_size_mb}MB")
|
||
return None
|
||
|
||
final_size_mb = len(video_bytes) / 1024 / 1024
|
||
logger.info(f"✅ 视频下载完成，实际大小: {final_size_mb:.2f}MB")
|
||
return bytes(video_bytes)
|
||
|
||
except asyncio.TimeoutError:
|
||
logger.error("❌ 下载超时")
|
||
return None
|
||
except aiohttp.ClientError as e:
|
||
logger.error(f"❌ 网络请求失败: {e}")
|
||
return None
|
||
except Exception as e:
|
||
logger.error(f"❌ 下载视频时发生未知错误: {e}")
|
||
logger.exception("详细错误信息:")
|
||
return None
|
||
|
||
async def analyze_bilibili_video(self, url: str, prompt: str = None) -> dict[str, Any]:
|
||
"""分析哔哩哔哩视频并返回详细信息和AI分析结果"""
|
||
try:
|
||
logger.info(f"🎬 开始分析哔哩哔哩视频: {url}")
|
||
|
||
# 1. 获取视频基本信息
|
||
video_info = await self.get_video_info(url)
|
||
if not video_info:
|
||
logger.error("❌ 无法获取视频基本信息")
|
||
return {"error": "无法获取视频信息"}
|
||
|
||
logger.info(f"📺 视频标题: {video_info['title']}")
|
||
logger.info(f"👤 UP主: {video_info['owner']}")
|
||
logger.info(f"⏱️ 时长: {video_info['duration']}秒")
|
||
|
||
# 2. 获取视频流URL
|
||
stream_url = await self.get_video_stream_url(video_info["aid"], video_info["cid"])
|
||
if not stream_url:
|
||
logger.warning("⚠️ 无法获取视频流，仅返回基本信息")
|
||
return {"video_info": video_info, "error": "无法获取视频流，仅返回基本信息"}
|
||
|
||
# 3. 下载视频
|
||
video_bytes = await self.download_video_bytes(stream_url)
|
||
if not video_bytes:
|
||
logger.warning("⚠️ 视频下载失败，仅返回基本信息")
|
||
return {"video_info": video_info, "error": "视频下载失败，仅返回基本信息"}
|
||
|
||
# 4. 构建增强的元数据信息
|
||
enhanced_metadata = {
|
||
"title": video_info["title"],
|
||
"uploader": video_info["owner"],
|
||
"duration": video_info["duration"],
|
||
"view_count": video_info["view"],
|
||
"like_count": video_info["like"],
|
||
"description": video_info["desc"],
|
||
"bvid": video_info["bvid"],
|
||
"aid": video_info["aid"],
|
||
"file_size": len(video_bytes),
|
||
"source": "bilibili",
|
||
}
|
||
|
||
# 5. 使用新的视频分析API，传递完整的元数据
|
||
logger.info("🤖 开始AI视频分析...")
|
||
analysis_result = await self.video_analyzer.analyze_video_from_bytes(
|
||
video_bytes=video_bytes,
|
||
filename=f"{video_info['title']}.mp4",
|
||
prompt=prompt, # 使用新API的prompt参数而不是user_question
|
||
)
|
||
|
||
# 6. 检查分析结果
|
||
if not analysis_result or not analysis_result.get("summary"):
|
||
logger.error("❌ 视频分析失败或返回空结果")
|
||
return {"video_info": video_info, "error": "视频分析失败，仅返回基本信息"}
|
||
|
||
# 7. 格式化返回结果
|
||
duration_str = f"{video_info['duration'] // 60}分{video_info['duration'] % 60}秒"
|
||
|
||
result = {
|
||
"video_info": {
|
||
"标题": video_info["title"],
|
||
"UP主": video_info["owner"],
|
||
"时长": duration_str,
|
||
"播放量": f"{video_info['view']:,}",
|
||
"点赞": f"{video_info['like']:,}",
|
||
"投币": f"{video_info['coin']:,}",
|
||
"收藏": f"{video_info['favorite']:,}",
|
||
"转发": f"{video_info['share']:,}",
|
||
"简介": video_info["desc"][:200] + "..." if len(video_info["desc"]) > 200 else video_info["desc"],
|
||
},
|
||
"ai_analysis": analysis_result.get("summary", ""),
|
||
"success": True,
|
||
"metadata": enhanced_metadata, # 添加元数据信息
|
||
}
|
||
|
||
logger.info("✅ 哔哩哔哩视频分析完成")
|
||
return result
|
||
|
||
except Exception as e:
|
||
error_msg = f"分析哔哩哔哩视频时发生异常: {e!s}"
|
||
logger.error(f"❌ {error_msg}")
|
||
logger.exception("详细错误信息:") # 记录完整的异常堆栈
|
||
return {"error": f"分析失败: {e!s}"}
|
||
|
||
|
||
# 全局实例
|
||
_bilibili_analyzer = None
|
||
|
||
|
||
def get_bilibili_analyzer() -> BilibiliVideoAnalyzer:
|
||
"""获取哔哩哔哩视频分析器实例（单例模式）"""
|
||
global _bilibili_analyzer
|
||
if _bilibili_analyzer is None:
|
||
_bilibili_analyzer = BilibiliVideoAnalyzer()
|
||
return _bilibili_analyzer
|