python之SSH（交互式和非交互式）

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”