mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-05-06 20:42:52 +08:00
04dad3b72a
- 新增 abogus.py 文件，实现 a_bogus 参数的生成逻辑 - 代码源自 JoeanAmier/TikTokDownloader 项目，并进行了适配和优化 - 功能包括生成用户代理字符串、加密 URL 参数和生成最终的 a_bogus值 - 提供了详细的注释和函数说明，便于理解和维护
26 lines
741 B
Python
26 lines
741 B
Python
from pydantic import AnyUrl, validator, BaseModel
|
|
import re
|
|
|
|
SUPPORTED_PLATFORMS = {
|
|
"bilibili": r"(https?://)?(www\.)?bilibili\.com/video/[a-zA-Z0-9]+",
|
|
"youtube": r"(https?://)?(www\.)?(youtube\.com/watch\?v=|youtu\.be/)[\w\-]+",
|
|
"douyin": r"'http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F])"
|
|
|
|
}
|
|
|
|
|
|
|
|
def is_supported_video_url(url: str) -> bool:
|
|
return any(re.match(pattern, url) for pattern in SUPPORTED_PLATFORMS.values())
|
|
|
|
|
|
class VideoRequest(BaseModel):
|
|
url: AnyUrl
|
|
platform: str
|
|
|
|
@validator("url")
|
|
def validate_video_url(cls, v):
|
|
if not is_supported_video_url(str(v)):
|
|
raise ValueError("暂不支持该视频平台或链接格式无效")
|
|
return v
|