使用paramiko执行远程linux主机命令

　　paramiko是python的SSH库，可用来连接远程linux主机，然后执行linux命令或者通过SFTP传输文件。

　　关于使用paramiko执行远程主机命令可以找到很多参考资料了，本文在此基础上做一些封装，便于扩展与编写脚本。

　　下面直接给出代码：

　　

 # coding: utf-8

 import paramiko
 import re
 from time import sleep

 # 定义一个类，表示一台远端linux主机
 class Linux(object):
     # 通过IP, 用户名，密码，超时时间初始化一个远程Linux主机
     def __init__(self, ip, username, password, timeout=30):
         self.ip = ip
         self.username = username
         self.password = password
         self.timeout = timeout
         # transport和chanel
         self.t = ''
         self.chan = ''
         # 链接失败的重试次数
         self.try_times = 3

     # 调用该方法连接远程主机
     def connect(self):
         while True:
             # 连接过程中可能会抛出异常，比如网络不通、链接超时
             try:
                 self.t = paramiko.Transport(sock=(self.ip, 22))
                 self.t.connect(username=self.username, password=self.password)
                 self.chan = self.t.open_session()
                 self.chan.settimeout(self.timeout)
                 self.chan.get_pty()
                 self.chan.invoke_shell()
                 # 如果没有抛出异常说明连接成功，直接返回
                 print u'连接%s成功' % self.ip
                 # 接收到的网络数据解码为str
                 print self.chan.recv(65535).decode('utf-8')
                 return
             # 这里不对可能的异常如socket.error, socket.timeout细化，直接一网打尽
             except Exception, e1:
                 if self.try_times != 0:
                     print u'连接%s失败，进行重试' %self.ip
                     self.try_times -= 1
                 else:
                     print u'重试3次失败，结束程序'
                     exit(1)

     # 断开连接
     def close(self):
         self.chan.close()
         self.t.close()

     # 发送要执行的命令
     def send(self, cmd):
         cmd += '\r'
         # 通过命令执行提示符来判断命令是否执行完成
         p = re.compile(r':~ #')

         result = ''
         # 发送要执行的命令
         self.chan.send(cmd)
         # 回显很长的命令可能执行较久，通过循环分批次取回回显
         while True:
             sleep(0.5)
             ret = self.chan.recv(65535)
             ret = ret.decode('utf-8')
             result += ret
             if p.search(ret):
                 print result
                 return result

　　下面进行测试：

 # 主机IP错误，无法连接的情况
 if __name__ == '__main__':
     host = Linux('192.168.180.12', 'root', 'xxxx')
     host.connect()
          host.send('ls -l')
     host.close()

 # 链接正常的情况
 if __name__ == '__main__':
     host = Linux('192.168.180.128', 'root', 'love')
     host.connect()
     host.send('ls -l')
     host.close()

 运行结果：
 连接192.168.180.128成功
 Last login: Sat May 21 07:25:39 2016 from 192.168.180.1
 Have a lot of fun...

 ls -l
 192:~ # ls -l
 total 28
 -rw------- 1 root root 18 May 21 07:17 .bash_history
 drwxr-xr-x 1 root root 28 May 21 06:02 .config
 drwx------ 1 root root 22 May 21 05:57 .dbus
 drwx------ 1 root root  0 Sep 25  2014 .gnupg
 drwxr-xr-x 1 root root 10 May 21 06:15 .local
 -rw------- 1 root root 55 May 21 06:03 .xauth5mesuo
 -rw------- 1 root root 55 May 21 07:22 .xauthEYqDmK
 -rw------- 1 root root 55 May 21 07:25 .xauthGTrohO
 -rw------- 1 root root 55 May 21 07:09 .xauthP90TnG
 -rw------- 1 root root 48 May 21 07:40 .xauthjW8pI9
 -rw------- 1 root root 48 May 21 07:40 .xauthx8T4ED
 drwxr-xr-x 1 root root  0 Sep 25  2014 bin
 drwxr-xr-x 1 root root 38 May 21 05:43 inst-sys
 192:~ # 

 Process finished with exit code 0

 连接192.168.180.12失败，进行重试
 连接192.168.180.12失败，进行重试
 连接192.168.180.12失败，进行重试
 重试3次失败，结束程序

 Process finished with exit code 1

　　从上面代码可以看出，通过简单封装之后，执行远端linux主机就非常简单了，只需简单几行代码即可。