water level settings end points #2

nodeorc_api/crud/disk_management.py

@@ -5,4 +5,4 @@ def get(db: Session):
     # there should always only be one disk management config. Hence retrieve the first.
     dms = db.query(models.DiskManagement)
     if dms.count() > 0:
-        return db.query(models.DiskManagement).first()
+        return dms.first()

nodeorc_api/main.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI, Depends, Request
 from fastapi.middleware.cors import CORSMiddleware
 
-from nodeorc_api.routers import device, video, disk_management
+from nodeorc_api.routers import device, video, disk_management, water_level
 app = FastAPI()
 
 # origins = ["http://localhost:5173"]
@@ -20,6 +20,7 @@
 app.include_router(device.router)
 app.include_router(video.router)
 app.include_router(disk_management.router)
+app.include_router(water_level.router)
 
 @app.get("/")
 async def root():