141 lines
4.2 KiB
Python
141 lines
4.2 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Extract VideoId from various video platform URLs.
|
|
|
|
Supports: 小红书, 抖音, TikTok, B站, YouTube, 快手等
|
|
"""
|
|
|
|
import re
|
|
from urllib.parse import urlparse, parse_qs
|
|
|
|
|
|
def extract_video_id(url: str) -> str:
|
|
"""
|
|
Extract VideoId from video platform URL.
|
|
|
|
Args:
|
|
url: Video URL from supported platforms
|
|
|
|
Returns:
|
|
VideoId string
|
|
|
|
Raises:
|
|
ValueError: If URL format is not recognized
|
|
"""
|
|
url = url.strip()
|
|
|
|
# 小红书短链接: http://xhslink.com/o/6VbNVltFQRX
|
|
if 'xhslink.com' in url or 'xiaohongshu.com' in url:
|
|
match = re.search(r'/o/([A-Za-z0-9]+)', url)
|
|
if match:
|
|
return match.group(1)
|
|
|
|
# 抖音: https://v.douyin.com/xxx/ or https://www.douyin.com/video/xxx
|
|
if 'douyin.com' in url:
|
|
# 短链接
|
|
match = re.search(r'v\.douyin\.com/([A-Za-z0-9]+)', url)
|
|
if match:
|
|
return match.group(1)
|
|
# 长链接
|
|
match = re.search(r'/video/(\d+)', url)
|
|
if match:
|
|
return match.group(1)
|
|
|
|
# TikTok: https://www.tiktok.com/@user/video/1234567890
|
|
if 'tiktok.com' in url:
|
|
match = re.search(r'/video/(\d+)', url)
|
|
if match:
|
|
return match.group(1)
|
|
# 短链接 https://vm.tiktok.com/xxx/
|
|
match = re.search(r'vm\.tiktok\.com/([A-Za-z0-9]+)', url)
|
|
if match:
|
|
return match.group(1)
|
|
|
|
# B站: https://www.bilibili.com/video/BVxxx or https://b23.tv/xxx
|
|
if 'bilibili.com' in url or 'b23.tv' in url:
|
|
# BV号
|
|
match = re.search(r'/(BV[A-Za-z0-9]+)', url)
|
|
if match:
|
|
return match.group(1)
|
|
# av号
|
|
match = re.search(r'/av(\d+)', url)
|
|
if match:
|
|
return f"av{match.group(1)}"
|
|
# 短链接
|
|
match = re.search(r'b23\.tv/([A-Za-z0-9]+)', url)
|
|
if match:
|
|
return match.group(1)
|
|
|
|
# YouTube: https://www.youtube.com/watch?v=xxx or https://youtu.be/xxx
|
|
if 'youtube.com' in url or 'youtu.be' in url:
|
|
# youtu.be 短链接
|
|
match = re.search(r'youtu\.be/([A-Za-z0-9_-]+)', url)
|
|
if match:
|
|
return match.group(1)
|
|
# 标准链接
|
|
parsed = urlparse(url)
|
|
if parsed.query:
|
|
params = parse_qs(parsed.query)
|
|
if 'v' in params:
|
|
return params['v'][0]
|
|
|
|
# 快手: https://www.kuaishou.com/short-video/xxx
|
|
if 'kuaishou.com' in url:
|
|
match = re.search(r'/short-video/(\d+)', url)
|
|
if match:
|
|
return match.group(1)
|
|
match = re.search(r'\.com/([A-Za-z0-9]+)', url)
|
|
if match:
|
|
return match.group(1)
|
|
|
|
# 如果都不匹配，尝试提取 URL 最后的路径部分
|
|
parsed = urlparse(url)
|
|
path_parts = [p for p in parsed.path.split('/') if p]
|
|
if path_parts:
|
|
# 返回最后一个非空路径部分
|
|
return path_parts[-1]
|
|
|
|
raise ValueError(f"无法从 URL 提取 VideoId: {url}")
|
|
|
|
|
|
def extract_urls(text: str) -> list[str]:
|
|
"""
|
|
Extract video URLs from text.
|
|
|
|
Args:
|
|
text: Text containing video URLs
|
|
|
|
Returns:
|
|
List of unique video URLs
|
|
"""
|
|
pattern = r'https?://[^\s]+(?:douyin|tiktok|youtube|youtu\.be|xiaohongshu|xhslink|bilibili|b23\.tv|kuaishou)[^\s]*'
|
|
urls = re.findall(pattern, text)
|
|
return list(dict.fromkeys(urls)) # 保持顺序去重
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# 测试用例
|
|
test_urls = [
|
|
'http://xhslink.com/o/6VbNVltFQRX',
|
|
'http://xhslink.com/o/16X9vM9CTBo',
|
|
'https://v.douyin.com/eFyQRjc/',
|
|
'https://www.douyin.com/video/7123456789',
|
|
'https://www.tiktok.com/@user/video/7123456789',
|
|
'https://vm.tiktok.com/ZMeAbCdEf/',
|
|
'https://www.bilibili.com/video/BV1xx411c7mD',
|
|
'https://b23.tv/av12345678',
|
|
'https://www.youtube.com/watch?v=dQw4w9WgXcQ',
|
|
'https://youtu.be/dQw4w9WgXcQ',
|
|
'https://www.kuaishou.com/short-video/1234567890',
|
|
]
|
|
|
|
print("VideoId 提取测试:\n")
|
|
for url in test_urls:
|
|
try:
|
|
video_id = extract_video_id(url)
|
|
print(f"✅ {url}")
|
|
print(f" → {video_id}\n")
|
|
except ValueError as e:
|
|
print(f"❌ {url}")
|
|
print(f" → {e}\n")
|