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 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()
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user