refactor Post methods in schema files

1 files changed, 1 insertions, 6 deletions

diff --git a/api/routes/stations.py b/api/routes/stations.py
index 0a753fe..c7cdf7c 100644
--- a/api/routes/stations.py
+++ b/api/routes/stations.py
@@ -2,7 +2,7 @@ from fastapi import APIRouter, HTTPException
 from sqlalchemy import select
 from ..database import SessionLocal
 from ..models import Station, SensorDevice, Location
-from pydantic import BaseModel
+from ..schemas.station import PostStation
 from datetime import datetime
 
 router = APIRouter(
@@ -35,11 +35,6 @@ async def get_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()