from fastapi import APIRouter, Depends, HTTPException, Query from sqlalchemy.orm import Session from ..database import get_db from ..models import Event, ParseJob from ..schemas import ( ConsumerCreate, ConsumerRead, EventRead, ParseJobCreate, ParseJobRead, ) from ..services.filtering import create_consumer from ..services.jobs import enqueue_job router = APIRouter(prefix="/admin", tags=["admin"]) @router.post("/jobs", response_model=ParseJobRead, status_code=201) def create_parse_job(payload: ParseJobCreate, db: Session = Depends(get_db)): job = ParseJob( source_type=payload.source_type, source_config=payload.source_config, schedule=payload.schedule, status="queued", ) db.add(job) db.commit() db.refresh(job) enqueue_job(job.id, job.source_type, job.source_config) return job @router.get("/jobs", response_model=list[ParseJobRead]) def list_parse_jobs(db: Session = Depends(get_db)): return db.query(ParseJob).order_by(ParseJob.id.desc()).all() @router.get("/jobs/{job_id}", response_model=ParseJobRead) def get_parse_job(job_id: int, db: Session = Depends(get_db)): job = db.query(ParseJob).filter(ParseJob.id == job_id).first() if not job: raise HTTPException(status_code=404, detail="Job not found") return job @router.get("/events", response_model=list[EventRead]) def list_events( limit: int = Query(default=100, ge=1, le=1000), offset: int = Query(default=0, ge=0), db: Session = Depends(get_db), ): return ( db.query(Event) .order_by(Event.ingested_at.desc()) .offset(offset) .limit(limit) .all() ) @router.post("/consumers", response_model=ConsumerRead, status_code=201) def create_pi_consumer(payload: ConsumerCreate, db: Session = Depends(get_db)): consumer, api_key = create_consumer( db, name=payload.name, regions=payload.regions, topics=payload.topics, date_from=payload.date_from, ) return ConsumerRead( id=consumer.id, name=consumer.name, is_active=consumer.is_active, created_at=consumer.created_at, api_key=api_key, )