[client] Add indicator score at creation (#69)

Author: Samuel Hassine

pycti/entities/opencti_indicator.py

@@ -338,6 +338,7 @@ def create(self, **kwargs):
         pattern_type = kwargs.get("pattern_type", None)
         valid_from = kwargs.get("valid_from", None)
         valid_until = kwargs.get("valid_until", None)
+        score = kwargs.get("score", None)
         id = kwargs.get("id", None)
         stix_id_key = kwargs.get("stix_id_key", None)
         created = kwargs.get("created", None)
@@ -350,6 +351,7 @@ def create(self, **kwargs):
                     entity_type
                     name
                     description
+                    score
                     ... on Indicator {
                         observableRefs {
                             edges {
@@ -385,17 +387,26 @@ def create(self, **kwargs):
         if object_result is not None:
             if update:
                 # name
-                if object_result["name"] != name:
+                if name is not None and object_result["name"] != name:
                     self.opencti.stix_domain_entity.update_field(
                         id=object_result["id"], key="name", value=name
                     )
                     object_result["name"] = name
                 # description
-                if object_result["description"] != description:
+                if (
+                    description is not None
+                    and object_result["description"] != description
+                ):
                     self.opencti.stix_domain_entity.update_field(
                         id=object_result["id"], key="description", value=description
                     )
                     object_result["description"] = description
+                # score
+                if score is not None and object_result["score"] != score:
+                    self.opencti.stix_domain_entity.update_field(
+                        id=object_result["id"], key="score", value=score
+                    )
+                    object_result["score"] = score
             return object_result
         else:
             return self.create_raw(
@@ -406,6 +417,7 @@ def create(self, **kwargs):
                 pattern_type=pattern_type,
                 valid_from=valid_from,
                 valid_until=valid_until,
+                score=score,
                 id=id,
                 stix_id_key=stix_id_key,
                 created=created,
@@ -518,6 +530,8 @@ def to_stix2(self, **kwargs):
             indicator["modified"] = self.opencti.stix2.format_date(entity["modified"])
             if self.opencti.not_empty(entity["alias"]):
                 indicator[CustomProperties.ALIASES] = entity["alias"]
+            if self.opencti.not_empty(entity["score"]):
+                indicator[CustomProperties.SCORE] = entity["score"]
             indicator[CustomProperties.ID] = entity["id"]
             return self.opencti.stix2.prepare_export(
                 entity, indicator, mode, max_marking_definition_entity

requirements.txt

@@ -4,7 +4,7 @@ setuptools==46.1.3
 python-dateutil==2.8.1
 datefinder==0.7.0
 stix2==1.4.0
-pytz==2019.3
+pytz==2020.1
 pika==1.1.0
 stix2-validator==1.1.2
 black==19.10b0

setup.py

@@ -5,7 +5,7 @@
 from setuptools import setup
 from setuptools.command.install import install
 
-VERSION = "3.2.1"
+VERSION = "3.2.2"
 
 with open("README.md", "r") as fh:
     long_description = fh.read()
@@ -56,7 +56,7 @@ def run(self):
         "datefinder==0.7.0",
         "stix2==1.4.0",
         "stix2-validator==1.1.2",
-        "pytz==2019.3",
+        "pytz==2020.1",
         "pika==1.1.0",
         "python-magic==0.4.15;sys.platform=='linux' or sys.platform=='darwin'",
         "python-magic-bin==0.4.14;sys.platform=='win32'",