insert_client.py

WSL0809 · Apr 17, 2024 · de95afc · de95afc
1 parent 778c30c
commit de95afc
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/api/insert_client.py b/api/insert_client.py
@@ -56,9 +56,10 @@ def update_client_and_room(db, insert_client_recv: InsertClientRecv):
     )
 
     try:
-        room_status = db.execute(check_room_status_sql, {"room": insert_client_recv.room}).fetchone()[0]
-        if room_status == occupied:
-            raise HTTPException(status_code=400, detail="room occupied")
+        if insert_client_recv.room is not None:
+            room_status = db.execute(check_room_status_sql, {"room": insert_client_recv.room}).fetchone()[0]
+            if room_status == occupied:
+                raise HTTPException(status_code=400, detail="room occupied")
     except SQLAlchemyError as e:
         db.rollback()
         raise HTTPException(status_code=400, detail=f"检查房间{insert_client_recv.room}是否存在,ERROR: {e}")