Skip to content
This repository was archived by the owner on Dec 5, 2025. It is now read-only.

Commit c3aa217

Browse files
author
Samuel Hassine
committed
[client] Be able to pass tags at the creation of entities (#80)
1 parent 7f7f302 commit c3aa217

15 files changed

+117
-33
lines changed

‎pycti/entities/opencti_attack_pattern.py‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -257,6 +257,7 @@ def create_raw(self, **kwargs):
257257
modified = kwargs.get("modified", None)
258258
created_by_ref = kwargs.get("createdByRef", None)
259259
marking_definitions = kwargs.get("markingDefinitions", None)
260+
tags = kwargs.get("tags", None)
260261
kill_chain_phases = kwargs.get("killChainPhases", None)
261262

262263
if name is not None and description is not None:
@@ -288,6 +289,7 @@ def create_raw(self, **kwargs):
288289
"modified": modified,
289290
"createdByRef": created_by_ref,
290291
"markingDefinitions": marking_definitions,
292+
"tags": tags,
291293
"killChainPhases": kill_chain_phases,
292294
}
293295
},
@@ -321,6 +323,7 @@ def create(self, **kwargs):
321323
modified = kwargs.get("modified", None)
322324
created_by_ref = kwargs.get("createdByRef", None)
323325
marking_definitions = kwargs.get("markingDefinitions", None)
326+
tags = kwargs.get("tags", None)
324327
kill_chain_phases = kwargs.get("killChainPhases", None)
325328
update = kwargs.get("update", False)
326329
custom_attributes = """
@@ -413,6 +416,7 @@ def create(self, **kwargs):
413416
modified=modified,
414417
createdByRef=created_by_ref,
415418
markingDefinitions=marking_definitions,
419+
tags=tags,
416420
killChainPhases=kill_chain_phases,
417421
)
418422

@@ -469,6 +473,9 @@ def import_from_stix2(self, **kwargs):
469473
markingDefinitions=extras["marking_definitions_ids"]
470474
if "marking_definitions_ids" in extras
471475
else [],
476+
tags=extras["tags_ids"]
477+
if "tags_ids" in extras
478+
else [],
472479
killChainPhases=extras["kill_chain_phases_ids"]
473480
if "kill_chain_phases_ids" in extras
474481
else [],

‎pycti/entities/opencti_campaign.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -218,6 +218,7 @@ def create_raw(self, **kwargs):
218218
modified = kwargs.get("modified", None)
219219
created_by_ref = kwargs.get("createdByRef", None)
220220
marking_definitions = kwargs.get("markingDefinitions", None)
221+
tags = kwargs.get("tags", None)
221222

222223
if name is not None and description is not None:
223224
self.opencti.log("info", "Creating Campaign {" + name + "}.")
@@ -248,6 +249,7 @@ def create_raw(self, **kwargs):
248249
"modified": modified,
249250
"createdByRef": created_by_ref,
250251
"markingDefinitions": marking_definitions,
252+
"tags": tags
251253
}
252254
},
253255
)
@@ -277,6 +279,7 @@ def create(self, **kwargs):
277279
modified = kwargs.get("modified", None)
278280
created_by_ref = kwargs.get("createdByRef", None)
279281
marking_definitions = kwargs.get("markingDefinitions", None)
282+
tags = kwargs.get("tags", None)
280283
update = kwargs.get("update", False)
281284
custom_attributes = """
282285
id
@@ -355,6 +358,7 @@ def create(self, **kwargs):
355358
modified=modified,
356359
createdByRef=created_by_ref,
357360
markingDefinitions=marking_definitions,
361+
tags=tags
358362
)
359363

360364
"""

‎pycti/entities/opencti_course_of_action.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -215,6 +215,7 @@ def create_raw(self, **kwargs):
215215
modified = kwargs.get("modified", None)
216216
created_by_ref = kwargs.get("createdByRef", None)
217217
marking_definitions = kwargs.get("markingDefinitions", None)
218+
tags = kwargs.get("tags", None)
218219

219220
if name is not None and description is not None:
220221
self.opencti.log("info", "Creating Course Of Action {" + name + "}.")
@@ -242,6 +243,7 @@ def create_raw(self, **kwargs):
242243
"modified": modified,
243244
"createdByRef": created_by_ref,
244245
"markingDefinitions": marking_definitions,
246+
"tags": tags
245247
}
246248
},
247249
)
@@ -271,6 +273,7 @@ def create(self, **kwargs):
271273
modified = kwargs.get("modified", None)
272274
created_by_ref = kwargs.get("createdByRef", None)
273275
marking_definitions = kwargs.get("markingDefinitions", None)
276+
tags = kwargs.get("tags", None)
274277
update = kwargs.get("update", False)
275278
custom_attributes = """
276279
id
@@ -323,6 +326,7 @@ def create(self, **kwargs):
323326
modified=modified,
324327
createdByRef=created_by_ref,
325328
markingDefinitions=marking_definitions,
329+
tags=tags
326330
)
327331

328332
"""

‎pycti/entities/opencti_identity.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -218,6 +218,7 @@ def create_raw(self, **kwargs):
218218
modified = kwargs.get("modified", None)
219219
created_by_ref = kwargs.get("createdByRef", None)
220220
marking_definitions = kwargs.get("markingDefinitions", None)
221+
tags = kwargs.get("tags", None)
221222

222223
if name is not None and description is not None:
223224
self.opencti.log("info", "Creating Identity {" + name + "}.")
@@ -246,6 +247,7 @@ def create_raw(self, **kwargs):
246247
"modified": modified,
247248
"createdByRef": created_by_ref,
248249
"markingDefinitions": marking_definitions,
250+
"tags": tags
249251
}
250252
},
251253
)
@@ -271,6 +273,7 @@ def create(self, **kwargs):
271273
modified = kwargs.get("modified", None)
272274
created_by_ref = kwargs.get("createdByRef", None)
273275
marking_definitions = kwargs.get("markingDefinitions", None)
276+
tags = kwargs.get("tags", None)
274277
update = kwargs.get("update", False)
275278
custom_attributes = """
276279
id
@@ -324,6 +327,7 @@ def create(self, **kwargs):
324327
modified=modified,
325328
createdByRef=created_by_ref,
326329
markingDefinitions=marking_definitions,
330+
tags=tags
327331
)
328332

329333
"""

‎pycti/entities/opencti_incident.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -228,6 +228,7 @@ def create_raw(self, **kwargs):
228228
modified = kwargs.get("modified", None)
229229
created_by_ref = kwargs.get("createdByRef", None)
230230
marking_definitions = kwargs.get("markingDefinitions", None)
231+
tags = kwargs.get("tags", None)
231232

232233
if name is not None and description is not None:
233234
self.opencti.log("info", "Creating Incident {" + name + "}.")
@@ -258,6 +259,7 @@ def create_raw(self, **kwargs):
258259
"modified": modified,
259260
"createdByRef": created_by_ref,
260261
"markingDefinitions": marking_definitions,
262+
"tags": tags
261263
}
262264
},
263265
)
@@ -285,6 +287,7 @@ def create(self, **kwargs):
285287
modified = kwargs.get("modified", None)
286288
created_by_ref = kwargs.get("createdByRef", None)
287289
marking_definitions = kwargs.get("markingDefinitions", None)
290+
tags = kwargs.get("tags", None)
288291
update = kwargs.get("update", False)
289292
custom_attributes = """
290293
id
@@ -363,6 +366,7 @@ def create(self, **kwargs):
363366
modified=modified,
364367
createdByRef=created_by_ref,
365368
markingDefinitions=marking_definitions,
369+
tags=tags
366370
)
367371

368372
"""

‎pycti/entities/opencti_indicator.py‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -275,6 +275,8 @@ def create_raw(self, **kwargs):
275275
modified = kwargs.get("modified", None)
276276
created_by_ref = kwargs.get("createdByRef", None)
277277
marking_definitions = kwargs.get("markingDefinitions", None)
278+
tags = kwargs.get("tags", None)
279+
kill_chain_phases = kwargs.get("killChainPhases", None)
278280

279281
if (
280282
name is not None
@@ -313,6 +315,8 @@ def create_raw(self, **kwargs):
313315
"modified": modified,
314316
"createdByRef": created_by_ref,
315317
"markingDefinitions": marking_definitions,
318+
"tags": tags,
319+
"killChainPhases": kill_chain_phases,
316320
}
317321
},
318322
)
@@ -345,6 +349,8 @@ def create(self, **kwargs):
345349
modified = kwargs.get("modified", None)
346350
created_by_ref = kwargs.get("createdByRef", None)
347351
marking_definitions = kwargs.get("markingDefinitions", None)
352+
tags = kwargs.get("tags", None)
353+
kill_chain_phases = kwargs.get("killChainPhases", None)
348354
update = kwargs.get("update", False)
349355
custom_attributes = """
350356
id
@@ -424,6 +430,8 @@ def create(self, **kwargs):
424430
modified=modified,
425431
createdByRef=created_by_ref,
426432
markingDefinitions=marking_definitions,
433+
tags=tags,
434+
killChainPhases=kill_chain_phases
427435
)
428436

429437
"""

‎pycti/entities/opencti_intrusion_set.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,7 @@ def create_raw(self, **kwargs):
227227
modified = kwargs.get("modified", None)
228228
created_by_ref = kwargs.get("createdByRef", None)
229229
marking_definitions = kwargs.get("markingDefinitions", None)
230+
tags = kwargs.get("tags", None)
230231

231232
if name is not None and description is not None:
232233
self.opencti.log("info", "Creating Intrusion-Set {" + name + "}.")
@@ -261,6 +262,7 @@ def create_raw(self, **kwargs):
261262
"modified": modified,
262263
"createdByRef": created_by_ref,
263264
"markingDefinitions": marking_definitions,
265+
"tags": tags
264266
}
265267
},
266268
)
@@ -297,6 +299,7 @@ def create(self, **kwargs):
297299
modified = kwargs.get("modified", None)
298300
created_by_ref = kwargs.get("createdByRef", None)
299301
marking_definitions = kwargs.get("markingDefinitions", None)
302+
tags = kwargs.get("tags", None)
300303
update = kwargs.get("update", False)
301304
custom_attributes = """
302305
id
@@ -427,6 +430,7 @@ def create(self, **kwargs):
427430
modified=modified,
428431
createdByRef=created_by_ref,
429432
markingDefinitions=marking_definitions,
433+
tags=tags
430434
)
431435

432436
"""

‎pycti/entities/opencti_malware.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -229,6 +229,7 @@ def create_raw(self, **kwargs):
229229
modified = kwargs.get("modified", None)
230230
created_by_ref = kwargs.get("createdByRef", None)
231231
marking_definitions = kwargs.get("markingDefinitions", None)
232+
tags = kwargs.get("tags", None)
232233

233234
if name is not None and description is not None:
234235
self.opencti.log("info", "Creating Malware {" + name + "}.")
@@ -256,6 +257,7 @@ def create_raw(self, **kwargs):
256257
"modified": modified,
257258
"createdByRef": created_by_ref,
258259
"markingDefinitions": marking_definitions,
260+
"tags": tags
259261
}
260262
},
261263
)
@@ -282,6 +284,7 @@ def create(self, **kwargs):
282284
modified = kwargs.get("modified", None)
283285
created_by_ref = kwargs.get("createdByRef", None)
284286
marking_definitions = kwargs.get("markingDefinitions", None)
287+
tags = kwargs.get("tags", None)
285288
update = kwargs.get("update", False)
286289
custom_attributes = """
287290
id
@@ -334,6 +337,7 @@ def create(self, **kwargs):
334337
modified=modified,
335338
createdByRef=created_by_ref,
336339
markingDefinitions=marking_definitions,
340+
tags=tags
337341
)
338342

339343
"""

‎pycti/entities/opencti_report.py‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -340,6 +340,7 @@ def create_raw(self, **kwargs):
340340
modified = kwargs.get("modified", None)
341341
created_by_ref = kwargs.get("createdByRef", None)
342342
marking_definitions = kwargs.get("markingDefinitions", None)
343+
tags = kwargs.get("tags", None)
343344

344345
if (
345346
name is not None
@@ -376,6 +377,7 @@ def create_raw(self, **kwargs):
376377
"modified": modified,
377378
"createdByRef": created_by_ref,
378379
"markingDefinitions": marking_definitions,
380+
"tags": tags
379381
}
380382
},
381383
)
@@ -410,6 +412,7 @@ def create(self, **kwargs):
410412
modified = kwargs.get("modified", None)
411413
created_by_ref = kwargs.get("createdByRef", None)
412414
marking_definitions = kwargs.get("markingDefinitions", None)
415+
tags = kwargs.get("tags", None)
413416
update = kwargs.get("update", False)
414417
custom_attributes = """
415418
id
@@ -482,6 +485,7 @@ def create(self, **kwargs):
482485
modified=modified,
483486
createdByRef=created_by_ref,
484487
markingDefinitions=marking_definitions,
488+
tags=tags
485489
)
486490
if external_reference_id is not None:
487491
self.opencti.stix_entity.add_external_reference(
@@ -648,6 +652,9 @@ def import_from_stix2(self, **kwargs):
648652
markingDefinitions=extras["marking_definitions_ids"]
649653
if "marking_definitions_ids" in extras
650654
else [],
655+
tags=extras["tags_ids"]
656+
if "tags_ids" in extras
657+
else [],
651658
update=update,
652659
)
653660
else:

‎pycti/entities/opencti_stix_observable.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -251,6 +251,7 @@ def create_raw(self, **kwargs):
251251
stix_id_key = kwargs.get("stix_id_key", None)
252252
created_by_ref = kwargs.get("createdByRef", None)
253253
marking_definitions = kwargs.get("markingDefinitions", None)
254+
tags = kwargs.get("tags", None)
254255
create_indicator = kwargs.get("createIndicator", False)
255256

256257
if type is not None and observable_value is not None:
@@ -284,6 +285,7 @@ def create_raw(self, **kwargs):
284285
"stix_id_key": stix_id_key,
285286
"createdByRef": created_by_ref,
286287
"markingDefinitions": marking_definitions,
288+
"tags": tags,
287289
"createIndicator": create_indicator,
288290
}
289291
},
@@ -309,6 +311,7 @@ def create(self, **kwargs):
309311
stix_id_key = kwargs.get("stix_id_key", None)
310312
created_by_ref = kwargs.get("createdByRef", None)
311313
marking_definitions = kwargs.get("markingDefinitions", None)
314+
tags = kwargs.get("tags", None)
312315
create_indicator = kwargs.get("createIndicator", False)
313316
update = kwargs.get("update", False)
314317
custom_attributes = """
@@ -340,6 +343,7 @@ def create(self, **kwargs):
340343
stix_id_key=stix_id_key,
341344
createdByRef=created_by_ref,
342345
markingDefinitions=marking_definitions,
346+
tags=tags,
343347
createIndicator=create_indicator,
344348
)
345349

0 commit comments

Comments
 (0)