128 lines
3.5 KiB
Python
128 lines
3.5 KiB
Python
"""
|
|
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
|