streamlit-octostar-utils 0.2.10__py3-none-any.whl → 0.2.12__py3-none-any.whl

streamlit_octostar_utils/api_crafter/nifi.py

@@ -23,7 +23,7 @@ from octostar.client import make_client
 from ..core.dict import recursive_update_dict, travel_dict, jsondict_hash
 from ..core.timestamp import now, string_to_datetime
 from .fastapi import DefaultErrorRoute, Route
-from ..ontology.inheritance import is_child_concept as is_child_concept_fn
+from ..ontology.inheritance import is_child_concept as is_child_concept_fn, get_label_keys
 from ..ontology.expand_entities import expand_entities
 
 RELATIONSHIP_ENTITY_NAME = "os_relationship"
@@ -69,6 +69,7 @@ class NifiEntityModel(BaseModel):
         class OntologyInfoModel(BaseModel):
             parents: List[str]
             relationships: List[str]
+            label_keys: List[str]
 
         class ContentsPointerModel(BaseModel):
             location: NifiContentsPointerLocationModel
@@ -806,15 +807,8 @@ class NifiEntity(object):
 
     @property
     def label(self):
-        if not self.context.ontology:
-            return None
-        label_fields = self.context.ontology["concepts"][
-            self.record.get("os_concept") or self.record.get("entity_type")
-        ]["labelKeys"]
-        label_fields = [field for field in label_fields if field]
-        label = " ".join(
-            [(self.record.get(field) or "") for field in label_fields]
-        ).strip()
+        label_keys = self.request["ontology_info"]["label_keys"]
+        label = " ".join([(self.record.get(field) or "") for field in label_keys]).strip()
         if not label:
             label = None
         return label
@@ -923,14 +917,21 @@ class NifiEntity(object):
         now_time = now()
         random_id = str(uuid.uuid4())
         username = self.jwt_data["username"]
+        if entity_type == self.record["entity_type"]:
+            ont_parents = self.request["ontology_info"]["parents"]
+            ont_relationships = self.request["ontology_info"]["relationships"]
+            ont_label_keys = self.request["ontology_info"]["label_keys"]
+        else:
+            ont_parents = self.context.ontology["concepts"][entity_type]["parents"]
+            ont_relationships = self.context.ontology["concepts"][entity_type]["relationships"]
+            ont_label_keys = get_label_keys(entity_type, self.context.ontology)
         child_request = {
             "jwt": self.request["jwt"],
             "ontology_name": self.request["ontology_name"],
             "ontology_info": {
-                "parents": self.context.ontology["concepts"][entity_type]["parents"],
-                "relationships": self.context.ontology["concepts"][entity_type][
-                    "relationships"
-                ],
+                "parents": ont_parents,
+                "relationships": ont_relationships,
+                "label_keys": ont_label_keys,
             },
             "entity_timestamp": None,
             "sync_params": {},

streamlit_octostar_utils/ontology/inheritance.py

@@ -1,2 +1,15 @@
 def is_child_concept(type, parent_type, ontology):
     return type == parent_type or parent_type in ontology["concepts"][type]["parents"]
+
+def get_label_keys(type, ontology):
+    parents = set(ontology["concepts"][type]["parents"])
+    parents.add(type)
+    parents = list(parents)
+    parents.reverse()
+    label_keys = {} # for guaranteed insertion order
+    for parent in parents:
+        for label_key in ontology["concepts"][parent]["labelKeys"]:
+            if not label_key:
+                continue
+            label_keys[label_key] = None
+    return list(label_keys.keys())

{streamlit_octostar_utils-0.2.10.dist-info → streamlit_octostar_utils-0.2.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: streamlit-octostar-utils
-Version: 0.2.10
+Version: 0.2.12
 Summary: 
 License: MIT
 License-File: LICENSE

{streamlit_octostar_utils-0.2.10.dist-info → streamlit_octostar_utils-0.2.12.dist-info}/RECORD

@@ -2,7 +2,7 @@ streamlit_octostar_utils/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-Y
 streamlit_octostar_utils/api_crafter/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 streamlit_octostar_utils/api_crafter/celery.py,sha256=BXOTGN9egdD75qf-PkccLGAoniilB9PZ_NRchFIjWdw,30051
 streamlit_octostar_utils/api_crafter/fastapi.py,sha256=RKQrStPzG1I1pxsPJvGs_DRrnjlMJbVmu9ObMF2LgZ0,14368
-streamlit_octostar_utils/api_crafter/nifi.py,sha256=yFs1HXpSVfWpOC1aJnNahjPofGzZ8fpuqvChloqM4rQ,45541
+streamlit_octostar_utils/api_crafter/nifi.py,sha256=6PSWIFKjv8nzlFGH9IFRI3VrYsISNjDIPyi1RvLJoKk,45810
 streamlit_octostar_utils/api_crafter/parser/__init__.py,sha256=YeYWF6sdQiCFV_RKNW2t9Vs6KJExE2pbXxWTe_DOayY,107
 streamlit_octostar_utils/api_crafter/parser/combine_fields.py,sha256=ddc44xkajw8MU0peAX_263DL7rPXbTKbHUjpOhRgvyU,8790
 streamlit_octostar_utils/api_crafter/parser/entities_parser.py,sha256=zOQoN-p1Gz6ZzxvoX4M1b4Fi3mfmQr5zaNUcp_8gCjw,30016
@@ -28,7 +28,7 @@ streamlit_octostar_utils/octostar/context.py,sha256=TpucK48EbeVy4vDqKd9UULEtr1JO
 streamlit_octostar_utils/octostar/permissions.py,sha256=G5nZQLR-k-5_Xeto4nDTb32828Ga-SHm1mvSB9tz-t4,1565
 streamlit_octostar_utils/ontology/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 streamlit_octostar_utils/ontology/expand_entities.py,sha256=bBt32Dnts3VSzu13QQtPyfYe05IRodD9WfnhNTiBg_w,22749
-streamlit_octostar_utils/ontology/inheritance.py,sha256=oSd6xDAlmI7iYOv3VJ7t8CRN2zK7_Cln26YHS20qAqw,138
+streamlit_octostar_utils/ontology/inheritance.py,sha256=8GA2an1hbHfa6p993tIyfFLrewJHRUIFOw7dmvL8geU,583
 streamlit_octostar_utils/ontology/validation.py,sha256=0cXxEq8vQ63qxn4WianTioTcsmpsg4jEXVyI4R6x1gE,1051
 streamlit_octostar_utils/style/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 streamlit_octostar_utils/style/common.py,sha256=TKfjV9-sIoJChGM7Ewg3uPsz5sMmPxFwmc0o3L4D8Qo,1496
@@ -36,7 +36,7 @@ streamlit_octostar_utils/threading/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzp
 streamlit_octostar_utils/threading/async_task_manager.py,sha256=q7N6YZwUvIYMzkSHmsJNheNVCv93c03H6Hyg9uH8pvk,4747
 streamlit_octostar_utils/threading/session_callback_manager.py,sha256=LvZVP4g6tvKtYmI13f2j1sX_7hm61Groqp5xJine9_k,3973
 streamlit_octostar_utils/threading/session_state_hot_swapper.py,sha256=6eeCQI6A42hp4DmW2NQw2rbeR-k9N8DhfBKQdN_fbLU,811
-streamlit_octostar_utils-0.2.10.dist-info/METADATA,sha256=YKHSxwF_9RwZOr2uzqwhjZA_Q9LWhsMl-GEvanSL9mE,2330
-streamlit_octostar_utils-0.2.10.dist-info/WHEEL,sha256=M5asmiAlL6HEcOq52Yi5mmk9KmTVjY2RDPtO4p9DMrc,88
-streamlit_octostar_utils-0.2.10.dist-info/licenses/LICENSE,sha256=dkwVPyV03fPHHtERnF6RnvRXcll__tud9gWca1RcgnQ,1073
-streamlit_octostar_utils-0.2.10.dist-info/RECORD,,
+streamlit_octostar_utils-0.2.12.dist-info/METADATA,sha256=F_7r20YG6UyKEMKxD4yxdm6vs7xbBeW0hakEMAI6kwg,2330
+streamlit_octostar_utils-0.2.12.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+streamlit_octostar_utils-0.2.12.dist-info/licenses/LICENSE,sha256=dkwVPyV03fPHHtERnF6RnvRXcll__tud9gWca1RcgnQ,1073
+streamlit_octostar_utils-0.2.12.dist-info/RECORD,,

{streamlit_octostar_utils-0.2.10.dist-info → streamlit_octostar_utils-0.2.12.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.2.0
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any