wehub-resource-sync
140 lines
5.0 KiB
Python
140 lines
5.0 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
测试链接导入项目缩略图功能
|
|
"""
|
|
|
|
import sys
|
|
import asyncio
|
|
from pathlib import Path
|
|
|
|
# 添加项目根目录到Python路径
|
|
project_root = Path(__file__).parent.parent.parent
|
|
if str(project_root) not in sys.path:
|
|
sys.path.insert(0, str(project_root))
|
|
|
|
from backend.utils.bilibili_downloader import BilibiliDownloader
|
|
import requests
|
|
import base64
|
|
|
|
async def test_bilibili_thumbnail_extraction():
|
|
"""测试B站缩略图提取功能"""
|
|
print("🧪 测试B站缩略图提取功能...")
|
|
|
|
# 使用一个公开的B站视频链接进行测试
|
|
test_url = "https://www.bilibili.com/video/BV1LSegzbEp9/"
|
|
|
|
try:
|
|
# 创建下载器
|
|
downloader = BilibiliDownloader()
|
|
|
|
# 获取视频信息
|
|
video_info = await downloader.get_video_info(test_url)
|
|
|
|
print(f"✅ 视频信息获取成功:")
|
|
print(f" 标题: {video_info.title}")
|
|
print(f" 上传者: {video_info.uploader}")
|
|
print(f" 缩略图URL: {video_info.thumbnail_url}")
|
|
|
|
# 测试缩略图下载
|
|
if video_info.thumbnail_url:
|
|
print("🖼️ 测试缩略图下载...")
|
|
response = requests.get(video_info.thumbnail_url, timeout=10)
|
|
if response.status_code == 200:
|
|
# 转换为base64
|
|
thumbnail_base64 = base64.b64encode(response.content).decode('utf-8')
|
|
thumbnail_data = f"data:image/jpeg;base64,{thumbnail_base64}"
|
|
|
|
print(f"✅ 缩略图下载成功，大小: {len(response.content)} bytes")
|
|
print(f" Base64长度: {len(thumbnail_base64)} 字符")
|
|
print(f" 数据URI前缀: {thumbnail_data[:50]}...")
|
|
|
|
return True
|
|
else:
|
|
print(f"❌ 缩略图下载失败: HTTP {response.status_code}")
|
|
return False
|
|
else:
|
|
print("⚠️ 没有缩略图URL")
|
|
return False
|
|
|
|
except Exception as e:
|
|
print(f"❌ 测试失败: {e}")
|
|
return False
|
|
|
|
async def test_youtube_thumbnail_extraction():
|
|
"""测试YouTube缩略图提取功能"""
|
|
print("\n🧪 测试YouTube缩略图提取功能...")
|
|
|
|
# 使用一个公开的YouTube视频链接进行测试
|
|
test_url = "https://www.youtube.com/watch?v=dQw4w9WgXcQ"
|
|
|
|
try:
|
|
import yt_dlp
|
|
|
|
ydl_opts = {
|
|
'quiet': True,
|
|
'no_warnings': True,
|
|
}
|
|
|
|
def extract_info_sync(url, ydl_opts):
|
|
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
|
return ydl.extract_info(url, download=False)
|
|
|
|
loop = asyncio.get_event_loop()
|
|
video_info = await loop.run_in_executor(None, extract_info_sync, test_url, ydl_opts)
|
|
|
|
print(f"✅ 视频信息获取成功:")
|
|
print(f" 标题: {video_info.get('title', 'Unknown')}")
|
|
print(f" 上传者: {video_info.get('uploader', 'Unknown')}")
|
|
print(f" 缩略图URL: {video_info.get('thumbnail', '')}")
|
|
|
|
# 测试缩略图下载
|
|
thumbnail_url = video_info.get('thumbnail', '')
|
|
if thumbnail_url:
|
|
print("🖼️ 测试缩略图下载...")
|
|
response = requests.get(thumbnail_url, timeout=10)
|
|
if response.status_code == 200:
|
|
# 转换为base64
|
|
thumbnail_base64 = base64.b64encode(response.content).decode('utf-8')
|
|
thumbnail_data = f"data:image/jpeg;base64,{thumbnail_base64}"
|
|
|
|
print(f"✅ 缩略图下载成功，大小: {len(response.content)} bytes")
|
|
print(f" Base64长度: {len(thumbnail_base64)} 字符")
|
|
print(f" 数据URI前缀: {thumbnail_data[:50]}...")
|
|
|
|
return True
|
|
else:
|
|
print(f"❌ 缩略图下载失败: HTTP {response.status_code}")
|
|
return False
|
|
else:
|
|
print("⚠️ 没有缩略图URL")
|
|
return False
|
|
|
|
except Exception as e:
|
|
print(f"❌ 测试失败: {e}")
|
|
return False
|
|
|
|
async def main():
|
|
"""主函数"""
|
|
print("🚀 开始测试链接导入项目缩略图功能...\n")
|
|
|
|
# 测试B站缩略图提取
|
|
bilibili_success = await test_bilibili_thumbnail_extraction()
|
|
|
|
# 测试YouTube缩略图提取
|
|
youtube_success = await test_youtube_thumbnail_extraction()
|
|
|
|
print(f"\n📊 测试结果:")
|
|
print(f" B站缩略图提取: {'✅ 成功' if bilibili_success else '❌ 失败'}")
|
|
print(f" YouTube缩略图提取: {'✅ 成功' if youtube_success else '❌ 失败'}")
|
|
|
|
if bilibili_success and youtube_success:
|
|
print("\n🎉 所有测试通过！链接导入项目缩略图功能正常")
|
|
return True
|
|
else:
|
|
print("\n⚠️ 部分测试失败，请检查相关功能")
|
|
return False
|
|
|
|
if __name__ == "__main__":
|
|
success = asyncio.run(main())
|
|
sys.exit(0 if success else 1)
|