streamlit-octostar-utils 0.6.4__tar.gz → 0.6.5.dev2__tar.gz

{streamlit_octostar_utils-0.6.4 → streamlit_octostar_utils-0.6.5.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: streamlit-octostar-utils
-Version: 0.6.4
+Version: 0.6.5.dev2
 Summary: 
 License: MIT
 License-File: LICENSE

{streamlit_octostar_utils-0.6.4 → streamlit_octostar_utils-0.6.5.dev2}/pyproject.toml

@@ -5,7 +5,7 @@ include = '\.pyi?$'
 
 [tool.poetry]
 name = "streamlit-octostar-utils"
-version = "0.6.4"
+version = "0.6.5-dev.2"
 description = ""
 license = "MIT"
 authors = ["Octostar"]

{streamlit_octostar_utils-0.6.4 → streamlit_octostar_utils-0.6.5.dev2}/streamlit_octostar_utils/api_crafter/nifi.py

@@ -668,6 +668,7 @@ class NifiContextManager(object):
         self.out_entities = None
         self.nonlazy_sync_ids = set()
         self.lazy_sync = lazy_sync
+        self.sync_allowed_flags = None
         self.client, self.ontology_name = self.get_client(json_data)
         self._ontology = None
         self._input_loader = input_loader
@@ -1068,17 +1069,34 @@ class NifiContextManager(object):
             if statuses:
                 update_processing_status.sync(statuses, client=self.client)
 
-    def sync_entities(self):
-        if not self.lazy_sync:
-            entities = self.out_entities
-        else:
-            entities = [e for e in self.out_entities if e.record["entity_id"] in self.nonlazy_sync_ids]
+    def sync_entities(self, allowed_flags=None, entities=None):
+        """Run sync for ``entities`` (defaults to ``self.out_entities`` honoring
+        ``lazy_sync``/``nonlazy_sync_ids``). Passing ``entities`` explicitly
+        bypasses the ``out_entities`` filter — useful for routes that want to
+        trigger a sync from inside their task body (i.e. before ``send_output``
+        has populated ``out_entities``) and catch failures locally.
+        """
+        effective_flags = allowed_flags if allowed_flags is not None else self.sync_allowed_flags
+        if entities is None:
+            if not self.lazy_sync:
+                entities = self.out_entities
+            else:
+                entities = [e for e in self.out_entities if e.record["entity_id"] in self.nonlazy_sync_ids]
         if not entities:
             return
 
-        attachables_to_write = self._find_attachables_to_write(entities)
-        entities_to_upsert = self._find_entities_to_upsert(entities)
-        fetch_rel_entities, fetch_concept_rels = self._find_relationships_to_fetch(entities)
+        run_writes = effective_flags is None or NifiContextManager.SyncFlag.WRITE_CONTENTS in effective_flags
+        run_upserts = effective_flags is None or (
+            NifiContextManager.SyncFlag.UPSERT_ENTITY_ALL in effective_flags
+            or NifiContextManager.SyncFlag.UPSERT_ENTITY_SPECIFIC_FIELDS in effective_flags
+        )
+        run_fetch = effective_flags is None or NifiContextManager.SyncFlag.FETCH_RELATIONSHIPS in effective_flags
+
+        attachables_to_write = self._find_attachables_to_write(entities) if run_writes else []
+        entities_to_upsert = self._find_entities_to_upsert(entities) if run_upserts else []
+        fetch_rel_entities, fetch_concept_rels = (
+            self._find_relationships_to_fetch(entities) if run_fetch else ({}, {})
+        )
 
         all_entities_to_modify = attachables_to_write + [e[0] for e in entities_to_upsert]
         with self.reindex_lock(all_entities_to_modify) as locked:
@@ -1091,8 +1109,14 @@ class NifiContextManager(object):
         for entity in all_entities_to_modify:
             entity.request["is_temporary"] = False
 
-        for entity in entities:
-            entity.sync_params = {}
+        if effective_flags is None:
+            for entity in entities:
+                entity.sync_params = {}
+        else:
+            for entity in entities:
+                entity.sync_params = {
+                    k: v for k, v in entity.sync_params.items() if k not in effective_flags
+                }
 
     def _find_attachables_to_write(self, entities) -> list:
         attachables_to_write = []