Просмотр исходного кода

a space could have multiple records now

chenhaiyang 4 лет назад
Родитель
Сommit
8996988d9a
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      app/crud/space/weight.py

+ 2 - 2
app/crud/space/weight.py

@@ -10,8 +10,8 @@ from app.schemas.sapce_weight import SpaceWeightCreate, SpaceWeightUpdate
 from app.utils.date import get_time_str
 
 
-def get_weight_by_space(db: Session, space_id: str) -> VAVRoomWeight:
-    return db.query(VAVRoomWeight).filter(VAVRoomWeight.space_id == space_id).first()
+def get_weights_by_space(db: Session, space_id: str) -> List[VAVRoomWeight]:
+    return db.query(VAVRoomWeight).filter(VAVRoomWeight.space_id == space_id).all()
 
 
 def get_weights_by_vav(db: Session, vav_id: str) -> List[VAVRoomWeight]: