add support for passing command args

This commit is contained in:
Frédéric Péters 2018-12-29 21:07:08 +01:00
parent c2bf345bcb
commit f0ab4feea3
1 changed files with 4 additions and 3 deletions

View File

@ -67,8 +67,8 @@ class Server:
def shell(self):
return 'ssh %s' % self.name
def cmd(self, cmd):
return 'ssh -t %s "%s"' % (self.name, cmd)
def cmd(self, cmd, *args):
return 'ssh -t %s "%s %s"' % (self.name, cmd, ' '.join(args))
def get_servers():
@ -98,6 +98,7 @@ def parse_args():
parser.add_argument('--status-window', dest='session_name', type=str)
parser.add_argument('-k', dest='keywords', type=str)
parser.add_argument('cmd', type=str, nargs='?', default=None)
parser.add_argument('args', nargs=argparse.REMAINDER)
args = parser.parse_args()
return args
@ -277,7 +278,7 @@ else:
session.new_window(
attach=False,
window_name=server.name,
window_shell=server.cmd(cmd))
window_shell=server.cmd(cmd, *args.args))
break
else:
time.sleep(0.1)