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