def test_parse_args(self):
args = "--config=./config.py server start --verbose --processes=8".split()
self.assertEqual(parse_args(args), (args[:1], args[1:]))
args = "server start --verbose --processes=8".split()
self.assertEqual(parse_args(args), ([], args))
args = "server --type=http start --verbose --processes=8".split()
self.assertEqual(parse_args(args), ([], args))
def process(self):
"""Process parsing
"""
# Configure self.parser options
self.add_parser_options()
commandor_args, commands = parse_args(self._args)
self._curdir = os.path.abspath(os.path.curdir)
# Commandor parsed options and args (that empty list)
# Because all args separated into commands_args
self._parsed_options, _ = self.parse_args(commandor_args)
res = self.run(self._parsed_options, commands)
# If need run only commandor without subcommands
if res is None:
return None
command, command_args = self.__class__.find_command(commands)
if issubclass(command, Commandor):
# if subcommands not specified, show help and exit
self.parser.print_help()
self.show_commands(commands)
self.exit()
else:
command_instance = command(cur_dir=self._curdir,
args=command_args, commandor_res=res)
return command_instance.process()
def test_commandors(self):
test_self = self
command_options = [Option("-v", "--verbose",
action="store_true",
default=False,
help="Switch verbose mode"),
Option("-p", "--processes",
default=7,
type="int",
dest="processes",
help="Number of processes")]
class Commandor1(Commandor):
"""Custom commandor
"""
test_commandor = False
def run(self, options, args):
if options.config == './config_exit.py':
return False
else:
test_self.assertEqual(options.config, "./config.py")
self.test_commandor = True
return {"param": "value1"}
class Server(Command):
"""Command description
"""
test_server = False
commandor = Commandor1
def run(self, options, args):
test_self.assertFalse(self.test_server)
test_self.assertEqual(self._commandor_res['param'], "value1")
return "server_name"
class Start(Command):
parent = Server
test_start = False
options = command_options
def run(self, verbose, processes):
test_self.assertTrue(verbose)
test_self.assertEqual(processes, 10)
test_self.assertFalse(self.test_start)
test_self.assertEqual(self._commandor_res['param'], "value1")
return "start_name"
parser = OptionParser(
usage="%prog [options] <commands>",
add_help_option=False)
commandor_options = [Option('-c', '--config',
metavar="FILE",
help='Commandor configuration file')]
commandor_args = "--config=./config.py server start --verbose --processes=10".split()
commandor = Commandor1(parser, args=commandor_args, options=commandor_options)
self.assertEqual(parse_args(commandor._args), (commandor_args[:1], commandor_args[1:]))
commandor.add_parser_options()
parsed_options, parsed_args = commandor.parse_args(["--config=./config.py"])
self.assertRaises(InvalidScriptOption, commandor.parse_args, ["--dddd=true"])
self.assertEqual(parsed_options.config, "./config.py")
self.assertEqual(parsed_args, [])
command, args = commandor.__class__.find_command(['sserver'])
self.assertEqual(command, commandor.__class__)
self.assertEqual(args, ['sserver'])
command, args = commandor.__class__.find_command(['server'])
self.assertEqual(command, Server)
self.assertEqual(args, [])
parser = OptionParser(
usage="%prog [options] <commands>",
add_help_option=False)
commandor = Commandor1(parser, args=commandor_args,
options=commandor_options)
self.assertEqual(commandor.process(), "start_name")
# Test commandor run
commandor_args = "--config=./config_exit.py".split()
parser = OptionParser(
usage="%prog [options] <commands>",
add_help_option=False)
commandor = Commandor1(parser, args=commandor_args,
options=commandor_options)
self.assertRaises(SystemExit, commandor.process)
commandor_args = []
parser = OptionParser(
usage="%prog [options] <commands>",
add_help_option=False)
class Commandor2(Commandor):
"""Custom commandor2
"""
def run(self, options, args):
return False
commandor = Commandor2(parser, args=commandor_args,
options=commandor_options)
self.assertRaises(SystemExit, commandor.process)
