small fixes

This commit is contained in:
VakarisZ 2019-07-08 14:51:57 +03:00
parent cc09aeefa1
commit 2fa45ef5bc
2 changed files with 3 additions and 3 deletions

View File

@ -50,7 +50,7 @@ class AttackTechnique(object):
def technique_status(cls):
"""
Gets the status of a certain attack technique.
:return: ScanStatus Enum object
:return: ScanStatus numeric value
"""
if mongo.db.telemetry.find_one({'telem_category': 'attack',
'data.status': ScanStatus.USED.value,
@ -97,7 +97,7 @@ class AttackTechnique(object):
def get_tech_base_data(cls):
"""
Gathers basic attack technique data into a dict.
:return: dict E.g. {'message': 'Brute force used', 'status': 'Used', 'title': 'T1110 Brute force'}
:return: dict E.g. {'message': 'Brute force used', 'status': 2, 'title': 'T1110 Brute force'}
"""
data = {}
status = cls.technique_status()

View File

@ -11,7 +11,7 @@ class T1107 extends React.Component {
}
static renderDelete(status){
if(status === 2){
if(status === scanStatus.USED){
return <span>Yes</span>
} else {
return <span>No</span>