Hotfix permission batch (#187)

* improved permission receival

* embedding force update

* PR fix potential no results

---------

Co-authored-by: andhreljaKern <[email protected]>

Author: LennartSchmidtKern

business_objects/embedding.py

@@ -772,18 +772,31 @@ def update_embedding_state_failed(
 
 
 def update_embedding_state_waiting(
-    project_id: str, embedding_id: str, with_commit: bool = False
+    project_id: str,
+    embedding_id: str,
+    with_commit: bool = False,
+    force_update: bool = False,
 ) -> None:
     __update_embedding_state(
-        project_id, embedding_id, enums.EmbeddingState.WAITING.value, with_commit
+        project_id,
+        embedding_id,
+        enums.EmbeddingState.WAITING.value,
+        with_commit,
+        force_update,
     )
 
 
 def __update_embedding_state(
-    project_id: str, embedding_id: str, state: str, with_commit=False
+    project_id: str,
+    embedding_id: str,
+    state: str,
+    with_commit: bool = False,
+    force_update: bool = False,
 ) -> None:
     embedding_item = get(project_id, embedding_id)
-    if embedding_item and not embedding_item.state == enums.EmbeddingState.FAILED.value:
+    if embedding_item and (
+        not embedding_item.state == enums.EmbeddingState.FAILED.value or force_update
+    ):
         embedding_item.state = state
         general.flush_or_commit(with_commit)
 

cognition_objects/integration.py

@@ -278,3 +278,23 @@ def get_sharepoint_permissions_by_integration_id(
         ON x.id = s.id
     """
     return session.execute(query).all()
+
+
+def get_distinct_item_ids_for_all_permissions(
+    integration_id: str,
+) -> List[str]:
+    integration_id = prevent_sql_injection(
+        integration_id, isinstance(integration_id, str)
+    )
+    query = f"""SELECT DISTINCT x.object_id
+    FROM (
+        SELECT json_array_elements_text(permissions) permission_id, MAX(object_id::TEXT) object_id
+        FROM integration.sharepoint
+        WHERE integration_id = '{integration_id}'
+        GROUP BY 1
+    ) x;"""
+    results = session.execute(query).all()
+    if not results:
+        return []
+
+    return [row[0] for row in results if row and row[0]]