[client] Pass update to observables

Author: SamuelHassine

pycti/connector/opencti_connector_helper.py

@@ -18,7 +18,6 @@
 from typing import Callable, Dict, List, Optional, Union
 
 import pika
-import stix2.utils
 from filigran_sseclient import SSEClient
 from pika.adapters.asyncio_connection import AsyncioConnection
 from pika.exceptions import NackError, UnroutableError

pycti/entities/opencti_stix_cyber_observable.py

@@ -659,6 +659,7 @@ def create(self, **kwargs):
                     $objectLabel: [String],
                     $objectOrganization: [String],
                     $externalReferences: [String],
+                    $update: Boolean,
                     $AutonomousSystem: AutonomousSystemAddInput,
                     $Directory: DirectoryAddInput,
                     $DomainName: DomainNameAddInput,
@@ -698,6 +699,7 @@ def create(self, **kwargs):
                         createdBy: $createdBy,
                         objectMarking: $objectMarking,
                         objectLabel: $objectLabel,
+                        update: $update,
                         externalReferences: $externalReferences,
                         objectOrganization: $objectOrganization,
                         AutonomousSystem: $AutonomousSystem,

pycti/utils/opencti_stix2.py

@@ -1990,6 +1990,7 @@ def export_entity(
         mode: str = "simple",
         max_marking_definition: Dict = None,
         no_custom_attributes: bool = False,
+        only_entity: bool = False,
     ) -> Dict:
         max_marking_definition_entity = (
             self.opencti.marking_definition.read(id=max_marking_definition)
@@ -2057,6 +2058,7 @@ def export_entity(
         if entity is None:
             API_LOGGER.error("Cannot export entity (not found)")
             return bundle
+        entity_standard_id = entity["standard_id"]
         stix_objects = self.prepare_export(
             self.generate_export(entity, no_custom_attributes),
             mode,
@@ -2065,6 +2067,10 @@ def export_entity(
         )
         if stix_objects is not None:
             bundle["objects"].extend(stix_objects)
+        if only_entity:
+            return [e for e in bundle["objects"] if e.get("id") == entity_standard_id][
+                0
+            ]
         return bundle
 
     def export_list(