attached

pkgs/clan-cli/clan_cli/webui/sql_crud.py

@@ -134,24 +134,30 @@ def get_entities(
 ) -> List[sql_models.Entity]:
     return db.query(sql_models.Entity).offset(skip).limit(limit).all()
 
+# get attached
+def get_attached_entities(
+    db: Session, skip: int = 0, limit: int = 100
+) -> List[sql_models.Entity]:
+    return db.query(sql_models.Entity).offset(skip).limit(limit).filter(sql_models.Entity.attached == True)
+
 
 def get_entity_by_did(db: Session, did: str) -> Optional[sql_models.Entity]:
     return db.query(sql_models.Entity).filter(sql_models.Entity.did == did).first()
 
 
-def check_health(db: Session, entity_did: str, message: str) -> str:
+# set attached
+    # None if did not found
+    # Returns same entity if setting didnt changed something
+def set_attached_by_entity_did(db: Session, entity_did: str, value: bool) -> bool:
     # ste attached to true
     db_entity = get_entity_by_did(db, entity_did)
     if db_entity is not None:
         # db_entity.attached = Column(True)
-        setattr(db_entity, "attached", True)
+        setattr(db_entity, "attached", value)
         # save changes in db
         db.add(db_entity)
         db.commit()
         db.refresh(db_entity)
-        # check every 5 secounds if status has changed
-        while db_entity.attached:
-            time.sleep(5)
-        return f"{entity_did} message"
+        return db_entity
     else:
-        return f"{entity_did} not found and not attached"
+        return db_entity