From eea417a0d9727f1797a7bcc781f18a17a5b2a042 Mon Sep 17 00:00:00 2001 From: pemontto Date: Wed, 17 Apr 2019 16:31:23 +1000 Subject: [PATCH] fix cvss info severity --- vulnwhisp/vulnwhisp.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vulnwhisp/vulnwhisp.py b/vulnwhisp/vulnwhisp.py index b4677c3..e6bc285 100755 --- a/vulnwhisp/vulnwhisp.py +++ b/vulnwhisp/vulnwhisp.py @@ -282,7 +282,7 @@ class vulnWhispererBase(object): # Map CVSS to severity name df.loc[df['cvss'] == '', 'cvss'] = None df['cvss'] = df['cvss'].astype('float') - df.loc[df['cvss'] == 0, 'cvss_severity'] = 'info' + df.loc[df['cvss'].isnull(), 'cvss_severity'] = 'info' df.loc[(df['cvss'] > 0) & (df['cvss'] < 3), 'cvss_severity'] = 'low' df.loc[(df['cvss'] >= 3) & (df['cvss'] < 6), 'cvss_severity'] = 'medium' df.loc[(df['cvss'] >= 6) & (df['cvss'] < 9), 'cvss_severity'] = 'high' @@ -291,9 +291,9 @@ class vulnWhispererBase(object): if 'cvss3' in df: self.logger.info('Normalising CVSS3 severity') # Map CVSS to severity name - df.loc[df['cvss3'] =='', 'cvss3'] = None + df.loc[df['cvss3'] == '', 'cvss3'] = None df['cvss3'] = df['cvss3'].astype('float') - df.loc[df['cvss3'] == 0, 'cvss3_severity'] = 'info' + df.loc[df['cvss3'].isnull(), 'cvss3_severity'] = 'info' df.loc[(df['cvss3'] > 0) & (df['cvss3'] < 3), 'cvss3_severity'] = 'low' df.loc[(df['cvss3'] >= 3) & (df['cvss3'] < 6), 'cvss3_severity'] = 'medium' df.loc[(df['cvss3'] >= 6) & (df['cvss3'] < 9), 'cvss3_severity'] = 'high'