diff --git a/libcxx/utils/run.py b/libcxx/utils/run.py index e88fff12824821..cdfa2387b22d57 100755 --- a/libcxx/utils/run.py +++ b/libcxx/utils/run.py @@ -23,12 +23,9 @@ def main(): parser.add_argument('--execdir', type=str, required=True) parser.add_argument('--codesign_identity', type=str, required=False, default=None) parser.add_argument('--env', type=str, nargs='*', required=False, default=dict()) - (args, remaining) = parser.parse_known_args(sys.argv[1:]) - - if len(remaining) < 2: - sys.stderr.write('Missing actual commands to run') - exit(1) - commandLine = remaining[1:] # Skip the '--' + parser.add_argument("command", nargs=argparse.ONE_OR_MORE) + args = parser.parse_args() + commandLine = args.command # Do any necessary codesigning. if args.codesign_identity: diff --git a/libcxx/utils/ssh.py b/libcxx/utils/ssh.py index f74ed785bbbfc8..876e35460dc7d8 100755 --- a/libcxx/utils/ssh.py +++ b/libcxx/utils/ssh.py @@ -29,13 +29,9 @@ def main(): parser.add_argument('--execdir', type=str, required=True) parser.add_argument('--codesign_identity', type=str, required=False, default=None) parser.add_argument('--env', type=str, nargs='*', required=False, default=dict()) - (args, remaining) = parser.parse_known_args(sys.argv[1:]) - - if len(remaining) < 2: - sys.stderr.write('Missing actual commands to run') - return 1 - - commandLine = remaining[1:] # Skip the '--' + parser.add_argument("command", nargs=argparse.ONE_OR_MORE) + args = parser.parse_args() + commandLine = args.command ssh = lambda command: ['ssh', '-oBatchMode=yes', args.host, command] scp = lambda src, dst: ['scp', '-q', '-oBatchMode=yes', src, '{}:{}'.format(args.host, dst)]