def run_battery(destination=None, experiments=None, experiment_folder=None, subject_id=None, battery_folder=None, port=None, time=30): '''run_battery runs or previews an entire battery locally with the --run tag. If no experiments are provided, all in the folder will be used. :param destination: destination folder for battery. If none provided, tmp directory is used :param experiments: list of experiment tags to add to battery :param experiment_folder: the folder of experiments to deploy the battery from. :param subject_id: subject id to embed into battery. If none, will be randomly generated :param battery_folder: full path to battery folder to use as a template. If none specified, the expfactory-battery repo will be used. :param port: the port number, default will be randomly generated between 8000 and 9999 :param time: total number of minutes for experiments to add to battery ''' print( "Generating custom battery selecting from experiments for maximum of %s minutes, please wait..." % (time)) if destination == None: destination = tempfile.mkdtemp() shutil.rmtree(destination) # Deploy experiment with battery to temporary directory tmpdir = generate_local(battery_dest=destination, subject_id=subject_id, battery_repo=battery_folder, experiment_repo=experiment_folder, experiments=experiments, warning=False, time=time) os.chdir(tmpdir) try: if port == None: # port = choice(range(8000,9999)) port = get_open_port() Handler = SimpleHTTPServer.SimpleHTTPRequestHandler httpd = SocketServer.TCPServer(("", port), Handler) print("Preview experiment at localhost:%s" % port) webbrowser.open("http://localhost:%s" % (port)) httpd.serve_forever() except: print("Stopping web server...") httpd.server_close() shutil.rmtree(tmpdir)
def run_battery(destination=None,experiments=None,experiment_folder=None,subject_id=None,battery_folder=None,port=None,time=30): '''run_battery runs or previews an entire battery locally with the --run tag. If no experiments are provided, all in the folder will be used. :param destination: destination folder for battery. If none provided, tmp directory is used :param experiments: list of experiment tags to add to battery :param experiment_folder: the folder of experiments to deploy the battery from. :param subject_id: subject id to embed into battery. If none, will be randomly generated :param battery_folder: full path to battery folder to use as a template. If none specified, the expfactory-battery repo will be used. :param port: the port number, default will be randomly generated between 8000 and 9999 :param time: total number of minutes for experiments to add to battery ''' print "Generating custom battery selecting from experiments for maximum of %s minutes, please wait..." %(time) if destination == None: destination = tempfile.mkdtemp() shutil.rmtree(destination) # Deploy experiment with battery to temporary directory tmpdir = generate_local(battery_dest=destination, subject_id=subject_id, battery_repo=battery_folder, experiment_repo=experiment_folder, experiments=experiments, warning=False, time=time) os.chdir(tmpdir) try: if port == None: port = choice(range(8000,9999),1)[0] Handler = SimpleHTTPServer.SimpleHTTPRequestHandler httpd = SocketServer.TCPServer(("", port), Handler) print "Preview experiment at localhost:%s" %port webbrowser.open("http://localhost:%s" %(port)) httpd.serve_forever() except: print "Stopping web server..." httpd.server_close() shutil.rmtree(tmpdir)
def main(): parser = argparse.ArgumentParser( description="generate experiments and infrastructure to serve them.") parser.add_argument("--folder", dest='folder', help="full path to single experiment folder (for single experiment run with --run) or folder with many experiments (for battery run with --run)", type=str, default=None) parser.add_argument("--subid", dest='subid', help="subject id to embed in experiments data in the case of a battery run with --run", type=str, default=None) parser.add_argument("--experiments", dest='experiments', help="comma separated list of experiments for a local battery", type=str, default=None) parser.add_argument("--port", dest='port', help="port to preview experiment", type=int, default=None) parser.add_argument("--battery", dest='battery_folder', help="full path to local battery folder to use as template", type=str, default=None) parser.add_argument("--time", dest='time', help="maximum number of minutes for battery to endure, to select experiments", type=int, default=99999) parser.add_argument('--preview', help="preview an experiment locally (development function)", dest='preview', default=False, action='store_true') parser.add_argument('--run', help="run a single experiment/survey or battery locally", dest='run', default=False, action='store_true') parser.add_argument("--survey", dest='survey', help="survey to run for a local assessment", type=str, default=None) parser.add_argument("--game", dest='game', help="game to run for a local assessment", type=str, default=None) parser.add_argument('--validate', dest='validate', help="validate an experiment folder", default=False, action='store_true') parser.add_argument('--psiturk', dest='psiturk', help="to be used with the --generate command, to generate a psiturk battery instead of local folder deployment", default=False, action='store_true') parser.add_argument('--generate', dest='generate', help="generate (and don't run) a battery with --experiments to a --folder", default=False, action='store_true') parser.add_argument("--output", dest='output', help="output folder for --generate command, if a temporary directory is not desired. Must not exist.", type=str, default=None) parser.add_argument('--test', dest='test', help="test an experiment folder with the experiment robot", default=False, action='store_true') try: args = parser.parse_args() except: parser.print_help() sys.exit(0) # Check if the person wants to preview experiment or battery if args.preview == True: preview_experiment(folder=args.folder,battery_folder=args.battery_folder,port=args.port) # Generate a local battery folder (static) elif args.generate == True: if args.experiments != None: # Deploy a psiturk battery folder experiments = args.experiments.split(",") if args.psiturk == True: outdir = generate(battery_dest=args.output, battery_repo=args.battery_folder, experiment_repo=args.folder, experiments=experiments, make_config=True, warning=False) # Deploy a regular battery folder else: outdir = generate_local(battery_dest=args.output, subject_id="expfactory_battery_result", battery_repo=args.battery_folder, experiment_repo=args.folder, experiments=experiments, warning=False, time=args.time) print "Battery generation complete: static files are in %s" %(outdir) else: print "Please specify list of comma separated experiments with --experiments" # Run a local battery elif args.run == True: # Warn the user about using repos for experiments and battery if args.battery_folder == None: print "No battery folder specified. Will pull latest battery from expfactory-battery repo" if args.folder == None: print "No experiments, games, or surveys folder specified. Will pull latest from expfactory-experiments repo" if args.survey != None: survey = args.survey.split(",") if len(survey) > 0: print "Currently only support running one survey, will run first in list." survey = survey[0] run_single(exp_id=survey, repo_type="surveys", source_repo=args.folder, battery_repo=args.battery_folder, port=args.port, subject_id=args.subid) if args.game != None: game = args.game.split(",") if len(game) > 0: print "Currently only support running one game, will run first in list." game = game[0] run_single(exp_id=game, repo_type="games", source_repo=args.folder, battery_repo=args.battery_folder, port=args.port, subject_id=args.subid) if args.experiments != None: experiments = args.experiments.split(",") run_battery(experiments=experiments, experiment_folder=args.folder, subject_id=args.subid, battery_folder=args.battery_folder, port=args.port, time=args.time) else: print "Please specify list of comma separated experiments with --experiments" # Validate a config.json elif args.validate == True: if args.folder == None: folder = os.getcwd() validate(experiment_folder=folder) # If a survey, and if validates, also validate survey.tsv experiment = load_experiment(folder)[0] if experiment["template"] == "survey": print "Validating survey.tsv..." survey_repo = os.path.dirname(folder) validate_surveys(experiment["exp_id"],survey_repo) # Run the experiment robot elif args.test == True: from expfactory.tests import test_experiment test_experiment(folder=args.folder,battery_folder=args.battery_folder,port=args.port) # Otherwise, just open the expfactory interface else: from expfactory.interface import start start(port=args.port)