[client] Fix import relations

Author: Samuel Hassine

examples/import_stix2_file.py

@@ -3,7 +3,7 @@
 from pycti import OpenCTIApiClient
 
 # Variables
-api_url = 'https://demo.opencti.io'
+api_url = 'https://reference.opencti.io'
 api_token = 'c2d944bb-aea6-4bd6-b3d7-6c10451e2256'
 
 # OpenCTI initialization

pycti/entities/opencti_attack_pattern.py

@@ -177,7 +177,7 @@ def read(self, **kwargs):
             else:
                 return None
         else:
-            self.opencti.log('error', 'Missing parameters: id or filters')
+            self.opencti.log('error', '[opencti_attack_pattern] Missing parameters: id or filters')
             return None
 
     """
@@ -224,7 +224,7 @@ def create_raw(self, **kwargs):
             })
             return self.opencti.process_multiple_fields(result['data']['attackPatternAdd'])
         else:
-            self.opencti.log('error', 'Missing parameters: name and description')
+            self.opencti.log('error', '[opencti_attack_pattern] Missing parameters: name and description')
 
     """
         Create a Attack-Pattern object only if it not exists, update it on request
@@ -322,7 +322,7 @@ def import_from_stix2(self, **kwargs):
                 update=update
             )
         else:
-            self.opencti.log('error', 'Missing parameters: stixObject')
+            self.opencti.log('error', '[opencti_attack_pattern] Missing parameters: stixObject')
 
     """
         Export an Attack-Pattern object in STIX2
@@ -358,4 +358,4 @@ def to_stix2(self, **kwargs):
             attack_pattern[CustomProperties.ID] = entity['id']
             return self.opencti.stix2.prepare_export(entity, attack_pattern, mode, max_marking_definition_entity)
         else:
-            self.opencti.log('error', 'Missing parameters: id or entity')
+            self.opencti.log('error', '[opencti_attack_pattern] Missing parameters: id or entity')

pycti/entities/opencti_report.py

@@ -268,7 +268,7 @@ def create_raw(self, **kwargs):
             })
             return self.opencti.process_multiple_fields(result['data']['reportAdd'])
         else:
-            self.opencti.log('error', 'Missing parameters: name and description and published')
+            self.opencti.log('error', '[opencti_report] Missing parameters: name and description and published')
 
     """
          Create a Report object only if it not exists, update it on request
@@ -384,7 +384,7 @@ def add_stix_entity(self, **kwargs):
             if report is None:
                 report = self.read(id=id)
             if report is None:
-                self.opencti.log('error', 'Cannot add Object Ref, report not found')
+                self.opencti.log('error', '[opencti_report] Cannot add Object Ref, report not found')
                 return False
             refs_ids = []
             for ref in report['objectRefs']:
@@ -416,7 +416,7 @@ def add_stix_entity(self, **kwargs):
                 })
                 return True
         else:
-            self.opencti.log('error', 'Missing parameters: id and entity_id')
+            self.opencti.log('error', '[opencti_report] Missing parameters: id and entity_id')
             return False
 
     """
@@ -458,4 +458,4 @@ def to_stix2(self, **kwargs):
             report[CustomProperties.ID] = entity['id']
             return self.opencti.stix2.prepare_export(entity, report, mode, max_marking_definition_entity)
         else:
-            self.opencti.log('error', 'Missing parameters: id or entity')
+            self.opencti.log('error', '[opencti_report] Missing parameters: id or entity')

pycti/utils/opencti_stix2.py

@@ -188,7 +188,7 @@ def extract_embedded_relationships(self, stix_object, types=None):
                 if object_ref in self.mapping_cache:
                     object_ref_result = self.mapping_cache[object_ref]
                 elif 'relationship' in object_ref:
-                    object_ref_result = self.opencti.stix_relation.read(stix_id_key=object_ref)
+                    object_ref_result = self.opencti.stix_relation.read(id=object_ref)
                 else:
                     object_ref_result = self.opencti.stix_entity.read(id=object_ref)