forked from Anteros-Code-Mentoria/poc-mvc-ocr
Rotas de tag, doc_tag e ocr_results.
This commit is contained in:
parent
f94f7a99cd
commit
29b7b39184
26
app/routes/doc_tag_routes.py
Normal file
26
app/routes/doc_tag_routes.py
Normal 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()
|
||||||
26
app/routes/ocr_results_routes.py
Normal file
26
app/routes/ocr_results_routes.py
Normal 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
25
app/routes/tag_routes.py
Normal 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()
|
||||||
Loading…
Reference in New Issue
Block a user