add room

api/__init__.py

@@ -15,4 +15,5 @@
 from .manage_plan import router as manage_plan_router
 from .allocate_room import router as allocate_room_router
 from .get_plan_by_id import router as get_plan_by_id_router
-from .update_client_status import router as update_client_status_router
+from .update_client_status import router as update_client_status_router
+from .room import router as room_router

api/room.py

@@ -16,6 +16,8 @@ class AddRoomRecv(BaseModel):
     room_number: str
     notes: Union[str, None]
 
+class DeleteRoomRecv(BaseModel):
+    room_number: str
 
 class AddRoomResp(BaseModel):
     msg: str
@@ -36,3 +38,17 @@ async def add_room(room_info: AddRoomRecv, db: Session = Depends(get_db)):
         return AddRoomResp(msg=f"Database error: {str(e)}", success=False)
     except Exception as e:
         return AddRoomResp(msg=f"Unexpected error: {str(e)}", success=False)
+
+
+@router.post("/delete_room")
+@roles_required("admin")
+async def delete_room(room: DeleteRoomRecv, db: Session = Depends(get_db)):
+    try:
+        db.query(Room).filter(Room.room_number == room.room_number).delete()
+        db.commit()
+        return AddRoomResp(msg="Room deleted successfully", success=True)
+    except SQLAlchemyError as e:
+        db.rollback()
+        return AddRoomResp(msg=f"Database error: {str(e)}", success=False)
+    except Exception as e:
+        return AddRoomResp(msg=f"Unexpected error: {str(e)}", success=False)

main.py

@@ -20,7 +20,8 @@
                  check_out_router, get_room_by_room_name_router, get_all_clients_router,
                  fault_registration_router, get_baby_nurse_router, insert_baby_nurse_router,
                  reminder_router, get_client_by_room_number_router, product_router, insert_client_router,
-                 manage_plan_router, allocate_room_router, get_plan_by_id_router, update_client_status_router)
+                 manage_plan_router, allocate_room_router, get_plan_by_id_router, update_client_status_router,
+                 room_router)
 from fastapi.middleware.cors import CORSMiddleware
 
 model.Base.metadata.create_all(bind=engine)
@@ -44,6 +45,7 @@
 app.include_router(allocate_room_router)
 app.include_router(get_plan_by_id_router)
 app.include_router(update_client_status_router)
+app.include_router(room_router)
 origins = [
     "*"
 ]