minor updates

This commit is contained in:
pemontto
2019-04-11 21:17:24 +10:00
parent 0710b38de3
commit 8f9932e56b
2 changed files with 8 additions and 2 deletions

View File

@ -223,6 +223,9 @@ class NessusAPI(object):
dataframe['cve'] = dataframe['cve'].str.upper()
dataframe['protocol'] = dataframe['protocol'].str.lower()
# Copy asset to IP
dataframe['ip'] = dataframe['asset']
# Map risk to a SEVERITY MAPPING value
self.logger.debug('Mapping risk to severity number')
dataframe['risk_number'] = dataframe['risk'].str.lower()

View File

@ -85,7 +85,8 @@ class qualysVulnScan:
'cvss3_base': 'cvss3',
'ip_status': 'state',
'os': 'operating_system',
'qid': 'plugin_id'
'qid': 'plugin_id',
'title': 'plugin_name'
}
def __init__(
@ -198,7 +199,9 @@ class qualysVulnScan:
.str.rstrip("/nan")
.fillna("")
)
dataframe.drop('cvss_temporal_vector', axis=1, inplace=True)
# Convert Qualys severity to standardised risk number
dataframe['risk_number'] = dataframe['severity'].astype(int)-1