hahaju/aiagents-stock
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

tushare

Browse Source
This commit is contained in:
songzhuoyuan
2026-08-11 20:41:31 +08:00
parent 2ce13e5bae
commit befdc32aea
24 changed files with 1311 additions and 530 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fund_flow_akshare.py
+47 -50
View File
@@ -109,56 +109,54 @@ class FundFlowAkshareDataFetcher:
def _get_individual_fund_flow(self, symbol, market):
"""获取个股资金流向数据（支持akshare和tushare自动切换）"""
try:
# 优先使用akshare的stock_individual_fund_flow接口
print(f" [Akshare] 正在获取资金流向 (市场: {market})...")
df = ak.stock_individual_fund_flow(stock=symbol, market=market)
# 优先使用tushare
if data_source_manager.tushare_available:
try:
print(f" [Tushare] 正在获取资金流向数据（主要数据源）...")
ts_code = data_source_manager._convert_to_ts_code(symbol)
# 计算日期范围（最近N个交易日）
end_date = datetime.now().strftime('%Y%m%d')
start_date = (datetime.now() - timedelta(days=self.days * 2)).strftime('%Y%m%d')
df = data_source_manager.tushare_api.moneyflow(
ts_code=ts_code,
start_date=start_date,
end_date=end_date
)
if df is not None and not df.empty:
# 标准化列名以匹配akshare格式（金额单位：千元→元）
df = df.rename(columns={
'trade_date': '日期',
'close': '收盘价',
'pct_chg': '涨跌幅',
'net_mf_amount': '主力净流入-净额'
})
df['主力净流入-净额'] = df['主力净流入-净额'] * 1000
df['超大单净流入-净额'] = (df['buy_elg_amount'] - df['sell_elg_amount']) * 1000
df['大单净流入-净额'] = (df['buy_lg_amount'] - df['sell_lg_amount']) * 1000
df['中单净流入-净额'] = (df['buy_md_amount'] - df['sell_md_amount']) * 1000
df['小单净流入-净额'] = (df['buy_sm_amount'] - df['sell_sm_amount']) * 1000
# tushare按日期倒序返回，取最近N天后转为正序（与akshare一致）
df = df.head(self.days)
df = df.iloc[::-1].reset_index(drop=True)
print(f" [Tushare] ✅ 成功获取 {len(df)} 条资金流向数据")
else:
print(f" [Tushare] ❌ 未找到资金流向数据，尝试备用数据源")
df = None
except Exception as te:
print(f" [Tushare] ❌ 获取失败: {te}")
df = None
else:
df = None
# tushare不可用或失败时，回退akshare
if df is None or df.empty:
print(f" [Akshare] 未找到资金流向数据，尝试备用数据源...")
# akshare失败，尝试tushare
if data_source_manager.tushare_available:
try:
print(f" [Tushare] 正在获取资金流向数据（备用数据源）...")
ts_code = data_source_manager._convert_to_ts_code(symbol)
# 计算日期范围（最近N个交易日）
end_date = datetime.now().strftime('%Y%m%d')
start_date = (datetime.now() - timedelta(days=self.days * 2)).strftime('%Y%m%d')
# 获取资金流向数据
df = data_source_manager.tushare_api.moneyflow(
ts_code=ts_code,
start_date=start_date,
end_date=end_date
)
if df is not None and not df.empty:
# 标准化列名以匹配akshare格式
df = df.rename(columns={
'trade_date': '日期',
'buy_sm_amount': '小单买入',
'sell_sm_amount': '小单卖出',
'buy_md_amount': '中单买入',
'sell_md_amount': '中单卖出',
'buy_lg_amount': '大单买入',
'sell_lg_amount': '大单卖出',
'buy_elg_amount': '超大单买入',
'sell_elg_amount': '超大单卖出',
'net_mf_amount': '净额'
})
# 限制为最近N天
df = df.head(self.days)
print(f" [Tushare] ✅ 成功获取 {len(df)} 条资金流向数据")
else:
print(f" [Tushare] ❌ 未找到资金流向数据")
return None
except Exception as te:
print(f" [Tushare] ❌ 获取失败: {te}")
return None
else:
print(f" [Akshare] 正在获取资金流向 (市场: {market})（备用数据源）...")
df = ak.stock_individual_fund_flow(stock=symbol, market=market)
if df is None or df.empty:
print(f" [Akshare] 未找到资金流向数据")
return None
# akshare 返回的数据是按时间正序排列（从旧到新），所以使用 tail() 获取最近N天的数据
@@ -341,4 +339,3 @@ if __name__ == "__main__":
print(f"\n获取失败: {data.get('error', '未知错误')}")
print("\n")