86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
import json
|
|
import logging
|
|
from fastapi import APIRouter, Depends, HTTPException, status, Body, Query, Path
|
|
from cryptoai.api.adata_api import AStockAPI
|
|
from datetime import datetime
|
|
from typing import Dict, Any, List, Optional
|
|
from pydantic import BaseModel
|
|
|
|
from cryptoai.api.deepseek_api import DeepSeekAPI
|
|
from cryptoai.utils.config_loader import ConfigLoader
|
|
from fastapi.responses import StreamingResponse
|
|
from cryptoai.routes.user import get_current_user
|
|
import requests
|
|
from cryptoai.models.astock import AStockManager
|
|
from sqlalchemy.orm import Session
|
|
from cryptoai.utils.db_manager import get_db
|
|
# 创建路由
|
|
router = APIRouter()
|
|
|
|
logger = logging.getLogger(__name__)
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
@router.get("/stock/search")
|
|
async def search_stock(key: str, limit: int = 10, session: Session = Depends(get_db)):
|
|
manager = AStockManager(session)
|
|
result = manager.search_stock(key, limit)
|
|
|
|
return result
|
|
|
|
|
|
@router.get("/stock/base", summary="获取股票基础信息")
|
|
async def get_stock_base(stock_code: str, session: Session = Depends(get_db)):
|
|
api = AStockAPI()
|
|
manager = AStockManager(session)
|
|
|
|
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"))
|
|
|
|
# # 获取概念板块
|
|
# concept_east = api.get_concept_east(stock_code)
|
|
# result["concept_east"] = json.loads(concept_east.to_json(orient="records"))
|
|
|
|
# 获取板块
|
|
plate_east = api.get_plate_east(stock_code)
|
|
result["plate_east"] = json.loads(plate_east.to_json(orient="records"))
|
|
|
|
except Exception as e:
|
|
logger.error(f"获取股票基础信息失败: {e}")
|
|
return {}
|
|
|
|
return result
|
|
|
|
@router.get("/stock/data", summary="获取股票数据")
|
|
async def get_stock_data(stock_code: str, start_date: Optional[str] = None, end_date: Optional[str] = None):
|
|
api = AStockAPI()
|
|
|
|
result = {}
|
|
|
|
try:
|
|
if start_date is None:
|
|
start_date = "2025-01-01"
|
|
if end_date is None:
|
|
end_date = datetime.now().strftime("%Y-%m-%d")
|
|
|
|
# 获取市场数据
|
|
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, start_date, end_date)
|
|
result["flow_data"] = json.loads(flow_data.to_json(orient="records"))
|
|
|
|
|
|
except Exception as e:
|
|
logger.error(f"获取股票数据失败: {e}")
|
|
return {}
|
|
|
|
return result |