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

Commit 4c23629

Browse files
[client] Add score on Organization (opencti/#9811)
1 parent 5e26d5c commit 4c23629

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pycti/entities/opencti_identity.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@ def __init__(self, opencti):
5151
}
5252
... on Organization {
5353
x_opencti_organization_type
54+
x_opencti_score
5455
}
5556
... on Individual {
5657
x_opencti_firstname
@@ -110,6 +111,7 @@ def __init__(self, opencti):
110111
}
111112
... on Organization {
112113
x_opencti_organization_type
114+
x_opencti_score
113115
}
114116
"""
115117
self.properties_with_files = """
@@ -152,6 +154,7 @@ def __init__(self, opencti):
152154
}
153155
... on Organization {
154156
x_opencti_organization_type
157+
x_opencti_score
155158
}
156159
... on Individual {
157160
x_opencti_firstname
@@ -224,6 +227,7 @@ def __init__(self, opencti):
224227
}
225228
... on Organization {
226229
x_opencti_organization_type
230+
x_opencti_score
227231
}
228232
importFiles {
229233
edges {
@@ -412,6 +416,7 @@ def create(self, **kwargs):
412416
x_opencti_aliases = kwargs.get("x_opencti_aliases", None)
413417
x_opencti_organization_type = kwargs.get("x_opencti_organization_type", None)
414418
x_opencti_reliability = kwargs.get("x_opencti_reliability", None)
419+
x_opencti_score = kwargs.get("x_opencti_score", None)
415420
x_opencti_firstname = kwargs.get("x_opencti_firstname", None)
416421
x_opencti_lastname = kwargs.get("x_opencti_lastname", None)
417422
x_opencti_stix_ids = kwargs.get("x_opencti_stix_ids", None)
@@ -456,6 +461,7 @@ def create(self, **kwargs):
456461
x_opencti_organization_type
457462
)
458463
input_variables["x_opencti_reliability"] = x_opencti_reliability
464+
input_variables["x_opencti_score"] = x_opencti_score
459465
result_data_field = "organizationAdd"
460466
elif type == IdentityTypes.INDIVIDUAL.value:
461467
query = """
@@ -552,6 +558,10 @@ def import_from_stix2(self, **kwargs):
552558
stix_object["x_opencti_reliability"] = (
553559
self.opencti.get_attribute_in_extension("reliability", stix_object)
554560
)
561+
if "x_opencti_score" not in stix_object:
562+
stix_object["x_opencti_score"] = (
563+
self.opencti.get_attribute_in_extension("score", stix_object)
564+
)
555565
if "x_opencti_organization_type" not in stix_object:
556566
stix_object["x_opencti_organization_type"] = (
557567
self.opencti.get_attribute_in_extension(
@@ -630,6 +640,11 @@ def import_from_stix2(self, **kwargs):
630640
if "x_opencti_reliability" in stix_object
631641
else None
632642
),
643+
x_opencti_score=(
644+
stix_object["x_opencti_score"]
645+
if "x_opencti_score" in stix_object
646+
else None
647+
),
633648
x_opencti_firstname=(
634649
stix_object["x_opencti_firstname"]
635650
if "x_opencti_firstname" in stix_object

0 commit comments

Comments
 (0)