File size: 3,697 Bytes
c3bf538 c6fb015 8427d6e c6fb015 c3bf538 c6fb015 c3bf538 c6fb015 c3bf538 8427d6e c6fb015 c3bf538 8427d6e c3bf538 8427d6e c6fb015 8427d6e c6fb015 c3bf538 c6fb015 8427d6e c6fb015 c3bf538 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 |
# from fastapi import FastAPI, Depends, HTTPException
# from fastapi.middleware.cors import CORSMiddleware
# from sqlalchemy.orm import Session
# from uuid import UUID
# import models.analysis_job as model
# import schemas
# from core.database import SessionLocal, engine
# # Import only our single, main task
# from tasks.main_task import run_full_analysis
# # This ensures tables are created on startup (good for development)
# model.Base.metadata.create_all(bind=engine)
# app = FastAPI(
# title="Quantitative Analysis Platform API",
# version="0.1.0",
# )
# app.add_middleware(
# CORSMiddleware,
# allow_origins=["*"],
# allow_credentials=True,
# allow_methods=["*"],
# allow_headers=["*"],
# )
# # Dependency to get a database session for each API request
# def get_db():
# db = SessionLocal()
# try:
# yield db
# finally:
# db.close()
# @app.post("/jobs", response_model=schemas.Job, status_code=201)
# def create_analysis_job(job_request: schemas.JobCreate, db: Session = Depends(get_db)):
# """
# Creates a new analysis job and dispatches the main background task.
# """
# db_job = model.AnalysisJob(ticker=job_request.ticker.upper())
# db.add(db_job)
# db.commit()
# db.refresh(db_job)
# # Dispatch our single, all-in-one analysis task to run in the background
# run_full_analysis.delay(str(db_job.id), db_job.ticker)
# # Immediately return the created job to the frontend
# return db_job
# @app.get("/jobs/{job_id}", response_model=schemas.Job)
# def get_job_status(job_id: UUID, db: Session = Depends(get_db)):
# """
# Allows the frontend to poll for the status and result of an analysis job.
# """
# db_job = db.query(model.AnalysisJob).filter(model.AnalysisJob.id == job_id).first()
# if db_job is None:
# raise HTTPException(status_code=404, detail="Job not found")
# return db_job
from fastapi import FastAPI, Depends, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from sqlalchemy.orm import Session
from sqlalchemy import desc # Import desc for ordering
from uuid import UUID
from typing import List # Import List for the history endpoint
import models.analysis_job as model
import schemas
from core.database import SessionLocal, engine
from tasks.main_task import run_full_analysis
model.Base.metadata.create_all(bind=engine)
app = FastAPI(
title="Quantitative Analysis Platform API",
version="0.1.0",
)
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()
@app.post("/jobs", response_model=schemas.Job, status_code=201)
def create_analysis_job(job_request: schemas.JobCreate, db: Session = Depends(get_db)):
db_job = model.AnalysisJob(ticker=job_request.ticker.upper())
db.add(db_job)
db.commit()
db.refresh(db_job)
run_full_analysis.delay(str(db_job.id), db_job.ticker)
return db_job
@app.get("/jobs/{job_id}", response_model=schemas.Job)
def get_job_status(job_id: UUID, db: Session = Depends(get_db)):
db_job = db.query(model.AnalysisJob).filter(model.AnalysisJob.id == job_id).first()
if db_job is None:
raise HTTPException(status_code=404, detail="Job not found")
return db_job
# --- NEW ENDPOINT FOR HISTORY PANEL ---
@app.get("/jobs", response_model=List[schemas.Job])
def get_jobs_history(db: Session = Depends(get_db)):
db_jobs = db.query(model.AnalysisJob).order_by(desc(model.AnalysisJob.created_at)).limit(20).all()
return db_jobs |