start_shell: initialize SSH client the same way as for SCP

Juniper · Jan 13, 2017 · 35d3390 · 35d3390
1 parent 63ccdcc
commit 35d3390
Showing 1 changed file with 3 additions and 11 deletions.
diff --git a/lib/jnpr/junos/utils/start_shell.py b/lib/jnpr/junos/utils/start_shell.py
@@ -1,8 +1,9 @@
-import paramiko
 from select import select
 import re
 import datetime
 
+from jnpr.junos.utils.misc import get_ssh_client
+
 _JUNOS_PROMPT = '> '
 _SHELL_PROMPT = '(%|#)\s'
 _SELECT_WAIT = 0.1
@@ -82,16 +83,7 @@ def open(self):
         :class:`paramiko.SSHClient` instance.
         """
         junos = self._nc
-
-        client = paramiko.SSHClient()
-        client.load_system_host_keys()
-        client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
-        client.connect(hostname=junos.hostname,
-                       port=(22, junos._port)[junos.hostname == 'localhost'],
-                       username=junos._auth_user,
-                       password=junos._auth_password,
-                       )
-
+        client = get_ssh_client(junos)
         chan = client.invoke_shell()
         self._client = client
         self._chan = chan