add cwe value and value description to presenter

SK-CERT · Feb 25, 2024 · 0f5240d · 0f5240d
1 parent 3f0f0fd
commit 0f5240d
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/src/presenters/presenters/base_presenter.py b/src/presenters/presenters/base_presenter.py
@@ -105,13 +105,15 @@ def __init__(self, report_item, report_types, attribute_map):
             for attribute_group_item_id, attribute_group_item in attribute_group_items.items():
                 attr_type = attribute_map[attribute_group_item_id]
                 attr_key = attr_type.title.lower().replace(" ", "_")
-
-                max_occurrence = attr_type.max_occurrence
-                value_to_add = (
-                    attribute_group_item[0].value
-                    if max_occurrence == 1 and attribute_group_item
-                    else [attribute.value for attribute in attribute_group_item]
-                )
+                if attr_key.startswith("cwe"):
+                    value_to_add = {attribute.value: attribute.value_description for attribute in attribute_group_item}
+                else:
+                    max_occurrence = attr_type.max_occurrence
+                    value_to_add = (
+                        attribute_group_item[0].value
+                        if max_occurrence == 1 and attribute_group_item
+                        else [attribute.value for attribute in attribute_group_item]
+                    )
 
                 how_many_with_the_same_name = len(attribute_groups[attr_key])
                 if how_many_with_the_same_name == 1: