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