diff --git a/src/MISP_maltego/transforms/common/entities.py b/src/MISP_maltego/transforms/common/entities.py index 4880ed4..e697e04 100644 --- a/src/MISP_maltego/transforms/common/entities.py +++ b/src/MISP_maltego/transforms/common/entities.py @@ -32,6 +32,8 @@ class MISPEvent(Entity): # threat_level = EnumEntityField('type.enum', choices=['Undefined', 'Low', 'Medium', 'High'], display_name='Threat Level') # analysis = EnumEntityField('type.enum', choices=['Initial', 'Ongoing', 'Completed']) # org = StringEntityField('type.str', display_name='Organisation') + count_attributes = IntegerEntityField('count_attributes', display_name="# attributes", matching_rule=MatchingRule.Loose) + count_objects = IntegerEntityField('count_objects', display_name="# objects", matching_rule=MatchingRule.Loose) class MISPObject(Entity): diff --git a/src/MISP_maltego/transforms/common/util.py b/src/MISP_maltego/transforms/common/util.py index 14cc7f2..ec9e758 100644 --- a/src/MISP_maltego/transforms/common/util.py +++ b/src/MISP_maltego/transforms/common/util.py @@ -341,6 +341,8 @@ def event_to_entity(e, link_style=LinkStyle.Normal, link_direction=LinkDirection info=e['Event']['info'], link_style=link_style, link_direction=link_direction, + count_attributes=len(e['Event'].get('Attribute')), + count_objects=len(e['Event'].get('Object')), notes=notes, bookmark=Bookmark.Green)