本文整理汇总了Python中wrapper.Wrapper.error方法的典型用法代码示例。如果您正苦于以下问题:Python Wrapper.error方法的具体用法?Python Wrapper.error怎么用?Python Wrapper.error使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类wrapper.Wrapper
的用法示例。
在下文中一共展示了Wrapper.error方法的8个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: checkArgc
# 需要导入模块: from wrapper import Wrapper [as 别名]
# 或者: from wrapper.Wrapper import error [as 别名]
def checkArgc():
""" Check argument count """
if len(sys.argv) == 1:
Wrapper.error('-H for help and usage')
return
示例2: sendRequest
# 需要导入模块: from wrapper import Wrapper [as 别名]
# 或者: from wrapper.Wrapper import error [as 别名]
def sendRequest(self, method, resource, data='', token='', verify=False):
""" Send a request to nessus based on the specified data """
resource = self.url + resource
headers = { 'X-Cookie': 'token={0}'.format(self.token),
'Content-Type': 'application/json' }
data = json.dumps(data)
if method == 'POST':
r = requests.post(resource, data=data, headers=headers,
verify=verify)
elif method == 'PUT':
r = requests.put(resource, data=data, headers=headers,
verify=verify)
elif method == 'DELETE':
r = requests.delete(resource, data=data, headers=headers,
verify=verify)
else:
r = requests.get(resource, params=data, headers=headers,
verify=verify)
if r.status_code != 200:
e = r.json()
Wrapper.error(e['error'])
if 'download' in resource:
return r.content
# temp dirty fix. TODO: make it better
elif 'session' in resource and method == 'DELETE':
return
else:
return r.json()
return
示例3: checkReportFormat
# 需要导入模块: from wrapper import Wrapper [as 别名]
# 或者: from wrapper.Wrapper import error [as 别名]
def checkReportFormat(rformat):
""" Check if chosen report format exists """
formats = ('html', 'pdf')
if rformat not in formats:
Wrapper.error('Wrong report format: %s' % rformat)
return
示例4: checkArgs
# 需要导入模块: from wrapper import Wrapper [as 别名]
# 或者: from wrapper.Wrapper import error [as 别名]
def checkArgs(args):
""" Check if required arguments are given """
length = len(args)
for a in args:
if a not in sys.argv:
length -= 1
if length is 0:
Wrapper.error('Wrong usage. -H for help.')
return
示例5: getNessusLogin
# 需要导入模块: from wrapper import Wrapper [as 别名]
# 或者: from wrapper.Wrapper import error [as 别名]
def getNessusLogin(self):
""" get Nessus login from cmdline or read from ini config file """
login = {}
if self.opts['nessus_login']:
login['user'] = self.opts['nessus_login'].split(':')[0]
login['pass'] = self.opts['nessus_login'].split(':')[1]
elif self.ini.config['nessus']:
login['user'] = self.ini.config.get('nessus', 'user')
login['pass'] = self.ini.config.get('nessus', 'pass')
else:
Wrapper.error('no login specified')
return login
示例6: checkIniFile
# 需要导入模块: from wrapper import Wrapper [as 别名]
# 或者: from wrapper.Wrapper import error [as 别名]
def checkIniFile(ini):
""" Check ini file configuration """
sections = ini.sections()
if not sections:
Wrapper.error('Cannot read from ini file. Empty or wrong path?')
check = ('nessus', 'mail', 'addressbook')
for s in sections:
if s not in check:
Wrapper.warn('Missing or wrong section: [%s]' % s)
return
示例7: checkActions
# 需要导入模块: from wrapper import Wrapper [as 别名]
# 或者: from wrapper.Wrapper import error [as 别名]
def checkActions(action):
""" Check if chosen action is valid. Print actions if '?' given. """
if action == '?':
Wrapper.msg('Available actions: \n')
Wrapper.msg('report - export and download nessus reports',
verbose=True)
Wrapper.msg('summary - create and download a summary report\n',
verbose=True)
sys.exit(1337)
if action not in ACTIONS:
Wrapper.error('Please select a correct action')
return
示例8: processReports
# 需要导入模块: from wrapper import Wrapper [as 别名]
# 或者: from wrapper.Wrapper import error [as 别名]
def processReports(self, option):
""" Login to nessus, export and download nessus- or summary-reports """
names = []
Wrapper.msg('Login to Nessus on ' + "'" + self.opts['nessus_url'] + "'")
nessus = Nessus(self.opts['nessus_url'])
nessus.login(self.opts['nessus_login']['user'],
self.opts['nessus_login']['pass'])
if not nessus.login:
Wrapper.error('Cannot login in to Nessus')
# get scan lists
Wrapper.msg('Fetching the scan lists')
scan_list = nessus.getScanList()
# export and download nessus reports
if option == 'report':
Wrapper.msg('Exporting and downloading reports')
outdir = Wrapper.makeReportDir(self.opts['outdir'])
for scan in scan_list['scans']:
if scan['name'] in self.opts['addrbook']:
if scan['status'] == 'completed':
scan_details = nessus.getScan(scan['id'])
end_date = datetime.datetime.fromtimestamp(
int(scan_details['info']['scan_end'])
).strftime('%Y-%m-%d')
if end_date == TODAY or end_date == YESTERDAY:
for root, dirs, files in os.walk(
self.opts['outdir'] + '/' + YESTERDAY):
names.append(files)
for root, dirs, files in os.walk(
self.opts['outdir'] + '/' + TODAY):
names.append(files)
names = [i for sublist in names for i in sublist]
if scan['name'] + '.html' not in names:
file_id = nessus.exportScanResult(scan['id'],
self.opts['format'],
self.opts['nessus_chapters'])['file']
# wait before downloading reports if status is
# not OK
if not nessus.exportStatusCheck(scan['id'],
file_id):
time.sleep(int(self.opts['sleep']))
nessus.downloadReport(scan['id'], file_id,
self.opts['format'], scan['name'],
outdir)
# create and download summary
else:
Wrapper.msg('Creating the summary report')
outdir = Wrapper.makeReportDir(self.opts['outdir'])
for scan in scan_list['scans']:
if scan['name'] in self.opts['addrbook']:
if scan['status'] == 'completed':
scan_res = nessus.getScan(scan['id'])
for h in range(len(scan_res['hosts'])):
host_id = scan_res['hosts'][h]['host_id']
res = nessus.getHostDetails(scan['id'], host_id)
for r in res['vulnerabilities']:
# critical, high, medium
if r['severity'] in (4, 3, 2):
with open(os.path.join(outdir, 'summary.txt'), 'a') as f:
f.write('project: ' + scan['name'] +
' host: ' + r['hostname'] +
' issue: ' + r['plugin_name'] + '\n')
# logout of nessus
Wrapper.msg('Logout of nessusd')
nessus.logout()
return