Support other ssh-type commands via symlinking

cb844838 · Brandon Smith · 4d4fc71c · cb844838
Commit cb844838 authored Mar 02, 2016 by Brandon Smith
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

ssh-u2f.py ssh-u2f.py +6 -1

No files found.
--- a/ssh-u2f.py
+++ b/ssh-u2f.py
@@ -24,7 +24,12 @@ def winch_handler(signum, frame):

 signal.signal(signal.SIGWINCH, winch_handler)

-ssh = pexpect.spawn("ssh", sys.argv[1:])
+try:
+    command = os.path.splitext(os.path.basename(__file__))[0].split("-")[0]
+except:
+    command = None
+
+ssh = pexpect.spawn(command or "ssh", sys.argv[1:])
 winch_handler(None, None)

 index = -1