[client] Add note_types and likelihood property (#310)

Author: RomuDeuxfois

pycti/entities/opencti_note.py

@@ -112,6 +112,8 @@ def __init__(self, opencti):
             attribute_abstract
             content
             authors
+            note_types
+            likelihood
             objects {
                 edges {
                     node {
@@ -425,6 +427,8 @@ def create(self, **kwargs):
         abstract = kwargs.get("abstract", None)
         content = kwargs.get("content", None)
         authors = kwargs.get("authors", None)
+        note_types = kwargs.get("note_types", None)
+        likelihood = kwargs.get("likelihood", None)
         x_opencti_stix_ids = kwargs.get("x_opencti_stix_ids", None)
         granted_refs = kwargs.get("objectOrganization", None)
         update = kwargs.get("update", False)
@@ -460,6 +464,8 @@ def create(self, **kwargs):
                         "attribute_abstract": abstract,
                         "content": content,
                         "authors": authors,
+                        "note_types": note_types,
+                        "likelihood": likelihood,
                         "x_opencti_stix_ids": x_opencti_stix_ids,
                         "update": update,
                     }
@@ -627,6 +633,12 @@ def import_from_stix2(self, **kwargs):
                 if "x_opencti_stix_ids" in stix_object
                 else None,
                 authors=stix_object["authors"] if "authors" in stix_object else None,
+                note_types=stix_object["note_types"]
+                if "note_types" in stix_object
+                else None,
+                likelihood=stix_object["likelihood"]
+                if "likelihood" in stix_object
+                else None,
                 objectOrganization=stix_object["granted_refs"]
                 if "granted_refs" in stix_object
                 else None,

pycti/entities/opencti_stix_core_object.py

@@ -133,6 +133,8 @@ def __init__(self, opencti):
                 attribute_abstract
                 content
                 authors
+                note_types
+                likelihood
             }
             ... on ObservedData {
                 first_observed

pycti/entities/opencti_stix_cyber_observable.py

@@ -2160,6 +2160,8 @@ def notes(self, **kwargs):
                                     attribute_abstract
                                     content
                                     authors
+                                    note_types
+                                    likelihood
                                 }
                             }
                         }

pycti/entities/opencti_stix_domain_object.py

@@ -144,6 +144,8 @@ def __init__(self, opencti, file):
                 attribute_abstract
                 content
                 authors
+                note_types
+                likelihood
                 objects {
                     edges {
                         node {
@@ -1560,6 +1562,8 @@ def notes(self, **kwargs):
                                     attribute_abstract
                                     content
                                     authors
+                                    note_types
+                                    likelihood
                                 }
                             }
                         }

pycti/entities/opencti_stix_object_or_stix_relationship.py

@@ -129,6 +129,8 @@ def __init__(self, opencti):
                 attribute_abstract
                 content
                 authors
+                note_types
+                likelihood
             }
             ... on ObservedData {
                 first_observed

tests/cases/entities.py

@@ -509,6 +509,8 @@ def data(self) -> Dict:
             "content": "You would like to know that",
             "confidence": 50,
             "authors": ["you"],
+            "note_types": ["internal"],
+            "likelihood": 75,
             #    "lang": "en",
         }