diff --git a/robyn/argument_parser.py b/robyn/argument_parser.py index 982ee18aa..fbfc89d64 100644 --- a/robyn/argument_parser.py +++ b/robyn/argument_parser.py @@ -1,13 +1,35 @@ import argparse + class ArgumentParser(argparse.ArgumentParser): def __init__(self): - self.parser = argparse.ArgumentParser(description="Robyn, a fast async web framework with a rust runtime.") - self.parser.add_argument('--processes', type=int, default=1, required=False) - self.parser.add_argument('--workers', type=int, default=1, required=False) - self.parser.add_argument('--dev', default=False, type=lambda x: (str(x).lower() == 'true')) + self.parser = argparse.ArgumentParser( + description="Robyn, a fast async web framework with a rust runtime." + ) + self.parser.add_argument( + "--processes", + type=int, + default=1, + required=False, + help="Choose the number of processes. [Default: 1]", + ) + self.parser.add_argument( + "--workers", + type=int, + default=1, + required=False, + help="Choose the number of workers. [Default: 1]", + ) + self.parser.add_argument( + "--dev", + dest="dev", + action="store_true", + default=False, + help="Development mode. It restarts the server based on file changes.", + ) + self.args = self.parser.parse_args() - + def num_processes(self): return self.args.processes @@ -16,7 +38,6 @@ def workers(self): def is_dev(self): _is_dev = self.args.dev - if _is_dev and ( self.num_processes() != 1 or self.workers() != 1 ): + if _is_dev and (self.num_processes() != 1 or self.workers() != 1): raise Exception("--processes and --workers shouldn't be used with --dev") return _is_dev -