dekun
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
"""API 路由：策略"""
|
|
|
|
from typing import List
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.database import get_db
|
|
from app import crud, schemas
|
|
|
|
router = APIRouter(prefix="/api/strategies", tags=["策略"])
|
|
|
|
|
|
@router.get("", response_model=List[schemas.StrategyOut])
|
|
def list_strategies(db: Session = Depends(get_db)):
|
|
return crud.get_strategies(db)
|
|
|
|
|
|
@router.post("", response_model=schemas.StrategyOut, status_code=201)
|
|
def create_strategy(data: schemas.StrategyCreate, db: Session = Depends(get_db)):
|
|
return crud.create_strategy(db, data)
|
|
|
|
|
|
@router.put("/{strategy_id}", response_model=schemas.StrategyOut)
|
|
def update_strategy(strategy_id: int, data: schemas.StrategyUpdate, db: Session = Depends(get_db)):
|
|
obj = crud.update_strategy(db, strategy_id, data)
|
|
if not obj:
|
|
raise HTTPException(404, "策略不存在")
|
|
return obj
|
|
|
|
|
|
@router.delete("/{strategy_id}")
|
|
def delete_strategy(strategy_id: int, db: Session = Depends(get_db)):
|
|
if not crud.delete_strategy(db, strategy_id):
|
|
raise HTTPException(404, "策略不存在")
|
|
return {"ok": True}
|