feat: add market data API endpoints

- GET /api/market/stocks/{ticker}
- GET /api/market/stocks/{ticker}/prices
- GET /api/market/search

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
zephyrdark 2026-02-03 09:01:48 +09:00
parent 6a6ccc9c57
commit bc484fcb07
3 changed files with 116 additions and 2 deletions

View File

@ -2,5 +2,6 @@ from app.api.auth import router as auth_router
from app.api.admin import router as admin_router
from app.api.portfolio import router as portfolio_router
from app.api.strategy import router as strategy_router
from app.api.market import router as market_router
__all__ = ["auth_router", "admin_router", "portfolio_router", "strategy_router"]
__all__ = ["auth_router", "admin_router", "portfolio_router", "strategy_router", "market_router"]

112
backend/app/api/market.py Normal file
View File

@ -0,0 +1,112 @@
"""
Market data API endpoints.
"""
from typing import List, Optional
from datetime import date, timedelta
from fastapi import APIRouter, Depends, HTTPException, Query
from sqlalchemy.orm import Session
from app.core.database import get_db
from app.api.deps import CurrentUser
from app.models.stock import Stock, Valuation, Price, Sector
from app.schemas.strategy import StockInfo, StockSearchResult, PriceData
router = APIRouter(prefix="/api/market", tags=["market"])
@router.get("/stocks/{ticker}", response_model=StockInfo)
async def get_stock(
ticker: str,
current_user: CurrentUser,
db: Session = Depends(get_db),
):
"""Get stock information."""
stock = db.query(Stock).filter(Stock.ticker == ticker).first()
if not stock:
raise HTTPException(status_code=404, detail="Stock not found")
valuation = (
db.query(Valuation)
.filter(Valuation.ticker == ticker)
.order_by(Valuation.base_date.desc())
.first()
)
sector = db.query(Sector).filter(Sector.ticker == ticker).first()
return StockInfo(
ticker=stock.ticker,
name=stock.name,
market=stock.market,
sector_name=sector.sector_name if sector else None,
stock_type=stock.stock_type.value if stock.stock_type else "common",
close_price=stock.close_price,
market_cap=int(stock.market_cap / 100_000_000) if stock.market_cap else None,
per=valuation.per if valuation else None,
pbr=valuation.pbr if valuation else None,
psr=valuation.psr if valuation else None,
pcr=valuation.pcr if valuation else None,
dividend_yield=valuation.dividend_yield if valuation else None,
eps=stock.eps,
bps=stock.bps,
base_date=stock.base_date,
)
@router.get("/stocks/{ticker}/prices", response_model=List[PriceData])
async def get_stock_prices(
ticker: str,
current_user: CurrentUser,
db: Session = Depends(get_db),
days: int = Query(default=365, ge=1, le=3650),
):
"""Get historical prices for a stock."""
start_date = date.today() - timedelta(days=days)
prices = (
db.query(Price)
.filter(Price.ticker == ticker)
.filter(Price.date >= start_date)
.order_by(Price.date.desc())
.all()
)
return [
PriceData(
date=p.date,
open=p.open,
high=p.high,
low=p.low,
close=p.close,
volume=p.volume,
)
for p in prices
]
@router.get("/search", response_model=List[StockSearchResult])
async def search_stocks(
q: str = Query(..., min_length=1),
current_user: CurrentUser = None,
db: Session = Depends(get_db),
limit: int = Query(default=20, ge=1, le=100),
):
"""Search stocks by ticker or name."""
stocks = (
db.query(Stock)
.filter(
(Stock.ticker.ilike(f"%{q}%")) | (Stock.name.ilike(f"%{q}%"))
)
.limit(limit)
.all()
)
return [
StockSearchResult(
ticker=s.ticker,
name=s.name,
market=s.market,
)
for s in stocks
]

View File

@ -4,7 +4,7 @@ Galaxy-PO Backend API
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from app.api import auth_router, admin_router, portfolio_router, strategy_router
from app.api import auth_router, admin_router, portfolio_router, strategy_router, market_router
app = FastAPI(
title="Galaxy-PO API",
@ -25,6 +25,7 @@ app.include_router(auth_router)
app.include_router(admin_router)
app.include_router(portfolio_router)
app.include_router(strategy_router)
app.include_router(market_router)
@app.get("/health")