使用python中有一个paramiko模块来实现python SSH客户端,与SSH服务器交互时,需要注意有交互式和非交互式的区别。
只执行单条命令,之后就断开链接,可以使用非交互方式。执行多条命令,或者基于前面的输出结果来判断后续要执行的命令,需要使用交互式方式。
我在写自动化测试用例时,就尝试使用非交互方式去连接一个只支持交互方式的SSH服务器,就怎么也读不到返回结果。换成交互式后就可以了。
需要注意的是,命令后面记得加“\n”。
下面内容转自: https://blog.csdn.net/u012322855/article/details/77839929/
python中有一个paramiko,功能强大,用来做SSH比较方便
先上代码
import paramiko
class SSHConnection(object):
def __init__(self, host, port, username, password):
self._host = host
self._port = port
self._username = username
self._password = password
self._transport = None
self._sftp = None
self._client = None
self._connect() # 建立连接 def _connect(self):
transport = paramiko.Transport((self._host, self._port))
transport.connect(username=self._username, password=self._password)
self._transport = transport #下载
def download(self, remotepath, localpath):
if self._sftp is None:
self._sftp = paramiko.SFTPClient.from_transport(self._transport)
self._sftp.get(remotepath, localpath) #上传
def put(self, localpath, remotepath):
if self._sftp is None:
self._sftp = paramiko.SFTPClient.from_transport(self._transport)
self._sftp.put(localpath, remotepath) #执行命令
def exec_command(self, command):
if self._client is None:
self._client = paramiko.SSHClient()
self._client._transport = self._transport
stdin, stdout, stderr = self._client.exec_command(command)
data = stdout.read()
if len(data) > 0:
print data.strip() #打印正确结果
return data
err = stderr.read()
if len(err) > 0:
print err.strip() #输出错误结果
return err def close(self):
if self._transport:
self._transport.close()
if self._client:
self._client.close()
接下来就简单测试一下exec_command这个命令,比较常用
if __name__ == "__main__":
conn = SSHConnection('ip', port, 'username', 'password') conn.exec_command('ls -ll')
conn.exec_command('cd /home/test;pwd') #cd需要特别处理
conn.exec_command('pwd')
conn.exec_command('tree /home/test')
exec_command这个函数如果想cd,可以使用pwd这样可以到当前目录而不是初始目录,但是有些情况下,比如chroot,是做不到的,这个时候就需要新的方法
上代码
ssh = paramiko.SSHClient() #创建sshclient
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) #目的是接受不在本地Known_host文件下的主机。
ssh.connect("ip",port,'username','password')
command='chroot /xxx\n'
#conn.write(command)
chan=ssh.invoke_shell()#新函数
chan.send(command+'\n')
#\n是执行命令的意思,没有\n不会执行
time.sleep(10)#等待执行,这种方式比较慢
#这个时候就可以在chroot目录下执行命令了
res=chan.recv(1024)#非必须,接受返回消息
chan.close()
注意invoke_shell这个函数即可
另外使用这个函数命令后面记得加“\n”