streamlit-octostar-utils 0.4.2.dev20__tar.gz → 0.4.2.dev21__tar.gz

{streamlit_octostar_utils-0.4.2.dev20 → streamlit_octostar_utils-0.4.2.dev21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: streamlit-octostar-utils
-Version: 0.4.2.dev20
+Version: 0.4.2.dev21
 Summary: 
 License: MIT
 License-File: LICENSE

{streamlit_octostar_utils-0.4.2.dev20 → streamlit_octostar_utils-0.4.2.dev21}/pyproject.toml

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

{streamlit_octostar_utils-0.4.2.dev20 → streamlit_octostar_utils-0.4.2.dev21}/streamlit_octostar_utils/api_crafter/nifi.py

@@ -208,6 +208,7 @@ class NifiFragmenter(object):
         if count > 100_000:
             raise ValueError("Cannot have more than 100k entities for fragmentation")
         identifier = str(uuid.uuid4())
+        root_uid = fragments[0].record["os_entity_uid"]
         for i, entity in enumerate(fragments):
             travel_dict(entity.request["nifi_attributes"], fragmenter_keylist.split("."), "w")(
                 {"identifier": identifier, "count": count, "index": i}
@@ -221,7 +222,7 @@ class NifiFragmenter(object):
                 "fragments_stack"
             ]
             travel_dict(entity.request["config"]["fragment"], fragmenter_keylist.split("."), "w")(
-                {"identifier": identifier, "count": count, "index": i}
+                {"identifier": identifier, "count": count, "index": i, "root_uid": root_uid}
             )
 
     def push_defragment_strategy(fragment, defragmenter_config):
@@ -793,6 +794,33 @@ class NifiEntity(object):
         fragment = entity.request.get("config", {}).get("fragment", {})
         return not _is_sub_fragment_recursive(fragment)
 
+    def get_fragment_root_uid(self, fragment_name_or_idx) -> str:
+        fragment_config = self.request.get("config", {}).get("fragment", {})
+        fragments_stack = fragment_config.get("fragments_stack", [])
+        if isinstance(fragment_name_or_idx, int):
+            try:
+                fragment_key = fragments_stack[fragment_name_or_idx]
+            except IndexError:
+                raise IndexError(
+                    f"Fragment stack index {fragment_name_or_idx} out of range "
+                    f"(stack size: {len(fragments_stack)})"
+                )
+        else:
+            fragment_key = fragment_name_or_idx
+            if fragment_key not in fragments_stack:
+                raise KeyError(
+                    f"Fragment '{fragment_key}' not found in "
+                    f"fragments_stack: {fragments_stack}"
+                )
+        fragment_data = travel_dict(
+            fragment_config, fragment_key.split("."), mode="r"
+        )
+        if not isinstance(fragment_data, dict) or "root_uid" not in fragment_data:
+            raise KeyError(
+                f"No root_uid found in fragment config for '{fragment_key}'"
+            )
+        return fragment_data["root_uid"]
+
     def to_json(self):
         if self.drop_on_output:
             return