|
|
|
|
|
"""
|
|
|
|
|
|
股票数据路由
|
|
|
|
|
|
"""
|
|
|
|
|
|
from typing import List
|
|
|
|
|
|
from fastapi import APIRouter, Depends, Query
|
|
|
|
|
|
from sqlalchemy.orm import Session
|
|
|
|
|
|
|
|
|
|
|
|
from app.db.session import get_db
|
|
|
|
|
|
from app.schemas.base import ResponseModel
|
|
|
|
|
|
from app.schemas.kline import KlineRequest, BatchKlineRequest
|
|
|
|
|
|
from app.services.stock_service import StockService
|
|
|
|
|
|
from app.core.security import get_current_user
|
|
|
|
|
|
from app.models.user import User
|
|
|
|
|
|
from app.utils.date_utils import parse_date
|
|
|
|
|
|
|
|
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@router.get("/kline", response_model=ResponseModel)
|
|
|
|
|
|
async def get_stock_kline(
|
|
|
|
|
|
codes: str = Query(..., description="股票代码，多个用逗号分隔"),
|
|
|
|
|
|
start_date: str = Query(..., description="开始日期(YYYYMMDD)"),
|
|
|
|
|
|
end_date: str = Query(..., description="结束日期(YYYYMMDD)"),
|
|
|
|
|
|
period: str = Query("daily", description="周期: daily, min1, min5, min15, min30, min60"),
|
|
|
|
|
|
db: Session = Depends(get_db),
|
|
|
|
|
|
current_user: User = Depends(get_current_user)
|
|
|
|
|
|
):
|
|
|
|
|
|
"""获取股票K线数据"""
|
|
|
|
|
|
service = StockService(db)
|
|
|
|
|
|
code_list = [c.strip() for c in codes.split(",")]
|
|
|
|
|
|
start = parse_date(start_date)
|
|
|
|
|
|
end = parse_date(end_date)
|
|
|
|
|
|
|
|
|
|
|
|
data = service.get_kline(code_list, start, end, period)
|
|
|
|
|
|
return ResponseModel(data=data)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@router.post("/kline/batch", response_model=ResponseModel)
|
|
|
|
|
|
async def batch_get_stock_kline(
|
|
|
|
|
|
request: BatchKlineRequest,
|
|
|
|
|
|
db: Session = Depends(get_db),
|
|
|
|
|
|
current_user: User = Depends(get_current_user)
|
|
|
|
|
|
):
|
|
|
|
|
|
"""批量获取股票K线数据"""
|
|
|
|
|
|
service = StockService(db)
|
|
|
|
|
|
start = parse_date(request.start_date)
|
|
|
|
|
|
end = parse_date(request.end_date)
|
|
|
|
|
|
|
|
|
|
|
|
data = service.get_kline(request.codes, start, end, request.period)
|
|
|
|
|
|
return ResponseModel(data=data)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@router.get("/kline/{code}/chart", response_model=ResponseModel)
|
|
|
|
|
|
async def get_stock_kline_chart(
|
|
|
|
|
|
code: str,
|
|
|
|
|
|
start_date: str = Query(..., description="开始日期(YYYYMMDD)"),
|
|
|
|
|
|
end_date: str = Query(..., description="结束日期(YYYYMMDD)"),
|
|
|
|
|
|
period: str = Query("daily", description="周期: daily, min1, min5, min15, min30, min60"),
|
|
|
|
|
|
db: Session = Depends(get_db),
|
|
|
|
|
|
current_user: User = Depends(get_current_user)
|
|
|
|
|
|
):
|
|
|
|
|
|
"""获取股票K线图数据（ECharts格式）"""
|
|
|
|
|
|
service = StockService(db)
|
|
|
|
|
|
start = parse_date(start_date)
|
|
|
|
|
|
end = parse_date(end_date)
|
|
|
|
|
|
|
|
|
|
|
|
data = service.get_kline_chart_data(code, start, end, period)
|
|
|
|
|
|
return ResponseModel(data=data)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@router.get("/snapshot", response_model=ResponseModel)
|
|
|
|
|
|
async def get_stock_snapshot(
|
|
|
|
|
|
codes: str = Query(..., description="股票代码，多个用逗号分隔"),
|
|
|
|
|
|
start_date: str = Query(..., description="开始日期(YYYYMMDD)"),
|
|
|
|
|
|
end_date: str = Query(..., description="结束日期(YYYYMMDD)"),
|
|
|
|
|
|
db: Session = Depends(get_db),
|
|
|
|
|
|
current_user: User = Depends(get_current_user)
|
|
|
|
|
|
):
|
|
|
|
|
|
"""获取股票历史快照数据"""
|
|
|
|
|
|
# 这里可以实现快照数据查询
|
|
|
|
|
|
return ResponseModel(data={"message": "功能开发中"})
|
