56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
from abc import ABC, abstractmethod
|
|
import requests
|
|
from typing import Dict, Optional
|
|
|
|
class BaseParser(ABC):
|
|
"""解析器基类"""
|
|
|
|
def __init__(self, api_url: str, api_key: Optional[str] = None, timeout: int = 30):
|
|
self.api_url = api_url
|
|
self.api_key = api_key
|
|
self.timeout = timeout
|
|
|
|
@abstractmethod
|
|
def parse(self, video_url: str) -> Dict:
|
|
"""
|
|
解析视频链接
|
|
返回统一格式:
|
|
{
|
|
"cover": "封面URL",
|
|
"video_url": "视频URL",
|
|
"title": "标题",
|
|
"description": "简介"
|
|
}
|
|
"""
|
|
pass
|
|
|
|
def _make_request(self, url: str, params: Dict = None, headers: Dict = None, verify: bool = True) -> requests.Response:
|
|
"""发送HTTP请求"""
|
|
try:
|
|
# 设置默认请求头
|
|
default_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',
|
|
'Accept': 'application/json, text/plain, */*',
|
|
'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8',
|
|
}
|
|
|
|
# 合并自定义请求头
|
|
if headers:
|
|
default_headers.update(headers)
|
|
|
|
response = requests.get(url, params=params, headers=default_headers, timeout=self.timeout, verify=verify)
|
|
response.raise_for_status()
|
|
return response
|
|
except requests.RequestException as e:
|
|
raise Exception(f"请求失败: {str(e)}")
|
|
|
|
def _normalize_response(self, cover: str, video_url: str, title: str, description: str, author: str = "") -> Dict:
|
|
"""标准化返回数据"""
|
|
return {
|
|
"cover": cover or "",
|
|
"video_url": video_url or "",
|
|
"title": title or "",
|
|
"description": description or "",
|
|
"author": author or ""
|
|
}
|