diff --git a/crussh.py b/crussh.py index 909e952..77bf22b 100755 --- a/crussh.py +++ b/crussh.py @@ -7,8 +7,6 @@ # Requires: python-gtk2 python-vte import sys -import argparse -import getpass import math try: import gtk @@ -22,19 +20,6 @@ except: error.run() sys.exit(1) -### Parse CLI Args ### -parser = argparse.ArgumentParser(description="Connect to multiple servers in parallel.") -parser.add_argument('hosts', metavar='HOST', nargs='+', - help="Host(s) to connect to.") -parser.add_argument('-l', '--login', dest='login', default=None, - help="Login name to use.") -parser.add_argument('-p', '--port', dest='port', type=int, default=None, - help="Alternate SSH port to use.") -parser.add_argument('-s', '--fontsize', dest='fontsize', type=int, default=10, - help="Font size to use. (default=10)") - -args = parser.parse_args() - ### CruSSH! ### class CruSSH: ### Config Vars ### @@ -138,10 +123,7 @@ class CruSSH: # give EntryBox default focus on init self.EntryBox.props.has_focus = True - def __init__(self, hosts, login=None, port=None): - self.Config["login"] = login - self.Config["port"] = port - + def __init__(self, hosts, ssh_args=None): # init all terminals for host in hosts: terminal = vte.Terminal() @@ -152,10 +134,8 @@ class CruSSH: # TODO: disable only this terminal widget on child exit # v.connect("child-exited", lambda term: gtk.main_quit()) cmd_str = "/usr/bin/ssh" - if self.Config["login"] != None: - cmd_str += " -l " + args.login - if self.Config["port"] != None: - cmd_str += " -p " + str(args.port) + if ssh_args is not None: + cmd_str += " " + ssh_args cmd_str += " " + host cmd = cmd_str.split(' ') terminal.fork_command(command=cmd[0], argv=cmd) @@ -167,6 +147,27 @@ class CruSSH: self.initGUI() self.reflow(force=True) -### Start Execution ### -crussh = CruSSH(args.hosts, args.login, args.port) -gtk.main() +if __name__ == "__main__": + import argparse + + ### Parse CLI Args ### + parser = argparse.ArgumentParser(description="Connect to multiple servers in parallel.", usage="%(prog)s [OPTIONS] HOST [HOST ...]") + parser.add_argument('-s', '--fontsize', dest='fontsize', type=int, default=10, + help="Font size to use. (default=10)") + + (args, hosts) = parser.parse_known_args() + + if len(hosts) == 0: + parser.print_usage() + sys.exit(1) + + if "--" in hosts: + offset = hosts.index("--") + 1 + ssh_args = " ".join(hosts[0:offset]) + hosts = hosts[offset:] + else: + ssh_args = None + + ### Start Execution ### + crussh = CruSSH(hosts, ssh_args) + gtk.main()