feat: add admin API for data collection management
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
parent
8cc2d3fa41
commit
d2c5f91b2b
@ -1,3 +1,4 @@
|
|||||||
from app.api.auth import router as auth_router
|
from app.api.auth import router as auth_router
|
||||||
|
from app.api.admin import router as admin_router
|
||||||
|
|
||||||
__all__ = ["auth_router"]
|
__all__ = ["auth_router", "admin_router"]
|
||||||
|
|||||||
127
backend/app/api/admin.py
Normal file
127
backend/app/api/admin.py
Normal file
@ -0,0 +1,127 @@
|
|||||||
|
"""
|
||||||
|
Admin API for data collection management.
|
||||||
|
"""
|
||||||
|
from typing import List
|
||||||
|
|
||||||
|
from fastapi import APIRouter, Depends, HTTPException
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
from app.core.database import get_db
|
||||||
|
from app.api.deps import CurrentUser
|
||||||
|
from app.models.stock import JobLog
|
||||||
|
from app.services.collectors import (
|
||||||
|
StockCollector,
|
||||||
|
SectorCollector,
|
||||||
|
PriceCollector,
|
||||||
|
ValuationCollector,
|
||||||
|
)
|
||||||
|
|
||||||
|
router = APIRouter(prefix="/api/admin", tags=["admin"])
|
||||||
|
|
||||||
|
|
||||||
|
class JobLogResponse(BaseModel):
|
||||||
|
id: int
|
||||||
|
job_name: str
|
||||||
|
status: str
|
||||||
|
started_at: str
|
||||||
|
finished_at: str | None
|
||||||
|
records_count: int | None
|
||||||
|
error_msg: str | None
|
||||||
|
|
||||||
|
class Config:
|
||||||
|
from_attributes = True
|
||||||
|
|
||||||
|
|
||||||
|
class CollectResponse(BaseModel):
|
||||||
|
message: str
|
||||||
|
job_id: int
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/collect/stocks", response_model=CollectResponse)
|
||||||
|
async def collect_stocks(
|
||||||
|
current_user: CurrentUser,
|
||||||
|
db: Session = Depends(get_db),
|
||||||
|
biz_day: str = None,
|
||||||
|
):
|
||||||
|
"""Collect stock master data from KRX."""
|
||||||
|
try:
|
||||||
|
collector = StockCollector(db, biz_day=biz_day)
|
||||||
|
job_log = collector.run()
|
||||||
|
return CollectResponse(
|
||||||
|
message=f"Stock collection completed: {job_log.records_count} records",
|
||||||
|
job_id=job_log.id,
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
raise HTTPException(status_code=500, detail=str(e))
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/collect/sectors", response_model=CollectResponse)
|
||||||
|
async def collect_sectors(
|
||||||
|
current_user: CurrentUser,
|
||||||
|
db: Session = Depends(get_db),
|
||||||
|
biz_day: str = None,
|
||||||
|
):
|
||||||
|
"""Collect sector classification data from WISEindex."""
|
||||||
|
try:
|
||||||
|
collector = SectorCollector(db, biz_day=biz_day)
|
||||||
|
job_log = collector.run()
|
||||||
|
return CollectResponse(
|
||||||
|
message=f"Sector collection completed: {job_log.records_count} records",
|
||||||
|
job_id=job_log.id,
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
raise HTTPException(status_code=500, detail=str(e))
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/collect/prices", response_model=CollectResponse)
|
||||||
|
async def collect_prices(
|
||||||
|
current_user: CurrentUser,
|
||||||
|
db: Session = Depends(get_db),
|
||||||
|
start_date: str = None,
|
||||||
|
end_date: str = None,
|
||||||
|
):
|
||||||
|
"""Collect price data using pykrx."""
|
||||||
|
try:
|
||||||
|
collector = PriceCollector(db, start_date=start_date, end_date=end_date)
|
||||||
|
job_log = collector.run()
|
||||||
|
return CollectResponse(
|
||||||
|
message=f"Price collection completed: {job_log.records_count} records",
|
||||||
|
job_id=job_log.id,
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
raise HTTPException(status_code=500, detail=str(e))
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/collect/valuations", response_model=CollectResponse)
|
||||||
|
async def collect_valuations(
|
||||||
|
current_user: CurrentUser,
|
||||||
|
db: Session = Depends(get_db),
|
||||||
|
biz_day: str = None,
|
||||||
|
):
|
||||||
|
"""Collect valuation data from KRX."""
|
||||||
|
try:
|
||||||
|
collector = ValuationCollector(db, biz_day=biz_day)
|
||||||
|
job_log = collector.run()
|
||||||
|
return CollectResponse(
|
||||||
|
message=f"Valuation collection completed: {job_log.records_count} records",
|
||||||
|
job_id=job_log.id,
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
raise HTTPException(status_code=500, detail=str(e))
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/collect/status", response_model=List[JobLogResponse])
|
||||||
|
async def get_collection_status(
|
||||||
|
current_user: CurrentUser,
|
||||||
|
db: Session = Depends(get_db),
|
||||||
|
limit: int = 20,
|
||||||
|
):
|
||||||
|
"""Get recent job execution status."""
|
||||||
|
jobs = (
|
||||||
|
db.query(JobLog)
|
||||||
|
.order_by(JobLog.started_at.desc())
|
||||||
|
.limit(limit)
|
||||||
|
.all()
|
||||||
|
)
|
||||||
|
return jobs
|
||||||
@ -4,7 +4,7 @@ Galaxy-PO Backend API
|
|||||||
from fastapi import FastAPI
|
from fastapi import FastAPI
|
||||||
from fastapi.middleware.cors import CORSMiddleware
|
from fastapi.middleware.cors import CORSMiddleware
|
||||||
|
|
||||||
from app.api import auth_router
|
from app.api import auth_router, admin_router
|
||||||
|
|
||||||
app = FastAPI(
|
app = FastAPI(
|
||||||
title="Galaxy-PO API",
|
title="Galaxy-PO API",
|
||||||
@ -22,6 +22,7 @@ app.add_middleware(
|
|||||||
|
|
||||||
# Include routers
|
# Include routers
|
||||||
app.include_router(auth_router)
|
app.include_router(auth_router)
|
||||||
|
app.include_router(admin_router)
|
||||||
|
|
||||||
|
|
||||||
@app.get("/health")
|
@app.get("/health")
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user