Skip to content

Commit

Permalink
Merge pull request #65 from rolobio/quiet-swap
Browse files Browse the repository at this point in the history
Quiet swap
  • Loading branch information
rolobio committed Apr 8, 2015
2 parents c0ada92 + 9c93c5a commit 0ad2381
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 17 deletions.
11 changes: 2 additions & 9 deletions sshm/main.py
Expand Up @@ -42,12 +42,8 @@ def get_argparse_args(args=None):
help='Remove any whitespace surrounding the output of each instance.')
parser.add_argument('-d', '--disable-formatting', action='store_true', default=False,
help='Disable command formatting.')
parser.add_argument('-q', '--quiet', action='store_true', default=False,
help='Hide server information on output. This implies sorted.')
parser.add_argument('-u', '--ssh-quiet', action='store_true', default=False,
help='Pass -q to the SSH command to quiet its output.')
parser.add_argument('-w', '--workers', type=int, default=20,
help='The maximum number of simultaneous SSH connections.')
parser.add_argument('-u', '--quiet', action='store_true', default=False,
help="Hide SSHM's server information on output (this implies sorted).")
parser.add_argument('--version', action='version', version='%(prog)s '+__version__)
args, extra_args = parser.parse_known_args(args=args)

Expand All @@ -69,9 +65,6 @@ def get_argparse_args(args=None):
if args.quiet:
args.sorted_output = True

# Pass -q to SSH command
if args.ssh_quiet:
extra_args.insert(0, '-q')
return (args, args.command, extra_args)


Expand Down
9 changes: 1 addition & 8 deletions sshm/test/test_main.py
Expand Up @@ -71,7 +71,7 @@ def test_get_argparse_args(self):
self.assertEqual(extra_args, [])

# You can hide server information output
provided = ['-q', 'example.com', 'ls']
provided = ['-u', 'example.com', 'ls']
args, command, extra_args = get_argparse_args(provided)
self.assertEqual(args.servers, ['example.com'])
self.assertTrue(args.sorted_output)
Expand All @@ -86,13 +86,6 @@ def test_get_argparse_args(self):
self.assertEqual(command, 'ls')
self.assertEqual(extra_args, [])

# You can quiet SSH (passing -q to ssh) by using -u
provided = ['example.com', 'command', '-o Something', '-u']
args, command, extra_args = get_argparse_args(provided)
self.assertEqual(args.servers, ['example.com',])
self.assertEqual(command, 'command')
self.assertEqual(extra_args, ['-q', '-o Something'])


def test__print_handling_newlines(self):
"""
Expand Down

0 comments on commit 0ad2381

Please sign in to comment.