post station

1 files changed, 25 insertions, 2 deletions

diff --git a/api/routes/stations.py b/api/routes/stations.py
index 00a5fa8..0a753fe 100644
--- a/api/routes/stations.py
+++ b/api/routes/stations.py
@@ -1,7 +1,9 @@
-from fastapi import APIRouter
+from fastapi import APIRouter, HTTPException
 from sqlalchemy import select
 from ..database import SessionLocal
-from ..models import Station
+from ..models import Station, SensorDevice, Location
+from pydantic import BaseModel
+from datetime import datetime
 
 router = APIRouter(
     prefix='/stations',
@@ -32,3 +34,24 @@ async def get_stations():
     dc['size'] = len(dc['stations'])
     session.close()
     return dc
+
+class PostStation(BaseModel):
+    name: str
+    installation_date: int
+    location_id: int
+
+@router.post('/')
+async def post_station(station: PostStation):
+    session = SessionLocal()
+    session.begin()
+    location = session.get(Location, station.location_id)
+    if not location:
+        session.close()
+        raise HTTPException(
+            status_code=404,
+            detail=f"Location {station.location_id} não encontrada."
+        )
+    session.add(Station(name=station.name, installation_date=datetime.fromtimestamp(station.installation_date), location_id=station.location_id, is_active=True))
+    session.commit()
+    session.close()
+    return {'msg': 'Estação adicionada com sucesso.'}