a65adcc2e5
Sources: extracted from two upstream archives (skill-repo, skills-main),
merged with the following policy:
- 15 broken symlinks (pointing to /Users/jameslee/.cc-switch/skills or
../../.agents/skills on a foreign machine) discarded
- 3 real name collisions with identical content (ai-pair, ifind-http-api,
zhipu-websearch) kept as one copy
- Functional overlaps deduped keeping the strongest variant:
- docx family: kept docx (official, full toolchain) + docx-cn
(GB/T 9704 Chinese official-document constants),
dropped docx_writer (no scripts, name collided with docx)
- humanizer family: kept humanizer-zh (6 zh reference docs),
dropped humanizer (en, redundant for CN workflow)
- Skills that only ran in a foreign environment removed:
ablemind-ops, app-publish, hlb-design-system, openclaw-adj-skill,
claude-driver
- alphapai excluded from this public repo because its SKILL.md hard-coded
live credentials
Result: 25 skills, 572 files, ~7.5 MB.
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
93 lines
2.8 KiB
Python
93 lines
2.8 KiB
Python
"""
|
|
Tushare 数据连接器
|
|
"""
|
|
|
|
import pandas as pd
|
|
import tushare as ts
|
|
from datetime import datetime, timedelta
|
|
from ppt_station.connectors.base import BaseConnector, ConnectorFactory
|
|
from ppt_station.models.job import DataSource
|
|
from ppt_station.config import settings
|
|
|
|
|
|
class TushareConnector(BaseConnector):
|
|
"""Tushare 金融数据连接器"""
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self._pro = None
|
|
|
|
def _get_pro_api(self):
|
|
"""获取 Tushare Pro API 实例"""
|
|
if self._pro is None:
|
|
if not settings.tushare_token:
|
|
raise ValueError(
|
|
"Tushare token not configured. "
|
|
"Please set TUSHARE_TOKEN in environment or .env file"
|
|
)
|
|
self._pro = ts.pro_api(settings.tushare_token)
|
|
return self._pro
|
|
|
|
def load(self, spec: DataSource) -> pd.DataFrame:
|
|
"""从 Tushare 加载数据"""
|
|
pro = self._get_pro_api()
|
|
|
|
# 处理日期范围
|
|
start_date = spec.start_date
|
|
end_date = spec.end_date
|
|
|
|
if not end_date:
|
|
end_date = datetime.now().strftime("%Y%m%d")
|
|
|
|
if not start_date:
|
|
# 默认一年前
|
|
one_year_ago = datetime.now() - timedelta(days=365)
|
|
start_date = one_year_ago.strftime("%Y%m%d")
|
|
|
|
# 根据 api_name 调用不同的接口
|
|
api_name = spec.api_name or "index_daily"
|
|
|
|
if api_name == "index_daily":
|
|
# 获取指数日线数据
|
|
ts_code = spec.ts_code or spec.index_code
|
|
if not ts_code:
|
|
raise ValueError("Tushare index_daily requires 'ts_code' or 'index_code'")
|
|
|
|
df = pro.index_daily(
|
|
ts_code=ts_code,
|
|
start_date=start_date,
|
|
end_date=end_date,
|
|
fields=",".join(spec.fields) if spec.fields else None,
|
|
)
|
|
|
|
elif api_name == "pro_bar":
|
|
# 使用 pro_bar 获取数据（更通用）
|
|
ts_code = spec.ts_code or spec.index_code
|
|
if not ts_code:
|
|
raise ValueError("Tushare pro_bar requires 'ts_code' or 'index_code'")
|
|
|
|
df = ts.pro_bar(
|
|
ts_code=ts_code,
|
|
start_date=start_date,
|
|
end_date=end_date,
|
|
adj="qfq", # 前复权
|
|
)
|
|
|
|
else:
|
|
raise ValueError(f"Unsupported Tushare API: {api_name}")
|
|
|
|
# 排序数据（按日期升序）
|
|
if "trade_date" in df.columns:
|
|
df = df.sort_values("trade_date").reset_index(drop=True)
|
|
|
|
# 转换日期格式为 datetime
|
|
if "trade_date" in df.columns:
|
|
df["trade_date"] = pd.to_datetime(df["trade_date"], format="%Y%m%d")
|
|
|
|
return df
|
|
|
|
|
|
# 注册连接器
|
|
ConnectorFactory.register("tushare", TushareConnector)
|
|
|