69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
from fastapi import APIRouter, Depends
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy import select
|
|
from typing import List
|
|
from pydantic import BaseModel
|
|
|
|
from app.db.database import get_db
|
|
from app.db.models import MasterStock, DiscoveryRankingCache, StockItem
|
|
|
|
router = APIRouter()
|
|
|
|
class StockItemSchema(BaseModel):
|
|
code: str
|
|
name: str
|
|
market: str
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
class RankingItemSchema(BaseModel):
|
|
code: str
|
|
name: str
|
|
price: float
|
|
change: float
|
|
changePercent: float
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
@router.get("/rankings", response_model=List[RankingItemSchema])
|
|
async def get_rankings(type: str = "rise", limit: int = 10, db: AsyncSession = Depends(get_db)):
|
|
"""
|
|
Get Top Rankings based on cached StockItem data.
|
|
type: 'rise' (Top Gainers), 'fall' (Top Losers), 'volume' (Not impl yet)
|
|
"""
|
|
stmt = select(StockItem)
|
|
|
|
if type == "fall":
|
|
stmt = stmt.order_by(StockItem.changePercent.asc())
|
|
else:
|
|
stmt = stmt.order_by(StockItem.changePercent.desc())
|
|
|
|
stmt = stmt.limit(limit)
|
|
res = await db.execute(stmt)
|
|
items = res.scalars().all()
|
|
|
|
return items
|
|
|
|
@router.get("/stocks/search", response_model=List[StockItemSchema])
|
|
async def search_stocks(query: str, db: AsyncSession = Depends(get_db)):
|
|
# Search by name or code
|
|
stmt = select(MasterStock).where(
|
|
(MasterStock.name.like(f"%{query}%")) | (MasterStock.code.like(f"%{query}%"))
|
|
).limit(20)
|
|
result = await db.execute(stmt)
|
|
stocks = result.scalars().all()
|
|
return stocks
|
|
|
|
from app.services.master_service import master_service
|
|
|
|
@router.get("/kis/master-stocks")
|
|
async def sync_master_stocks(db: AsyncSession = Depends(get_db)):
|
|
# Trigger Sync
|
|
# Ideally should be BackgroundTasks, but for now await to show result
|
|
await master_service.sync_master_data(db)
|
|
|
|
# Return count
|
|
stmt = select(MasterStock).limit(1)
|
|
# Just return status
|
|
return {"status": "Sync Complete"}
|