diff --git a/app/routes/doc_tag_routes.py b/app/routes/doc_tag_routes.py new file mode 100644 index 0000000..6225de2 --- /dev/null +++ b/app/routes/doc_tag_routes.py @@ -0,0 +1,26 @@ +from fastapi import APIRouter, Depends +from sqlalchemy.orm import Session +from models.doc_tag import DocTag +from database import SessionLocal +import uuid + +router = APIRouter(prefix="/doc-tags", tags=["DocTag"]) + +def get_db(): + db = SessionLocal() + try: + yield db + finally: + db.close() + +@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() diff --git a/app/routes/ocr_results_routes.py b/app/routes/ocr_results_routes.py new file mode 100644 index 0000000..d4cfd6c --- /dev/null +++ b/app/routes/ocr_results_routes.py @@ -0,0 +1,26 @@ +from fastapi import APIRouter, Depends +from sqlalchemy.orm import Session +from models.ocr_result import OCRResult +from database import SessionLocal +import uuid + +router = APIRouter(prefix="/ocr", tags=["OCR"]) + +def get_db(): + db = SessionLocal() + try: + yield db + finally: + db.close() + +@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() diff --git a/app/routes/tag_routes.py b/app/routes/tag_routes.py new file mode 100644 index 0000000..55519bb --- /dev/null +++ b/app/routes/tag_routes.py @@ -0,0 +1,25 @@ +from fastapi import APIRouter, Depends +from sqlalchemy.orm import Session +from models.tag import Tag +from database import SessionLocal + +router = APIRouter(prefix="/tags", tags=["Tags"]) + +def get_db(): + db = SessionLocal() + try: + yield db + finally: + db.close() + +@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()