diff --git a/pycti/entities/opencti_report.py b/pycti/entities/opencti_report.py index f13246978..2dd5cf563 100644 --- a/pycti/entities/opencti_report.py +++ b/pycti/entities/opencti_report.py @@ -694,6 +694,8 @@ def create(self, **kwargs): created_by = kwargs.get("createdBy", None) objects = kwargs.get("objects", None) object_marking = kwargs.get("objectMarking", None) + object_assignee = kwargs.get("objectAssignee", None) + object_participant = kwargs.get("objectParticipant", None) object_label = kwargs.get("objectLabel", None) external_references = kwargs.get("externalReferences", None) revoked = kwargs.get("revoked", None) @@ -733,6 +735,8 @@ def create(self, **kwargs): "objectMarking": object_marking, "objectLabel": object_label, "objectOrganization": granted_refs, + "objectAssignee": object_assignee, + "objectParticipant": object_participant, "objects": objects, "externalReferences": external_references, "revoked": revoked, @@ -886,7 +890,16 @@ def import_from_stix2(self, **kwargs): ) if "x_opencti_content" in stix_object: stix_object["content"] = stix_object["x_opencti_content"] - + if "x_opencti_assignee_ids" not in stix_object: + stix_object["x_opencti_assignee_ids"] = ( + self.opencti.get_attribute_in_extension("assignee_ids", stix_object) + ) + if "x_opencti_participant_ids" not in stix_object: + stix_object["x_opencti_participant_ids"] = ( + self.opencti.get_attribute_in_extension( + "participant_ids", stix_object + ) + ) return self.create( stix_id=stix_object["id"], createdBy=( @@ -897,6 +910,16 @@ def import_from_stix2(self, **kwargs): if "object_marking_ids" in extras else None ), + objectAssignee=( + stix_object["x_opencti_assignee_ids"] + if "x_opencti_assignee_ids" in stix_object + else None + ), + objectParticipant=( + stix_object["x_opencti_participant_ids"] + if "x_opencti_participant_ids" in stix_object + else None + ), objectLabel=( extras["object_label_ids"] if "object_label_ids" in extras else None ),