def get_custom_report(options, grep_string='slack', module=None):
# ws = utils.get_workspace(options=options)
if not module and module is not False:
module = options.get('CURRENT_MODULE', False)
if module:
url = options.get(
'REMOTE_API') + "/api/reports/raw/?module={0}".format(module)
else:
url = options.get('REMOTE_API') + "/api/reports/raw/"
headers = send.osmedeus_headers
headers['Authorization'] = options.get('JWT')
r = send.send_get(url, data=None, headers=headers)
if not r:
return False
reports = utils.resolve_commands(options, r.json().get('reports'))
final_reports = []
if grep_string:
for item in reports:
if grep_string in item.get('note').lower():
final_reports.append(item.get('report_path'))
else:
final_reports = reports
if len(final_reports) == 1:
return final_reports[0]
return final_reports
def gen_commands(self):
self.methods = utils.get_methods(self)
raw_commands = execute.get_cmd(self.options)
self.pre_commands, self.mid_commands, self.post_commands = [], [], []
if raw_commands:
self.commands = utils.resolve_commands(self.options, raw_commands)
for command in self.commands:
# if command.get('pre_run') and command.get('pre_run') != '':
# self.pre_commands.append(command)
if command.get('waiting') == 'last':
self.post_commands.append(command)
elif command.get('waiting') == 'first':
self.pre_commands.append(command)
else:
self.mid_commands.append(command)