forked from Anteros-Code-Mentoria/poc-mvc-ocr
Refeita todas as rotas com FLASK.
This commit is contained in:
parent
29b7b39184
commit
318e5353e7
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user