def main(args):
"""Main function.
The main function parses the command line arguments, reads the input file
and inits the generator.
"""
# Parse arguments
parser = argparse.ArgumentParser(description='Start workload.')
parser.add_argument('--debug', action='store_true',
help="enable debug information")
parser.add_argument('--verbose', action='store_true',
help="makes generator more verbose")
parser.add_argument('--threads', type=int,
default=Options.DEFAULT_THREADS,
help="number of threads to run simultaneously")
parser.add_argument('--test', action='store_true',
help="run a scenario only once")
parser.add_argument('--slimerjs', action='store_true',
help="use slimerjs instead of phantomjs")
parser.add_argument('--screenshot', action='store_true',
help="provide screenshots after each step")
parser.add_argument('--user-agent', type=str, dest='userAgent',
help="provide a custom User-Agent")
parser.add_argument('--max-time', type=int,
default=Options.DEFAULT_MAXTIME, dest='maxTime',
help="provide a maximum runtime")
parser.add_argument('scenario')
args = parser.parse_args()
# Check if scenario exists
if not os.path.isfile(args.scenario):
print "scenario '%s' not found" % args.scenario
return
# Load from file and parse
with open(args.scenario, 'r') as content_file:
content = content_file.read()
scenario = commentjson.loads(content)
# Load in scenario
scenario = Scenario(scenario)
# Parse options
options = Options()
# Which browser are we using
if args.slimerjs:
options.setBrowser('slimerjs')
# Threads option
options.setThreads(args.threads)
# Test option
options.setTest(args.test)
# Output information
options.setVerbose(args.verbose)
# Debug mode
options.setDebug(args.debug)
# Screenshot mode
options.setScreenshot(args.screenshot)
# User agent
options.setUserAgent(args.userAgent)
# Create watchdog thread
options.setMaximumExectionTime(args.maxTime)
# Create new Felt class
felt = Felt(options, scenario)
# Start worker
felt.run()
