Post measure

1 files changed, 14 insertions, 0 deletions

diff --git a/api/routes/measures.py b/api/routes/measures.py
index 195da9a..b435ae3 100644
--- a/api/routes/measures.py
+++ b/api/routes/measures.py
@@ -2,6 +2,7 @@ from fastapi import APIRouter
 from sqlalchemy import select
 from ..database import SessionLocal
 from ..models import Measure
+from pydantic import BaseModel
 
 router = APIRouter(
     prefix='/measures',
@@ -24,3 +25,16 @@ async def get_measures():
     dc['size'] = len(dc['measures'])
     session.close()
     return dc
+
+class PostMeasure(BaseModel):
+    name: str
+    unit_code: str
+
+@router.post('/')
+async def post_measure(measure: PostMeasure):
+    session = SessionLocal()
+    session.begin()
+    session.add(Measure(name=measure.name, unit_code=measure.unit_code))
+    session.commit()
+    session.close()
+    return {'msg': 'Unidade adicionada com sucesso.'}