Cyberdefense/VulnWhisperer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update cvss_severity calculation

Browse Source
This commit is contained in:
pemontto
2019-05-13 10:01:51 +01:00
parent bf2d2f86cd
commit dfc4403779
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/data

Submodule tests/data updated: 8d3c7de526...1d0e07075e

10
vulnwhisp/vulnwhisp.py
View File

@ -318,12 +318,12 @@ class vulnWhispererBase(object):
# Get a single cvss score derived from cvss3 else cvss2
if not 'cvss' in df:
if 'cvss2' in df:
df.loc[df['cvss2'].notnull(), 'cvss'] = df.loc[df['cvss2'].notnull(), 'cvss2']
df.loc[df['cvss2'].notnull(), 'cvss_severity'] = df.loc[df['cvss2'].notnull(), 'cvss2_severity']
if 'cvss3' in df:
df['cvss'] = df['cvss3'].fillna(df['cvss2'])
df['cvss_severity'] = df['cvss3_severity'].fillna(df['cvss2_severity'])
elif 'cvss2' in df:
df['cvss'] = df['cvss2']
df['cvss_severity'] = df['cvss2_severity']
df.loc[df['cvss3'].notnull(), 'cvss'] = df.loc[df['cvss3'].notnull(), 'cvss3']
df.loc[df['cvss3'].notnull(), 'cvss_severity'] = df.loc[df['cvss3'].notnull(), 'cvss3_severity']
self.logger.debug('Creating Unique Document ID')
df['_unique'] = df.index.values