@@ -565,7 +565,7 @@ def create_marking_definition(self,
565565 'modified' : modified
566566 }
567567 })
568- return result ['data' ]['markingDefinitionAdd' ]
568+ return self . process_multiple_fields ( result ['data' ]['markingDefinitionAdd' ])
569569
570570 # TODO Move to MarkingDefinition
571571 def create_marking_definition_if_not_exists (self ,
@@ -658,7 +658,7 @@ def create_external_reference(self,
658658 'modified' : modified
659659 }
660660 })
661- return result ['data' ]['externalReferenceAdd' ]
661+ return self . process_multiple_fields ( result ['data' ]['externalReferenceAdd' ])
662662
663663 # TODO Move to ExternalReference
664664 def create_external_reference_if_not_exists (self ,
@@ -727,7 +727,7 @@ def create_kill_chain_phase(self,
727727 'modified' : modified
728728 }
729729 })
730- return result ['data' ]['killChainPhaseAdd' ]
730+ return self . process_multiple_fields ( result ['data' ]['killChainPhaseAdd' ])
731731
732732 # TODO Move to KillChainPhase
733733 def create_kill_chain_phase_if_not_exists (self ,
@@ -940,7 +940,7 @@ def create_threat_actor(self,
940940 'modified' : modified
941941 }
942942 })
943- return result ['data' ]['threatActorAdd' ]
943+ return self . process_multiple_fields ( result ['data' ]['threatActorAdd' ])
944944
945945 # TODO Move to ThreatActor
946946 def create_threat_actor_if_not_exists (self ,
@@ -1197,7 +1197,7 @@ def create_campaign(self,
11971197 'modified' : modified
11981198 }
11991199 })
1200- return result ['data' ]['campaignAdd' ]
1200+ return self . process_multiple_fields ( result ['data' ]['campaignAdd' ])
12011201
12021202 # TODO Move to Campaign
12031203 def create_campaign_if_not_exists (self ,
@@ -1438,7 +1438,7 @@ def create_malware(self, name, description, alias=None, id=None, stix_id_key=Non
14381438 'modified' : modified
14391439 }
14401440 })
1441- return result ['data' ]['malwareAdd' ]
1441+ return self . process_multiple_fields ( result ['data' ]['malwareAdd' ])
14421442
14431443 # TODO Move to Malware
14441444 def create_malware_if_not_exists (self , name , description , alias = None , id = None , stix_id_key = None , created = None ,
@@ -1579,7 +1579,7 @@ def create_tool(self, name, description, alias=None, id=None, stix_id_key=None,
15791579 'modified' : modified
15801580 }
15811581 })
1582- return result ['data' ]['toolAdd' ]
1582+ return self . process_multiple_fields ( result ['data' ]['toolAdd' ])
15831583
15841584 # TODO Move to Tool
15851585 def create_tool_if_not_exists (self , name , description , alias = None , id = None , stix_id_key = None , created = None ,
@@ -1720,7 +1720,7 @@ def create_vulnerability(self, name, description, alias=None, id=None, stix_id_k
17201720 'modified' : modified
17211721 }
17221722 })
1723- return result ['data' ]['vulnerabilityAdd' ]
1723+ return self . process_multiple_fields ( result ['data' ]['vulnerabilityAdd' ])
17241724
17251725 # TODO Move to Vulnerability
17261726 def create_vulnerability_if_not_exists (self , name , description , alias = None , id = None , stix_id_key = None , created = None ,
@@ -1916,7 +1916,7 @@ def create_course_of_action(self, name, description, alias=None, id=None, stix_i
19161916 'modified' : modified
19171917 }
19181918 })
1919- return result ['data' ]['courseOfActionAdd' ]
1919+ return self . process_multiple_fields ( result ['data' ]['courseOfActionAdd' ])
19201920
19211921 # TODO Move to CourseOfAction
19221922 def create_course_of_action_if_not_exists (self , name , description , alias = None , id = None , stix_id_key = None ,
0 commit comments