Rotas de tag, doc_tag e ocr_results.

This commit is contained in:
looccasgtr 2025-05-02 17:50:04 -04:00
parent f94f7a99cd
commit 29b7b39184
3 changed files with 77 additions and 0 deletions

View File

@ -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()

View File

@ -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()

25
app/routes/tag_routes.py Normal file
View File

@ -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()