Browse Source

a space could have multiple records now

chenhaiyang 4 years ago
parent
commit
8996988d9a
1 changed files with 2 additions and 2 deletions
  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]: