From a261e92e6720ff66e3f8b3b8fa44c5e2be062acf Mon Sep 17 00:00:00 2001
From: aaron <>
Date: Wed, 14 May 2025 00:03:22 +0800
Subject: [PATCH] update

---
 cryptoai/api/adata_api.py | 4 ++--
 cryptoai/routes/adata.py  | 8 ++++++--
 docker-compose.yml        | 2 +-
 3 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/cryptoai/api/adata_api.py b/cryptoai/api/adata_api.py
index 9a5e20d..b3c817c 100644
--- a/cryptoai/api/adata_api.py
+++ b/cryptoai/api/adata_api.py
@@ -98,12 +98,12 @@ class AStockAPI:
             return pd.DataFrame()
         
     @staticmethod
-    def get_capital_flow(stock_code: str) -> pd.DataFrame:
+    def get_capital_flow(stock_code: str, start_date: Optional[str] = None, end_date: Optional[str] = None) -> pd.DataFrame:
         """
         获取股票资金流向数据
         """
         try:
-            return adata.stock.market.get_capital_flow(stock_code=stock_code)
+            return adata.stock.market.get_capital_flow(stock_code=stock_code, start_date=start_date, end_date=end_date)
         except Exception as e:
             print(f"获取资金流向数据失败: {str(e)}")
             return pd.DataFrame()
diff --git a/cryptoai/routes/adata.py b/cryptoai/routes/adata.py
index d61ff18..3e0415a 100644
--- a/cryptoai/routes/adata.py
+++ b/cryptoai/routes/adata.py
@@ -3,6 +3,7 @@ import logging
 from fastapi import APIRouter, Depends, HTTPException, status, Body, Query, Path
 from cryptoai.api.adata_api import AStockAPI
 from cryptoai.utils.db_manager import get_db_manager
+from datetime import datetime
 
 # 创建路由
 router = APIRouter()
@@ -50,12 +51,15 @@ async def get_stock_data(stock_code: str):
     result = {}
 
     try:
+        start_date = "2025-01-01"   
+        end_date = datetime.now().strftime("%Y-%m-%d")
+
         # 获取市场数据
-        market_data = api.get_market_data(stock_code)
+        market_data = api.get_market_data(stock_code, start_date, end_date)
         result["market_data"] = json.loads(market_data.to_json(orient="records"))   
 
         # 获取资金流向数据
-        flow_data = api.get_capital_flow(stock_code)
+        flow_data = api.get_capital_flow(stock_code, start_date, end_date)
         result["flow_data"] = json.loads(flow_data.to_json(orient="records"))
 
     except Exception as e:
diff --git a/docker-compose.yml b/docker-compose.yml
index 9f56af8..d0c8361 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.20
+    image: cryptoai-api:0.0.21
     restart: always
     ports:
       - "8000:8000"