Added an argument for username and password, which takes precendece over nessus. Fixed #5

This commit is contained in:
Shaun McCullough
2017-11-27 10:02:53 -05:00
parent fedbb18bb2
commit 35b7093762
2 changed files with 19 additions and 4 deletions

View File

@ -16,13 +16,14 @@ import logging
class vulnWhisperer(object):
def __init__(self, config=None, db_name='report_tracker.db', purge=False, verbose=None, debug=False):
def __init__(self, config=None, db_name='report_tracker.db', purge=False, verbose=None, debug=False, username=None, password=None):
self.verbose = verbose
self.nessus_connect = False
self.develop = True
self.purge = purge
if config is not None:
try:
self.config = vwConfig(config_in=config)
@ -31,8 +32,18 @@ class vulnWhisperer(object):
if self.nessus_enabled:
self.nessus_hostname = self.config.get('nessus', 'hostname')
self.nessus_port = self.config.get('nessus', 'port')
self.nessus_username = self.config.get('nessus', 'username')
self.nessus_password = self.config.get('nessus', 'password')
if password:
self.nessus_password = password
else:
self.nessus_password = self.config.get('nessus', 'password')
if username:
self.nessus_username = username
else:
self.nessus_username = self.config.get('nessus', 'username')
self.nessus_writepath = self.config.get('nessus', 'write_path')
self.nessus_dbpath = self.config.get('nessus', 'db_path')
self.nessus_trash = self.config.getbool('nessus', 'trash')