From 27348fb2be23e474c5e96034539a1c52a45fee98 Mon Sep 17 00:00:00 2001
From: aaron <>
Date: Sat, 17 May 2025 21:11:32 +0800
Subject: [PATCH] =?UTF-8?q?add=20=E8=B4=A2=E6=8A=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cryptoai/api/adata_api.py | 11 +++++++++++
 cryptoai/routes/adata.py  |  6 ++++--
 docker-compose.yml        |  2 +-
 3 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/cryptoai/api/adata_api.py b/cryptoai/api/adata_api.py
index a41fecb..9726076 100644
--- a/cryptoai/api/adata_api.py
+++ b/cryptoai/api/adata_api.py
@@ -56,6 +56,17 @@ class AStockAPI:
         except Exception as e:
             print(f"获取板块失败: {str(e)}")
             return pd.DataFrame()
+        
+    @staticmethod
+    def get_stock_finance_core_index(stock_code: str) -> pd.DataFrame:
+        """
+        获取核心财务指标
+        """
+        try:
+            return adata.stock.finance.get_core_index(stock_code) 
+        except Exception as e:
+            print(f"获取核心财务指标失败: {str(e)}")
+            return pd.DataFrame()
 
     @staticmethod
     def get_stock_shares(stock_code: str) -> pd.DataFrame:
diff --git a/cryptoai/routes/adata.py b/cryptoai/routes/adata.py
index cdcca6e..5947448 100644
--- a/cryptoai/routes/adata.py
+++ b/cryptoai/routes/adata.py
@@ -34,6 +34,10 @@ async def get_stock_base(stock_code: str):
     result = {}
 
     try:
+        # 获取核心财务指标
+        core_index = api.get_stock_finance_core_index(stock_code)
+        result["finance_core_index"] = json.loads(core_index.to_json(orient="records"))
+
         # 获取股本信息
         stock_shares = api.get_stock_shares(stock_code)
         result["stock_shares"] = json.loads(stock_shares.to_json(orient="records"))
@@ -124,8 +128,6 @@ async def get_stock_data_all(stock_code: str):
 
 @router.post('/{stock_code}/analysis', summary="获取股票分析数据")
 async def get_stock_analysis(stock_code: str, current_user: Dict[str, Any] = Depends(get_current_user)):
-    
-
     url = 'https://mate.aimateplus.com/v1/workflows/run'
     token = 'app-nWuCOa0YfQVtAosTY3Jr5vFV'
     headers = {
diff --git a/docker-compose.yml b/docker-compose.yml
index c933ae7..271f3f8 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.1.2
+    image: cryptoai-api:0.1.3
     restart: always
     ports:
       - "8000:8000"