Added all other missing API endpoints.

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

@@ -4,35 +4,11 @@ from sqlalchemy.orm import Session
 
 from . import schemas, sql_models
 
-
-def create_entity(db: Session, entity: schemas.EntityCreate) -> sql_models.Entity:
-    db_entity = sql_models.Entity(**entity.dict())
-    db.add(db_entity)
-    db.commit()
-    db.refresh(db_entity)
-    return db_entity
-
-
-def get_entities(
-    db: Session, skip: int = 0, limit: int = 100
-) -> List[sql_models.Entity]:
-    return db.query(sql_models.Entity).offset(skip).limit(limit).all()
-
-
-def create_consumer(
-    db: Session, consumer: schemas.ConsumerCreate
-) -> sql_models.Consumer:
-    db_consumer = sql_models.Consumer(**consumer.dict())
-    db.add(db_consumer)
-    db.commit()
-    db.refresh(db_consumer)
-    return db_consumer
-
-
-def get_consumers(
-    db: Session, skip: int = 0, limit: int = 100
-) -> List[sql_models.Consumer]:
-    return db.query(sql_models.Consumer).offset(skip).limit(limit).all()
+#########################
+#                       #
+#       Producer        #
+#                       #
+#########################
 
 
 def create_producer(
@@ -51,6 +27,58 @@ def get_producers(
     return db.query(sql_models.Producer).offset(skip).limit(limit).all()
 
 
+def get_producers_by_entity_did(
+    db: Session, entity_did: str, skip: int = 0, limit: int = 100
+) -> List[sql_models.Producer]:
+    return (
+        db.query(sql_models.Producer)
+        .filter(sql_models.Producer.entity_did == entity_did)
+        .offset(skip)
+        .limit(limit)
+        .all()
+    )
+
+
+#########################
+#                       #
+#       Consumer        #
+#                       #
+#########################
+
+
+def create_consumer(
+    db: Session, consumer: schemas.ConsumerCreate
+) -> sql_models.Consumer:
+    db_consumer = sql_models.Consumer(**consumer.dict())
+    db.add(db_consumer)
+    db.commit()
+    db.refresh(db_consumer)
+    return db_consumer
+
+
+def get_consumers(
+    db: Session, skip: int = 0, limit: int = 100
+) -> List[sql_models.Consumer]:
+    return db.query(sql_models.Consumer).offset(skip).limit(limit).all()
+
+
+def get_consumers_by_entity_did(
+    db: Session, entity_did: str, skip: int = 0, limit: int = 100
+) -> List[sql_models.Consumer]:
+    return (
+        db.query(sql_models.Consumer)
+        .filter(sql_models.Consumer.entity_did == entity_did)
+        .offset(skip)
+        .limit(limit)
+        .all()
+    )
+
+
+#########################
+#                       #
+#       REPOSITORY      #
+#                       #
+#########################
 def create_repository(
     db: Session, repository: schemas.RepositoryCreate
 ) -> sql_models.Repository:
@@ -85,3 +113,26 @@ def get_repository_by_did(
         .limit(limit)
         .all()
     )
+
+
+#########################
+#                       #
+#        Entity         #
+#                       #
+#########################
+def create_entity(db: Session, entity: schemas.EntityCreate) -> sql_models.Entity:
+    db_entity = sql_models.Entity(**entity.dict())
+    db.add(db_entity)
+    db.commit()
+    db.refresh(db_entity)
+    return db_entity
+
+
+def get_entities(
+    db: Session, skip: int = 0, limit: int = 100
+) -> List[sql_models.Entity]:
+    return db.query(sql_models.Entity).offset(skip).limit(limit).all()
+
+
+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()