def main_with_cmds(prog, desc, cmds, testfn=None):
parser = ArgumentParser(prog=prog, description=desc)
parser.add_argument('--verbose',
'-v',
action='store_true',
help='Enable verbose output')
subparsers = parser.add_subparsers(title='subcommands', dest='parser_name')
# Parser for the test command.
if testfn:
parser_test = subparsers.add_parser('test', help='Run tests')
# Parser for the sub-commands.
for cmd_name, cmd_info in cmds.items():
cmd_parser = subparsers.add_parser(
cmd_name, help='Execute the command: {}'.format(cmd_name))
if isinstance(cmd_info, dict) and 'args' in cmd_info:
args = OrderedDict()
for arg_info in cmd_info['args']:
name = arg_info.pop('name')
args[name] = arg_info
for arg_name, arg_params in args.items():
cmd_parser.add_argument(arg_name, **arg_params)
# Parse arguments.
args = parser.parse_args(sys.argv[1:])
if args.verbose:
enable_verbose()
# Call the right sub-command handler.
status = None
if args.parser_name is None:
print_error('You need to specify the command (see --help)')
elif args.parser_name == 'test':
status = testfn()
else:
for cmd_name, cmd_info in cmds.items():
if args.parser_name == cmd_name:
if isinstance(cmd_info, dict) and 'handler' in cmd_info:
status = cmd_info['handler'](args)
elif hasattr(cmd_info, '__call__'):
status = cmd_info(args)
else:
print_error('Invalid informations for command: {}'.format(
cmd_name))
status = -999
# Exit from the program with the status-code relative to the command exit
# status.
if status is None:
exit(0)
else:
exit(status)
def main_with_cmds(prog, desc, cmds, testfn=None):
parser = ArgumentParser(prog=prog, description=desc)
parser.add_argument("--verbose", "-v", action="store_true", help="Enable verbose output")
subparsers = parser.add_subparsers(title="subcommands", dest="parser_name")
# Parser for the test command.
if testfn:
parser_test = subparsers.add_parser("test", help="Run tests")
# Parser for the sub-commands.
for cmd_name, cmd_info in cmds.items():
cmd_parser = subparsers.add_parser(cmd_name, help="Execute the command: {}".format(cmd_name))
if isinstance(cmd_info, dict) and "args" in cmd_info:
args = OrderedDict()
for arg_info in cmd_info["args"]:
name = arg_info.pop("name")
args[name] = arg_info
for arg_name, arg_params in args.items():
cmd_parser.add_argument(arg_name, **arg_params)
# Parse arguments.
args = parser.parse_args(sys.argv[1:])
if args.verbose:
enable_verbose()
# Call the right sub-command handler.
status = None
if args.parser_name is None:
print_error("You need to specify the command (see --help)")
elif args.parser_name == "test":
status = testfn()
else:
for cmd_name, cmd_info in cmds.items():
if args.parser_name == cmd_name:
if isinstance(cmd_info, dict) and "handler" in cmd_info:
status = cmd_info["handler"](args)
elif hasattr(cmd_info, "__call__"):
status = cmd_info(args)
else:
print_error("Invalid informations for command: {}".format(cmd_name))
status = -999
# Exit from the program with the status-code relative to the command exit
# status.
if status is None:
exit(0)
else:
exit(status)
class TestIRCClient(TestCase):
def setUp(self):
self._irc_host = 'irc.freenode.net'
self._loop = get_event_loop()
def test_connect(self):
try:
client = IRCClient.connect(host=self._irc_host, wait=False)
except:
self.fail('Failed to connect to IRC Server')
_spawn(_delayed_disconnect(client))
try:
self._loop.run_until_complete(client.disconnect_future)
except:
self.fail('Failed to wait for disconnection')
# ------------------------------------------------------------------------------
# ENTRY POINT ------------------------------------------------------------------
if __name__ == '__main__':
enable_verbose()
main()
# ------------------------------------------------------------------------------
# vim: set filetype=python :
# ------------------------------------------------------------------------------
# UNIT TESTS -------------------------------------------------------------------
class TestIRCClient(TestCase):
def setUp(self):
self._irc_host = 'irc.freenode.net'
self._loop = get_event_loop()
def test_connect(self):
try:
client = IRCClient.connect(host=self._irc_host, wait=False)
except:
self.fail('Failed to connect to IRC Server')
_spawn(_delayed_disconnect(client))
try:
self._loop.run_until_complete(client.disconnect_future)
except:
self.fail('Failed to wait for disconnection')
# ------------------------------------------------------------------------------
# ENTRY POINT ------------------------------------------------------------------
if __name__ == '__main__':
enable_verbose()
main()
# ------------------------------------------------------------------------------
# vim: set filetype=python :