From 318e5353e7dadca680a6391e2bc2f9b76938fc0e Mon Sep 17 00:00:00 2001 From: looccasgtr Date: Thu, 8 May 2025 16:59:49 -0400 Subject: [PATCH] Refeita todas as rotas com FLASK. --- app/routes/ocr_results_routes.py | 35 +++++++++++++------------------- 1 file changed, 14 insertions(+), 21 deletions(-) diff --git a/app/routes/ocr_results_routes.py b/app/routes/ocr_results_routes.py index d4cfd6c..6ac1fc7 100644 --- a/app/routes/ocr_results_routes.py +++ b/app/routes/ocr_results_routes.py @@ -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