diff --git a/cryptoai/api/adata_api.py b/cryptoai/api/adata_api.py index 39ff75e..53ea5e9 100644 --- a/cryptoai/api/adata_api.py +++ b/cryptoai/api/adata_api.py @@ -96,7 +96,18 @@ class AStockAPI: except Exception as e: print(f"获取分钟线数据失败: {str(e)}") return pd.DataFrame() - + + @staticmethod + def get_capital_flow(stock_code: str) -> pd.DataFrame: + """ + 获取股票资金流向数据 + """ + try: + return adata.stock.market.get_capital_flow(stock_code=stock_code) + except Exception as e: + print(f"获取资金流向数据失败: {str(e)}") + return pd.DataFrame() + @staticmethod def get_capital_flow_min(stock_code: str) -> pd.DataFrame: """ diff --git a/cryptoai/routes/adata.py b/cryptoai/routes/adata.py index 6ef2c4b..ead303f 100644 --- a/cryptoai/routes/adata.py +++ b/cryptoai/routes/adata.py @@ -35,7 +35,7 @@ async def get_stock_data(stock_code: str): # result["min_data"] = json.loads(min_data.to_json(orient="records")) # 获取资金流向数据 - flow_data = api.get_capital_flow_min(stock_code) + flow_data = api.get_capital_flow(stock_code) result["flow_data"] = json.loads(flow_data.to_json(orient="records")) return result diff --git a/docker-compose.yml b/docker-compose.yml index 312a8c2..beba72a 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -29,7 +29,7 @@ services: cryptoai-api: build: . container_name: cryptoai-api - image: cryptoai-api:0.0.16 + image: cryptoai-api:0.0.17 restart: always ports: - "8000:8000"