[client] Implement organization seggregation at creation for individual / systems (#712)

Author: SamuelHassine

pycti/entities/opencti_identity.py

@@ -60,6 +60,11 @@ def __init__(self, opencti):
                 x_opencti_order
                 x_opencti_color
             }
+            objectOrganization {
+                id
+                standard_id
+                name
+            }
             objectLabel {
                 id
                 value
@@ -148,6 +153,11 @@ def __init__(self, opencti):
                 x_opencti_order
                 x_opencti_color
             }
+            objectOrganization {
+                id
+                standard_id
+                name
+            }
             objectLabel {
                 id
                 value
@@ -388,6 +398,7 @@ def create(self, **kwargs):
         x_opencti_firstname = kwargs.get("x_opencti_firstname", None)
         x_opencti_lastname = kwargs.get("x_opencti_lastname", None)
         x_opencti_stix_ids = kwargs.get("x_opencti_stix_ids", None)
+        granted_refs = kwargs.get("objectOrganization", None)
         x_opencti_workflow_id = kwargs.get("x_opencti_workflow_id", None)
         update = kwargs.get("update", False)
 
@@ -440,6 +451,7 @@ def create(self, **kwargs):
                         }
                     }
                 """
+                input_variables["objectOrganization"] = granted_refs
                 input_variables["x_opencti_firstname"] = x_opencti_firstname
                 input_variables["x_opencti_lastname"] = x_opencti_lastname
                 input_variables["x_opencti_reliability"] = x_opencti_reliability
@@ -455,6 +467,7 @@ def create(self, **kwargs):
                         }
                     }
                 """
+                input_variables["objectOrganization"] = granted_refs
                 input_variables["x_opencti_firstname"] = x_opencti_firstname
                 input_variables["x_opencti_lastname"] = x_opencti_lastname
                 input_variables["x_opencti_reliability"] = x_opencti_reliability
@@ -470,6 +483,7 @@ def create(self, **kwargs):
                         }
                     }
                 """
+                input_variables["objectOrganization"] = granted_refs
                 input_variables["type"] = type
                 result_data_field = "identityAdd"
             result = self.opencti.query(
@@ -540,6 +554,10 @@ def import_from_stix2(self, **kwargs):
                 stix_object["x_opencti_stix_ids"] = (
                     self.opencti.get_attribute_in_extension("stix_ids", stix_object)
                 )
+            if "x_opencti_granted_refs" not in stix_object:
+                stix_object["x_opencti_granted_refs"] = (
+                    self.opencti.get_attribute_in_extension("granted_refs", stix_object)
+                )
             if "x_opencti_workflow_id" not in stix_object:
                 stix_object["x_opencti_workflow_id"] = (
                     self.opencti.get_attribute_in_extension("workflow_id", stix_object)
@@ -611,6 +629,11 @@ def import_from_stix2(self, **kwargs):
                     if "x_opencti_stix_ids" in stix_object
                     else None
                 ),
+                objectOrganization=(
+                    stix_object["x_opencti_granted_refs"]
+                    if "x_opencti_granted_refs" in stix_object
+                    else None
+                ),
                 x_opencti_workflow_id=(
                     stix_object["x_opencti_workflow_id"]
                     if "x_opencti_workflow_id" in stix_object

pycti/entities/opencti_location.py

@@ -58,6 +58,11 @@ def __init__(self, opencti):
                 x_opencti_order
                 x_opencti_color
             }
+            objectOrganization {
+                id
+                standard_id
+                name
+            }
             objectLabel {
                 id
                 value
@@ -139,6 +144,11 @@ def __init__(self, opencti):
                 x_opencti_order
                 x_opencti_color
             }
+            objectOrganization {
+                id
+                standard_id
+                name
+            }
             objectLabel {
                 id
                 value