import json

from fastapi import APIRouter, Depends, HTTPException, status, Body, Query, Path
from cryptoai.api.adata_api import AStockAPI

# 创建路由
router = APIRouter()


@router.get("/stock/data")
async def get_stock_data(stock_code: str):
    
    result = {}

    api = AStockAPI()

    # 获取股本信息
    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"))

    # 获取市场数据
    market_data = api.get_market_data(stock_code)
    result["market_data"] = json.loads(market_data.to_json(orient="records"))

    # # 获取分钟线数据
    # min_data = api.get_market_min_data(stock_code)
    # result["min_data"] = json.loads(min_data.to_json(orient="records"))

    # 获取资金流向数据
    flow_data = api.get_capital_flow(stock_code)
    result["flow_data"] = json.loads(flow_data.to_json(orient="records"))
    
    return result