crud: get iot by BT

Signed-off-by: HeshamTB <hishaminv@gmail.com>

sql_app/crud.py

@@ -14,6 +14,9 @@ def get_iot_entity(db: Session, id: int):
 def get_iot_entity_by_description(db: Session, description: str):
     return db.query(models.IotEntity).filter(models.IotEntity.description == description).first()
 
+def get_iot_entity_by_bluetooth_mac(db: Session, bluetooth_mac: str):
+    return db.query(models.IotEntity).filter(models.IotEntity.bluetooth_mac == bluetooth_mac).first()
+
 def get_user_by_email(db: Session, email: str):
     return db.query(models.User).filter(models.User.email == email).first()
 

sql_app/main.py

@@ -59,8 +59,9 @@ def get_current_iot_device(current_device: schemas.IotBluetoothMac = Depends(),
     )
     payload = jwt.decode(token, auth_helper.JWT_SECRET, algorithms=[auth_helper.JWT_ALGO])
     mac_signed = payload.get("bluetooth_mac")
-    if (mac_signed == current_device): return mac_signed
-    else: raise credentials_exception
+    if (mac_signed != current_device): raise credentials_exception
+    device = crud.get_iot_entity_by_bluetooth_mac(db, mac_signed)
+    return device
 
 @app.post("/users/reg", response_model=schemas.User, tags=['Users'])
 def create_user(user: schemas.UserCreate, db: Session = Depends(get_db)):