def main():
"""Execute a command."""
def exit_on_int(sig, frame):
sys.exit(128 + sig)
signal.signal(signal.SIGINT, exit_on_int)
parser = argparse.ArgumentParser(
prog=RB_MAIN,
usage='%(prog)s [--version] <command> [options] [<args>]',
add_help=False)
for option in GLOBAL_OPTIONS:
option.add_to(parser)
opt = parser.parse_args()
if not opt.command:
help([], parser)
command_name = opt.command[0]
args = opt.command[1:]
if command_name == 'help':
help(args, parser)
elif opt.help or '--help' in args[:2] or '-h' in args[:2]:
help(opt.command, parser)
ep = find_entry_point_for_command(command_name)
if ep:
try:
command = ep.load()()
except ImportError:
# TODO: It might be useful to actual have the stack
# trace here, due to an import somewhere down the import
# chain failing.
sys.stderr.write('Could not load command entry point %s\n' %
ep.name)
sys.exit(1)
except Exception as e:
sys.stderr.write('Unexpected error loading command %s: %s\n' %
(ep.name, e))
sys.exit(1)
command.run_from_argv([RB_MAIN, command_name] + args)
else:
# A command class could not be found, so try and execute
# the "rb-<command>" on the system.
try:
sys.exit(
subprocess.call(['%s-%s' % (RB_MAIN, command_name)] + args,
stdin=sys.stdin,
stdout=sys.stdout,
stderr=sys.stderr,
env=os.environ.copy()))
except OSError:
# OSError is only raised in this scenario when subprocess.call
# cannot find an executable with the name rbt-<command_name>. If
# this command doesn't exist, we will check if an alias exists
# with the name before printing an error message.
pass
aliases = load_config().get('ALIASES', {})
if command_name in aliases:
sys.exit(run_alias(aliases[command_name], args))
else:
parser.error('"%s" is not a command' % command_name)
def main():
"""Execute a command."""
def exit_on_int(sig, frame):
sys.exit(128 + sig)
signal.signal(signal.SIGINT, exit_on_int)
parser = argparse.ArgumentParser(
prog=RB_MAIN,
usage='%(prog)s [--version] <command> [options] [<args>]',
add_help=False)
for option in GLOBAL_OPTIONS:
option.add_to(parser)
opt = parser.parse_args()
if not opt.command:
help([], parser)
command_name = opt.command[0]
args = opt.command[1:]
if command_name == 'help':
help(args, parser)
elif opt.help or '--help' in args or '-h' in args:
help(opt.command, parser)
# Attempt to retrieve the command class from the entry points. We
# first look in rbtools for the commands, and failing that, we look
# for third-party commands.
ep = pkg_resources.get_entry_info('rbtools', 'rbtools_commands',
command_name)
if not ep:
try:
ep = next(
pkg_resources.iter_entry_points('rbtools_commands',
command_name))
except StopIteration:
# There aren't any custom entry points defined.
pass
if ep:
try:
command = ep.load()()
except ImportError:
# TODO: It might be useful to actual have the strack
# trace here, due to an import somewhere down the import
# chain failing.
sys.stderr.write('Could not load command entry point %s\n' %
ep.name)
sys.exit(1)
except Exception as e:
sys.stderr.write('Unexpected error loading command %s: %s\n' %
(ep.name, e))
sys.exit(1)
command.run_from_argv([RB_MAIN, command_name] + args)
else:
# A command class could not be found, so try and execute
# the "rb-<command>" on the system.
try:
sys.exit(
subprocess.call(['%s-%s' % (RB_MAIN, command_name)] + args,
stdin=sys.stdin,
stdout=sys.stdout,
stderr=sys.stderr,
env=os.environ.copy()))
except OSError:
# OSError is only raised in this scenario when subprocess.call
# cannot find an executable with the name rbt-<command_name>. If
# this command doesn't exist, we will check if an alias exists
# with the name before printing an error message.
pass
aliases = load_config().get('ALIASES', {})
if command_name in aliases:
sys.exit(run_alias(aliases[command_name], args))
else:
parser.error('"%s" is not a command' % command_name)