app/routes/doc_tag_routes.py

@@ -1,20 +1,26 @@
-from flask import Blueprint, request, jsonify
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
 from models.doc_tag import DocTag
-from models.base import db
+from database import SessionLocal
 import uuid
 
-doc_tag_routes = Blueprint("doc_tag_routes", __name__, url_prefix="/doc-tags")
+router = APIRouter(prefix="/doc-tags", tags=["DocTag"])
 
-@doc_tag_routes.route("/", methods=["POST"])
-def create_doc_tag():
-    documento_id = request.json.get("documento_id")
-    tag_id = request.json.get("tag_id")
-    new_doc_tag = DocTag(documento_id=documento_id, tag_id=tag_id)
-    db.session.add(new_doc_tag)
-    db.session.commit()
-    return jsonify({"id": new_doc_tag.id, "documento_id": new_doc_tag.documento_id, "tag_id": new_doc_tag.tag_id}), 201
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
 
-@doc_tag_routes.route("/", methods=["GET"])
-def list_doc_tags():
-    doc_tags = DocTag.query.all()
-    return jsonify([{"id": dt.id, "documento_id": dt.documento_id, "tag_id": dt.tag_id} for dt in doc_tags]), 200
+@router.post("/")
+def create_doc_tag(documento_id: uuid.UUID, tag_id: uuid.UUID, db: Session = Depends(get_db)):
+    dt = DocTag(documento_id=documento_id, tag_id=tag_id)
+    db.add(dt)
+    db.commit()
+    db.refresh(dt)
+    return dt
+
+@router.get("/")
+def list_doc_tags(db: Session = Depends(get_db)):
+    return db.query(DocTag).all()

app/routes/ocr_results_routes.py

@@ -1,19 +1,26 @@
-from flask import Blueprint, request, jsonify
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
 from models.ocr_result import OCRResult
-from models.base import db
+from database import SessionLocal
 import uuid
 
-ocr_routes = Blueprint("ocr_routes", __name__, url_prefix="/ocr")
+router = APIRouter(prefix="/ocr", tags=["OCR"])
 
-@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
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
 
-@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
+@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()

app/routes/tag_routes.py

@@ -1,18 +1,25 @@
-from flask import Blueprint, request, jsonify
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
 from models.tag import Tag
-from models.base import db
+from database import SessionLocal
 
-tag_routes = Blueprint("tag_routes", __name__, url_prefix="/tags")
+router = APIRouter(prefix="/tags", tags=["Tags"])
 
-@tag_routes.route("/", methods=["POST"])
-def create_tag():
-    nome = request.json.get("nome")
-    new_tag = Tag(nome=nome)
-    db.session.add(new_tag)
-    db.session.commit()
-    return jsonify({"id": new_tag.id, "nome": new_tag.nome}), 201
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
 
-@tag_routes.route("/", methods=["GET"])
-def list_tags():
-    tags = Tag.query.all()
-    return jsonify([{"id": tag.id, "nome": tag.nome} for tag in tags]), 200
+@router.post("/")
+def create_tag(nome: str, db: Session = Depends(get_db)):
+    tag = Tag(nome=nome)
+    db.add(tag)
+    db.commit()
+    db.refresh(tag)
+    return tag
+
+@router.get("/")
+def list_tags(db: Session = Depends(get_db)):
+    return db.query(Tag).all()