使用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主机就非常简单了,只需简单几行代码即可。

上一篇:sqlite 跨数据库查询


下一篇:BestCoder Round #11 (Div. 2) 前三题题解