[client] [Playbook] Modify severity field with playbooks (#7648)

OpenCTI-Platform · Dec 18, 2024 · eafe157 · eafe157
1 parent 711f3a9
commit eafe157
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/pycti/entities/opencti_case_rfi.py b/pycti/entities/opencti_case_rfi.py
@@ -318,6 +318,8 @@ def __init__(self, opencti):
             modified
             name
             description
+            severity
+            priority
             information_types
             objects(all: true) {
                 edges {
@@ -678,6 +680,8 @@ def create(self, **kwargs):
         object_participant = kwargs.get("objectParticipant", None)
         external_references = kwargs.get("externalReferences", None)
         revoked = kwargs.get("revoked", None)
+        severity = kwargs.get("severity", None)
+        priority = kwargs.get("priority", None)
         confidence = kwargs.get("confidence", None)
         lang = kwargs.get("lang", None)
         created = kwargs.get("created", None)
@@ -716,6 +720,8 @@ def create(self, **kwargs):
                         "objects": objects,
                         "externalReferences": external_references,
                         "revoked": revoked,
+                        "severity": severity,
+                        "priority": priority,
                         "confidence": confidence,
                         "lang": lang,
                         "created": created,
@@ -881,6 +887,8 @@ def import_from_stix2(self, **kwargs):
                     else None
                 ),
                 revoked=stix_object["revoked"] if "revoked" in stix_object else None,
+                severity=stix_object["severity"] if "severity" in stix_object else None,
+                priority=stix_object["priority"] if "priority" in stix_object else None,
                 confidence=(
                     stix_object["confidence"] if "confidence" in stix_object else None
                 ),

diff --git a/pycti/entities/opencti_case_rft.py b/pycti/entities/opencti_case_rft.py
@@ -318,6 +318,8 @@ def __init__(self, opencti):
                 modified
                 name
                 description
+                severity
+                priority
                 takedown_types
                 objects(all: true) {
                     edges {
@@ -677,6 +679,8 @@ def create(self, **kwargs):
         object_participant = kwargs.get("objectParticipant", None)
         external_references = kwargs.get("externalReferences", None)
         revoked = kwargs.get("revoked", None)
+        severity = kwargs.get("severity", None)
+        priority = kwargs.get("priority", None)
         confidence = kwargs.get("confidence", None)
         lang = kwargs.get("lang", None)
         created = kwargs.get("created", None)
@@ -715,6 +719,8 @@ def create(self, **kwargs):
                         "objects": objects,
                         "externalReferences": external_references,
                         "revoked": revoked,
+                        "severity": severity,
+                        "priority": priority,
                         "confidence": confidence,
                         "lang": lang,
                         "created": created,
@@ -879,6 +885,8 @@ def import_from_stix2(self, **kwargs):
                     else None
                 ),
                 revoked=stix_object["revoked"] if "revoked" in stix_object else None,
+                severity=stix_object["severity"] if "severity" in stix_object else None,
+                priority=stix_object["priority"] if "priority" in stix_object else None,
                 confidence=(
                     stix_object["confidence"] if "confidence" in stix_object else None
                 ),