55 lines
2.2 KiB
Python
55 lines
2.2 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.api.binance_api import get_binance_api
|
|
from cryptoai.models.data_processor import DataProcessor
|
|
from datetime import timedelta
|
|
from cryptoai.models.token import TokenManager
|
|
# 创建路由
|
|
router = APIRouter()
|
|
|
|
logger = logging.getLogger(__name__)
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
@router.get("/search/{key}")
|
|
async def search_crypto(key: str):
|
|
manager = TokenManager()
|
|
result = manager.search_token(key)
|
|
return result
|
|
|
|
class CryptoAnalysisRequest(BaseModel):
|
|
symbol: str
|
|
timeframe: Optional[str] = None
|
|
|
|
@router.get("/kline/{symbol}")
|
|
async def get_crypto_kline(symbol: str, timeframe: Optional[str] = None, limit: Optional[int] = 100):
|
|
# 检查symbol是否存在
|
|
token_manager = TokenManager()
|
|
tokens = token_manager.search_token(symbol)
|
|
if not tokens or len(tokens) == 0:
|
|
raise HTTPException(status_code=400, detail="您输入的币种在币安不存在，请检查后重新输入。")
|
|
|
|
symbol = tokens[0]["symbol"]
|
|
|
|
binance_api = get_binance_api()
|
|
result = {}
|
|
|
|
if timeframe is None:
|
|
result["15m"] = binance_api.get_historical_klines(symbol=symbol, interval="15m", limit=limit).to_dict(orient="records")
|
|
result["1h"] = binance_api.get_historical_klines(symbol=symbol, interval="1h", limit=limit).to_dict(orient="records")
|
|
result["4h"] = binance_api.get_historical_klines(symbol=symbol, interval="4h", limit=limit).to_dict(orient="records")
|
|
# result["1d"] = binance_api.get_historical_klines(symbol=symbol, interval="1d", limit=limit).to_dict(orient="records")
|
|
else:
|
|
result[timeframe] = binance_api.get_historical_klines(symbol=symbol, interval=timeframe, limit=limit).to_dict(orient="records")
|
|
|
|
return result |