def call_scout_suite(args):
args = ['./scout.py'] + args
args.append('aws')
if TestScoutSuiteClass.profile_name:
args.append('--profile')
args.append(TestScoutSuiteClass.profile_name)
# TODO: FIXME this only tests AWS
args.append('--force')
args.append('--debug')
args.append('--no-browser')
if TestScoutSuiteClass.has_run_scout_suite:
args.append('--local')
TestScoutSuiteClass.has_run_scout_suite = True
sys = None
with mock.patch.object(sys, 'argv', args):
return run_from_cli()
#!/usr/bin/env python
import sys
from ScoutSuite.__main__ import run_from_cli
if __name__ == "__main__":
sys.exit(run_from_cli())
yield os.path.join(root, f)
def upload_file(filename):
mime_type = mime.guess_type(filename)
if mime_type[0] is not None:
extra_args = {'ContentType': mime_type[0]}
else:
extra_args = {}
s3.upload_file(Filename=filename,
Bucket=bucket,
Key=prefix + "/" + os.path.relpath(filename, directory),
ExtraArgs=extra_args)
with futures.ThreadPoolExecutor() as executor:
futures.wait(
[
executor.submit(upload_file, filename)
for filename in walk_directory(directory)
],
return_when=futures.FIRST_EXCEPTION,
)
if __name__ == "__main__":
run_from_cli()
output_bucket = os.environ.get("OUTPUT_BUCKET")
prefix = date.today().isoformat()
sys.exit(upload_directory("/opt/scoutsuite-report", output_bucket, prefix))