def parsing_argument(args):
if not args.client:
if not utils.connection_check('127.0.0.1', 5000):
p = Process(target=flask_run)
p.start()
# wait for flask API start
time.sleep(2)
else:
utils.print_info("Look like Flask API already ran")
# parsing agument
if args.config:
config_path = args.config
options = config.parsing_config(config_path, args)
if options['TARGET_LIST'] != "None":
# check if target list file exist and loop throught the target
targetlist = utils.just_read(options.get('TARGET_LIST'))
if targetlist:
for target in targetlist.splitlines():
options['TARGET'] = target
single_target(options)
utils.print_target(options.get('TARGET'))
else:
single_target(options)
def parsing_report(options):
global headers
mode = options.get('REPORT').lower()
modes = [
'full', 'raw', 'short', 'raw_html', 'list', 'ls', 'path', 'summary',
'sum'
]
if mode not in modes:
report_help()
return None
if not utils.connection_check('127.0.0.1', 5000):
utils.print_info(
"Look like API not turn on, reading from local log instead.")
else:
options['JWT'] = utils.get_jwt(options)
headers['Authorization'] = options['JWT']
# mapping mode
if 'list' in mode or 'ls' in mode:
workspace_list(options)
elif 'path' in mode:
short_report(options)
elif 'short' in mode:
short_report(options)
elif 'full' in mode:
full_report(options)
elif 'raw' in mode:
full_report(options)
elif 'sum' in mode:
summary_report(options)