streamlit-octostar-utils 0.2.1__tar.gz → 0.2.3__tar.gz

{streamlit_octostar_utils-0.2.1 → streamlit_octostar_utils-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: streamlit-octostar-utils
-Version: 0.2.1
+Version: 0.2.3
 Summary: 
 License: MIT
 Author: Octostar

{streamlit_octostar_utils-0.2.1 → streamlit_octostar_utils-0.2.3}/pyproject.toml

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

{streamlit_octostar_utils-0.2.1 → streamlit_octostar_utils-0.2.3}/streamlit_octostar_utils/api_crafter/nifi.py

@@ -81,6 +81,7 @@ class NifiEntityModel(BaseModel):
         sync_params: dict = Field(default_factory=dict)
         nifi_attributes: dict = Field(default_factory=dict)
         config: dict = Field(default_factory=dict)
+        metrics: dict = Field(default_factory=dict)
         contents_pointer: Optional[ContentsPointerModel] = None
         is_temporary: bool = False
         exception: dict = Field(default_factory=dict)
@@ -843,9 +844,26 @@ class NifiEntity(object):
                 entity_type == type or type in self.request["ontology_info"]["parents"]
             )
 
-    def is_fragmented(self):
+    def is_fragmented(self) -> bool:
         return bool(self.request["config"].get("fragment", {}).get("fragments_stack"))
 
+    def is_root_fragment(self, entity) -> bool:
+        def _is_sub_fragment_recursive(fragment: dict) -> bool:
+            if not isinstance(fragment, dict):
+                return False
+            if all(k in fragment for k in ["index", "count", "identifier"]):
+                return fragment.get("index", 0) != 0
+            for value in fragment.values():
+                if isinstance(value, dict):
+                    if _is_sub_fragment_recursive(value):
+                        return True
+            return False
+        
+        if not self.is_fragmented():
+            return True
+        fragment = entity.request.get("config", {}).get("fragment", {})
+        return not _is_sub_fragment_recursive(fragment)
+
     def to_json(self):
         def _safe_encode(contents):
             return base64.b64encode(contents) if contents else None
@@ -911,13 +929,16 @@ class NifiEntity(object):
                     "relationships"
                 ],
             },
+            "entity_timestamp": None,
             "sync_params": {},
             "nifi_attributes": {},
             "config": deepcopy(self.request["config"]),
-            "fallback_os_workspace": self.request["fallback_os_workspace"],
-            "is_temporary": True,
-            "entity_timestamp": None,
+            "metrics": {},
             "contents_pointer": None,
+            "is_temporary": True,
+            "exception": {},
+            "last_processor_name": None,
+            "fallback_os_workspace": self.request["fallback_os_workspace"],
         }
         child_entity = NifiEntity(
             self.context,