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
# 创建路由
router = APIRouter()

logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)

@router.get("/stock/search")
async def search_stock(key: str, limit: int = 10):
    manager = AStockManager()
    result = manager.search_stock(key, limit)

    return result


@router.get("/stock/base", summary="获取股票基础信息")
async def get_stock_base(stock_code: str):
    api = AStockAPI()
    
    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