[client] Improve vocabularies creation and edition

Author: richard-julien

pycti/entities/opencti_vocabulary.py

@@ -144,3 +144,34 @@ def read_or_create_unchecked(self, **kwargs):
             except ValueError:
                 return None
         return vocab
+
+    def update_field(self, **kwargs):
+        id = kwargs.get("id", None)
+        input = kwargs.get("input", None)
+        if id is not None and input is not None:
+            self.opencti.log("info", "Updating Vocabulary {" + id + "}.")
+            query = """
+                        mutation VocabularyEdit($id: ID!, $input: [EditInput!]!) {
+                            vocabularyFieldPatch(id: $id, input: $input) { 
+                                id
+                                standard_id
+                                entity_type
+                            }
+                        }
+                    """
+            result = self.opencti.query(
+                query,
+                {
+                    "id": id,
+                    "input": input,
+                },
+            )
+            return self.opencti.process_multiple_fields(
+                result["data"]["vocabularyFieldPatch"]
+            )
+        else:
+            self.opencti.log(
+                "error",
+                "[opencti_vocabulary] Missing parameters: id and key and value",
+            )
+            return None

pycti/utils/opencti_stix2.py

@@ -1968,8 +1968,8 @@ def import_bundle(
                         stix_id=item["id"],
                         name=item["name"],
                         category=item["category"],
-                        description=item["description"],
-                        aliases=item["aliases"],
+                        description=item["description"] if "description" in item else None,
+                        aliases=item["aliases"] if "aliases" in item else None,
                         x_opencti_stix_ids=stix_ids,
                         update=update,
                     )

pycti/utils/opencti_stix2_update.py

@@ -246,6 +246,8 @@ def update_attribute(self, entity_type, id, input):
             self.opencti.marking_definition.update_field(id=id, input=input)
         elif entity_type == "label":
             self.opencti.label.update_field(id=id, input=input)
+        elif entity_type == "vocabulary":
+            self.opencti.vocabulary.update_field(id=id, input=input)
         elif entity_type == "kill-chain-phase":
             self.opencti.kill_chain_phase.update_field(id=id, input=input)
         elif entity_type == "external-reference":