p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Refactored technique report basic data generation methods

This commit is contained in:
VakarisZ 2019-06-18 14:15:13 +03:00
parent 0422cd32db
commit dc2755173d
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
monkey/monkey_island/cc/services/attack/technique_reports/__init__.py
View File

@ -62,12 +62,16 @@ class AttackTechnique(object):
@staticmethod @staticmethod
def get_message_and_status(technique, status): def get_message_and_status(technique, status):
return {'message': technique.get_message_by_status(technique, status), 'status': status.name}
@staticmethod
def get_message_by_status(technique, status):
if status == ScanStatus.UNSCANNED: if status == ScanStatus.UNSCANNED:
return {'message': technique.unscanned_msg, 'status': ScanStatus.UNSCANNED.name} return technique.unscanned_msg
elif status == ScanStatus.SCANNED: elif status == ScanStatus.SCANNED:
return {'message': technique.scanned_msg, 'status': ScanStatus.SCANNED.name} return technique.scanned_msg
else: else:
return {'message': technique.used_msg, 'status': ScanStatus.USED.name} return technique.used_msg
@staticmethod @staticmethod
def technique_title(technique): def technique_title(technique):
@ -87,11 +91,7 @@ class AttackTechnique(object):
data = {} data = {}
status = AttackTechnique.technique_status(technique.tech_id) status = AttackTechnique.technique_status(technique.tech_id)
title = AttackTechnique.technique_title(technique.tech_id) title = AttackTechnique.technique_title(technique.tech_id)
data.update({'status': status.name, 'title': title}) data.update({'status': status.name,
if status == ScanStatus.UNSCANNED: 'title': title,
data.update({'message': technique.unscanned_msg}) 'message': technique.get_message_by_status(technique, status)})
elif status == ScanStatus.SCANNED:
data.update({'message': technique.scanned_msg})
else:
data.update({'message': technique.used_msg})
return data return data