diff --git a/examples/local_example.py b/examples/local_example.py index 39f7b77..0353629 100755 --- a/examples/local_example.py +++ b/examples/local_example.py @@ -11,6 +11,7 @@ Attach the logs to the token ''' +import argparse import logging import os import time @@ -78,14 +79,28 @@ def process_task(self, token): pass +def arg_parser(): + """ + Arguments parser for optional values of the example + returns: argparse object + """ + parser = argparse.ArgumentParser(description="Arguments used in the different classes in the example.") + parser.add_argument("-d", "--design_doc", default="Monitor", type=str, help="Select the designdoc used by the actor class") + parser.add_argument("-V", "--view", default="todo", type=str, help="Select the view used by the actor class") + parser.add_argument("-v", "--verbose", action="store_true", type=bool, help="Set verbose") # add v for optional verbosity in the future + return parser.parse_args() + + def main(): + # parse user arguments + args = arg_parser() # setup connection to db client = CouchDB(url=picasconfig.PICAS_HOST_URL, db=picasconfig.PICAS_DATABASE, username=picasconfig.PICAS_USERNAME, password=picasconfig.PICAS_PASSWORD) print("Connected to the database %s sucessfully. Now starting work..." %(picasconfig.PICAS_DATABASE)) # Create token modifier modifier = BasicTokenModifier() # Create actor - actor = ExampleActor(client, modifier) + actor = ExampleActor(client, modifier, view=args.view, design_doc=args.design_doc) # Start work! actor.run(max_token_time=1800, max_total_time=3600, max_tasks=10, max_scrub=2) diff --git a/examples/multi-core-job.sh b/examples/multi-core-job.sh new file mode 100644 index 0000000..a3fb48d --- /dev/null +++ b/examples/multi-core-job.sh @@ -0,0 +1,5 @@ +#!/bin/bash + +#SBATCH -c 4 + +python local_example.py -d MonitorMultiCore diff --git a/examples/single-core-job.sh b/examples/single-core-job.sh new file mode 100644 index 0000000..9fc915f --- /dev/null +++ b/examples/single-core-job.sh @@ -0,0 +1,5 @@ +#!/bin/bash + +#SBATCH -c 1 + +python local_example.py -d MonitorSingleCore