Refeita todas as rotas com FLASK.

This commit is contained in:
looccasgtr 2025-05-08 16:59:49 -04:00
parent 29b7b39184
commit 318e5353e7

View File

@ -1,26 +1,19 @@
from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session
from flask import Blueprint, request, jsonify
from models.ocr_result import OCRResult
from database import SessionLocal
from models.base import db
import uuid
router = APIRouter(prefix="/ocr", tags=["OCR"])
ocr_routes = Blueprint("ocr_routes", __name__, url_prefix="/ocr")
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()
@ocr_routes.route("/", methods=["POST"])
def create_ocr():
documento_id = request.json.get("documento_id")
new_ocr = OCRResult(documento_id=documento_id)
db.session.add(new_ocr)
db.session.commit()
return jsonify({"id": new_ocr.id, "documento_id": new_ocr.documento_id}), 201
@router.post("/")
def create_ocr(documento_id: uuid.UUID, db: Session = Depends(get_db)):
ocr = OCRResult(documento_id=documento_id)
db.add(ocr)
db.commit()
db.refresh(ocr)
return ocr
@router.get("/")
def list_ocr(db: Session = Depends(get_db)):
return db.query(OCRResult).all()
@ocr_routes.route("/", methods=["GET"])
def list_ocr():
ocrs = OCRResult.query.all()
return jsonify([{"id": ocr.id, "documento_id": ocr.documento_id} for ocr in ocrs]), 200