124 lines
4.9 KiB
Python
124 lines
4.9 KiB
Python
"""腾讯财经数据源适配器。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from datetime import datetime
|
||
from typing import List, Optional
|
||
from urllib.parse import urlencode
|
||
from urllib.request import Request, urlopen
|
||
|
||
from sidecar.models import KlineBar, Snapshot, parse_float
|
||
from sidecar.symbols import normalize_symbol
|
||
|
||
|
||
class TencentFinanceSource:
|
||
"""腾讯财经行情适配器。"""
|
||
|
||
def __init__(self, timeout_seconds: int = 8) -> None:
|
||
"""
|
||
功能说明：创建腾讯财经数据源。
|
||
参数说明：timeout_seconds 为 HTTP 请求超时时间。
|
||
返回值说明：无返回值。
|
||
注意事项：腾讯接口无需 Key，但字段为 GBK 编码和波浪线分隔。
|
||
"""
|
||
self.timeout_seconds = timeout_seconds
|
||
|
||
def fetch_kline(self, symbol: str, period: str, limit: int) -> List[KlineBar]:
|
||
"""
|
||
功能说明：获取 K 线数据。
|
||
参数说明：symbol 为股票代码，period 为周期，limit 为最大条数。
|
||
返回值说明：腾讯适配器暂不提供 K 线，固定抛出 NotImplementedError。
|
||
注意事项：本项目 K 线优先由 mootdx 提供。
|
||
"""
|
||
raise NotImplementedError("腾讯财经适配器暂不提供 K 线")
|
||
|
||
def fetch_snapshot(self, symbol: str) -> Snapshot:
|
||
"""
|
||
功能说明：获取腾讯财经实时行情快照。
|
||
参数说明：symbol 为股票代码。
|
||
返回值说明：返回统一 Snapshot 模型。
|
||
注意事项：PB 使用索引 46，PE_TTM 使用索引 39，避免常见错误映射。
|
||
"""
|
||
normalized = normalize_symbol(symbol)
|
||
query = urlencode({"q": normalized})
|
||
request = Request(
|
||
"https://qt.gtimg.cn/?" + query,
|
||
headers={"User-Agent": "Mozilla/5.0 TortoiseSidecar/0.1"},
|
||
)
|
||
with urlopen(request, timeout=self.timeout_seconds) as response:
|
||
text = response.read().decode("gbk", errors="ignore")
|
||
fields = parse_tencent_payload(text)
|
||
return snapshot_from_tencent_fields(normalized, fields)
|
||
|
||
|
||
def parse_tencent_payload(payload: str) -> List[str]:
|
||
"""
|
||
功能说明：解析腾讯财经原始响应。
|
||
参数说明：payload 为形如 v_sh600000="..." 的响应文本。
|
||
返回值说明：返回按波浪线拆分后的字段列表。
|
||
注意事项：响应为空或格式异常时抛出 ValueError。
|
||
"""
|
||
if '="' not in payload:
|
||
raise ValueError("腾讯财经响应格式异常")
|
||
content = payload.split('="', 1)[1].split('"', 1)[0]
|
||
fields = content.split("~")
|
||
if len(fields) < 53:
|
||
raise ValueError("腾讯财经响应字段不足")
|
||
return fields
|
||
|
||
|
||
def snapshot_from_tencent_fields(symbol: str, fields: List[str]) -> Snapshot:
|
||
"""
|
||
功能说明：把腾讯财经字段转换为统一快照。
|
||
参数说明：symbol 为标准股票代码，fields 为腾讯响应字段列表。
|
||
返回值说明：返回 Snapshot 模型。
|
||
注意事项：字段索引按项目文档校准，43 是振幅不是 PB。
|
||
"""
|
||
return Snapshot(
|
||
symbol=symbol,
|
||
name=_field(fields, 1) or None,
|
||
trade_time=_parse_trade_time(_field(fields, 30)),
|
||
price=parse_float(_field(fields, 3)),
|
||
previous_close=parse_float(_field(fields, 4)),
|
||
open=parse_float(_field(fields, 5)),
|
||
high=parse_float(_field(fields, 33)),
|
||
low=parse_float(_field(fields, 34)),
|
||
volume=parse_float(_field(fields, 6)),
|
||
amount=parse_float(_field(fields, 37)),
|
||
change=parse_float(_field(fields, 31)),
|
||
change_percent=parse_float(_field(fields, 32)),
|
||
turnover_rate=parse_float(_field(fields, 38)),
|
||
pe_ttm=parse_float(_field(fields, 39)),
|
||
pe_static=parse_float(_field(fields, 52)),
|
||
pb=parse_float(_field(fields, 46)),
|
||
market_cap=parse_float(_field(fields, 44)),
|
||
float_market_cap=parse_float(_field(fields, 45)),
|
||
limit_up=parse_float(_field(fields, 47)),
|
||
limit_down=parse_float(_field(fields, 48)),
|
||
source="tencent",
|
||
)
|
||
|
||
|
||
def _field(fields: List[str], index: int) -> str:
|
||
"""
|
||
功能说明：安全读取腾讯字段。
|
||
参数说明：fields 为字段列表，index 为字段索引。
|
||
返回值说明：字段存在返回原始字符串，不存在返回空字符串。
|
||
注意事项：腾讯字段偶发缺失时可避免 IndexError。
|
||
"""
|
||
return fields[index] if index < len(fields) else ""
|
||
|
||
|
||
def _parse_trade_time(value: str) -> Optional[datetime]:
|
||
"""
|
||
功能说明：解析腾讯交易时间字段。
|
||
参数说明：value 为 YYYYMMDDHHMMSS 格式字符串。
|
||
返回值说明：解析成功返回 datetime，空值或异常返回 None。
|
||
注意事项：返回值不带时区，表示交易所本地时间。
|
||
"""
|
||
if not value:
|
||
return None
|
||
try:
|
||
return datetime.strptime(value[:14], "%Y%m%d%H%M%S")
|
||
except ValueError:
|
||
return None |