add context information

Author: bilgehanertan

python_catalyst/client.py

@@ -540,6 +540,7 @@ def create_report_from_member_content_with_references(
         content_id = content.get("id")
         slug = content.get("slug", "")  # noqa: F841
         tlp = content.get("tlp", TLPLevel.CLEAR.value)
+        topics = content.get("topics", [])
         self.converter = self.get_stix_converter(tlp)
 
         if published_on:
@@ -559,6 +560,9 @@ def create_report_from_member_content_with_references(
             labels.append(content["category"])
         if content.get("sub_category") and content["sub_category"].get("name"):
             labels.append(content["sub_category"]["name"])
+        if len(topics) > 0:
+            for topic in topics:
+                labels.append(topic["name"])
 
         report_id = (
             f"report--{str(uuid.uuid5(uuid.NAMESPACE_URL, f'catalyst-{content_id}'))}"
@@ -618,14 +622,15 @@ def create_report_from_member_content_with_references(
                 entity_id = observable.get("id")
                 entity_value = observable.get("value")
                 entity_type = observable.get("type")
-
+                entity_context = observable.get("context", "")
                 if entity_id and entity_value and entity_type:
                     observable_data = {
                         "id": entity_id,
                         "value": entity_value,
                         "type": entity_type,
                         "post_id": content_id,
                         "tlp_marking": content_marking,
+                        "context": entity_context,
                     }
 
                     (

python_catalyst/stix_converter.py

@@ -696,7 +696,6 @@ def create_threat_actor(
         Returns:
             STIX Threat Actor object
         """
-        print(context)
         external_references = []
         if report_reference:
             external_references = [report_reference]
@@ -1097,11 +1096,15 @@ def create_indicator_from_observable(
         marking_ref = tlp_marking.id if tlp_marking else self.tlp_marking.id
 
         created_by_ref = self.get_created_by_ref()
+        description = f"Indicator for {observable_type}: {value}"
+        if "context" in observable_data:
+            ctx = observable_data["context"]
+            description = f"{description}\n\n{ctx}"
 
         return stix2.Indicator(
             id=indicator_id,
             name=indicator_name,
-            description=f"Indicator for {observable_type}: {value}",
+            description=description,
             pattern=pattern,
             pattern_type="stix",
             created_by_ref=created_by_ref,