修复代码格式和文件名大小写问题
This commit is contained in:
Windpicker-owo
@@ -17,51 +17,51 @@ 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/',
|
||||
"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) -> Optional[str]:
|
||||
"""从文本中提取哔哩哔哩视频链接"""
|
||||
# 哔哩哔哩短链接模式
|
||||
short_pattern = re.compile(r'https?://b23\.tv/[\w]+', re.IGNORECASE)
|
||||
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)
|
||||
|
||||
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) -> Optional[Dict[str, Any]]:
|
||||
"""获取哔哩哔哩视频基本信息"""
|
||||
try:
|
||||
logger.info(f"🔍 解析视频URL: {url}")
|
||||
|
||||
|
||||
# 如果是短链接，先解析为完整链接
|
||||
if 'b23.tv' in 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)
|
||||
|
||||
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}"
|
||||
@@ -73,7 +73,7 @@ class BilibiliVideoAnalyzer:
|
||||
else:
|
||||
logger.error("❌ 无法从URL中提取视频ID")
|
||||
return None
|
||||
|
||||
|
||||
# 获取视频信息
|
||||
logger.info("📡 正在获取视频信息...")
|
||||
timeout = aiohttp.ClientTimeout(total=30)
|
||||
@@ -83,38 +83,39 @@ class BilibiliVideoAnalyzer:
|
||||
logger.error(f"❌ API请求失败，状态码: {response.status}")
|
||||
return None
|
||||
data = await response.json()
|
||||
|
||||
if data.get('code') != 0:
|
||||
error_msg = data.get('message', '未知错误')
|
||||
|
||||
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']
|
||||
|
||||
|
||||
video_data = data["data"]
|
||||
|
||||
# 验证必要字段
|
||||
if not video_data.get('title'):
|
||||
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)
|
||||
"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
|
||||
@@ -125,15 +126,15 @@ class BilibiliVideoAnalyzer:
|
||||
logger.error(f"❌ 获取哔哩哔哩视频信息时发生未知错误: {e}")
|
||||
logger.exception("详细错误信息:")
|
||||
return None
|
||||
|
||||
|
||||
async def get_video_stream_url(self, aid: int, cid: int) -> Optional[str]:
|
||||
"""获取视频流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:
|
||||
@@ -141,38 +142,38 @@ class BilibiliVideoAnalyzer:
|
||||
logger.error(f"❌ 播放信息API请求失败，状态码: {response.status}")
|
||||
return None
|
||||
data = await response.json()
|
||||
|
||||
if data.get('code') != 0:
|
||||
error_msg = data.get('message', '未知错误')
|
||||
|
||||
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']
|
||||
|
||||
|
||||
play_data = data["data"]
|
||||
|
||||
# 尝试获取DASH格式的视频流
|
||||
if 'dash' in play_data and play_data['dash'].get('video'):
|
||||
videos = play_data['dash']['video']
|
||||
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')
|
||||
|
||||
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 'durl' in play_data and play_data['durl']:
|
||||
if "durl" in play_data and play_data["durl"]:
|
||||
logger.info("📹 使用FLV格式视频流")
|
||||
stream_url = play_data['durl'][0].get('url')
|
||||
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
|
||||
@@ -183,55 +184,55 @@ class BilibiliVideoAnalyzer:
|
||||
logger.error(f"❌ 获取视频流URL时发生未知错误: {e}")
|
||||
logger.exception("详细错误信息:")
|
||||
return None
|
||||
|
||||
|
||||
async def download_video_bytes(self, stream_url: str, max_size_mb: int = 100) -> Optional[bytes]:
|
||||
"""下载视频字节数据
|
||||
|
||||
|
||||
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')
|
||||
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
|
||||
@@ -242,93 +243,84 @@ class BilibiliVideoAnalyzer:
|
||||
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'])
|
||||
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": "无法获取视频流，仅返回基本信息"
|
||||
}
|
||||
|
||||
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": "视频下载失败，仅返回基本信息"
|
||||
}
|
||||
|
||||
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'],
|
||||
"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"
|
||||
"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
|
||||
prompt=prompt, # 使用新API的prompt参数而不是user_question
|
||||
)
|
||||
|
||||
|
||||
# 6. 检查分析结果
|
||||
if not analysis_result or not analysis_result.get('summary'):
|
||||
if not analysis_result or not analysis_result.get("summary"):
|
||||
logger.error("❌ 视频分析失败或返回空结果")
|
||||
return {
|
||||
"video_info": video_info,
|
||||
"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'],
|
||||
"标题": 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']
|
||||
"简介": video_info["desc"][:200] + "..." if len(video_info["desc"]) > 200 else video_info["desc"],
|
||||
},
|
||||
"ai_analysis": analysis_result.get('summary', ''),
|
||||
"ai_analysis": analysis_result.get("summary", ""),
|
||||
"success": True,
|
||||
"metadata": enhanced_metadata # 添加元数据信息
|
||||
"metadata": enhanced_metadata, # 添加元数据信息
|
||||
}
|
||||
|
||||
|
||||
logger.info("✅ 哔哩哔哩视频分析完成")
|
||||
return result
|
||||
|
||||
|
||||
except Exception as e:
|
||||
error_msg = f"分析哔哩哔哩视频时发生异常: {str(e)}"
|
||||
logger.error(f"❌ {error_msg}")
|
||||
@@ -339,9 +331,10 @@ class BilibiliVideoAnalyzer:
|
||||
# 全局实例
|
||||
_bilibili_analyzer = None
|
||||
|
||||
|
||||
def get_bilibili_analyzer() -> BilibiliVideoAnalyzer:
|
||||
"""获取哔哩哔哩视频分析器实例（单例模式）"""
|
||||
global _bilibili_analyzer
|
||||
if _bilibili_analyzer is None:
|
||||
_bilibili_analyzer = BilibiliVideoAnalyzer()
|
||||
return _bilibili_analyzer
|
||||
return _bilibili_analyzer
|
||||
|
||||
Reference in New Issue
Block a user