Python 实现远程服务器批量执行命令

paramiko 远程控制介绍

Python paramiko是一个相当好用的远程登录模块,采用ssh协议,可以实现linux服务器的ssh远程登录。首先来看一个简单的例子

import paramiko
#实例化ssh客户端
ssh = paramiko.SSHClient()
#创建默认的白名单
policy = paramiko.AutoAddPolicy()
#设置白名单
ssh.set_missing_host_key_policy(policy)
#链接服务器
ssh.connect(
hostname = "192.168.2.186", #服务器的ip
port = 22, #服务器的端口
username = "root", #服务器的用户名
password = "" #用户名对应的密码
)
#远程执行命令
stdin,stdout,stderr = ssh.exec_command("ls")
#exec_command 返回的对象都是类文件对象
#stdin 标准输入 用于向远程服务器提交参数,通常用write方法提交
#stdout 标准输出 服务器执行命令成功,返回的结果 通常用read方法查看
#stderr 标准错误 服务器执行命令错误返回的错误值 通常也用read方法
#查看结果,注意在Python3 字符串分为了:字符串和字节两种格式,文件返回的是字节
result = stdout.read().decode() print(result)

这是一段最简单的paramiko的代码,其中比较难理解的就是白名单概念,所谓的白名单就是ssh的know_hosts文件,用来记录paramiko可以信任的远程主机,在

paramiko当中,需要生成和采用两个步骤,来避免在远程链接的时候,被询问是否信任远程登录的服务器。

threaing多线程介绍

想要学习多线程,先要明白:

  进程:是程序的一次运行,我们运行一个程序,就会形成一个进程,进程有独立的内存空间,也被称为重量级进程。

  线程:线程是进程下的一个分支,也被称为轻量级进程。

threading是python多线程的基础模块,也可以说Python程序员接触多线程最初接触的模块,

我们来看一下threading最基础的案例

为了好理解我们来了解前置代码的编写

最简单的例子:

 from time import sleep
#通过这个例子我们来看计算机正常情况下的执行顺序 从左往右,从上到下
def loop0():
print("loop 0 is start")
sleep(3)
print("loop 0 is down") def loop1():
print("loop 1 is start")
sleep(2)
print("loop 1 is down") def main():
print("all is start")
loop0()
loop1()
print("all is down") if __name__ == "__main__":
main()

代码优化过的例子:

from time import sleep

def loop(num,sleeptime):
"""
当前函数作为功能函数
:param num: 函数的编号
:param sleeptime: 睡眠的时间
"""
print("loop %s is start"%num)
sleep(sleeptime)
print("loop %s is done"%num) def main():
sleep_list = [3,2] #睡眠时间
lenth = len(sleep_list) #获取列表长度
print("all is start")
for i in range(lenth):
loop(i,sleep_list[i]) #按照列表长度和列表内容调用函数
print("all is down") if __name__ == "__main__":
main()

多线程调用的例子:

 import threading
from time import sleep def loop(num,sleeptime):
"""
当前函数作为功能函数
:param num: 函数的编号
:param sleeptime: 睡眠的时间
"""
print("loop %s is start"%num)
sleep(sleeptime)
print("loop %s is done"%num) def main():
sleep_list = [3,2] #睡眠时间
lenth = len(sleep_list) #获取列表长度
thread_list = []
print("all is start")
for i in range(lenth):
#threading.Thread 就是用线程来执行我们的功能
t = threading.Thread(target = loop,args = (i,sleep_list[i])) #按照列表长度和列表内容调用函数
thread_list.append(t) #将生成的线程添加到列表里
for t in thread_list:
t.start() #开始执行线程
for t in thread_list:
t.join() #挂起线程,到所有线程结束
print("all is down")

通过上面例子的执行,我们需要掌握以几个点:

  1、threading.Thread方法用来创建线程,线程被创建的时候没有执行,需要通过start方法执行

  2、多线程调用实际上就是把原有的功能发到线程上面去跑,所以我们使用多线程,功能要写成函数

  3、Python的多线程是异步并发,而不是大家理解的并行,在这里

    并发:指的是同时出发,不同时执行

    并行:指的是同时进行,不一定同时出发

  希望各位同学注意上面的三点,而值得大家思考的就是上面代码最后的for循环可以简化吗?为啥?

paramiko+threading 实现远程服务器批量执行命令

完成上面的代码实例,我们进行代码的整合,完成远程服务器批量执行命令的效果

 #coding:utf-8

 import sys
import paramiko
import threading def getConnection(ip,username,password,command,port = 22):
"""
:param ip: 服务器的ip
:param username: 服务器的用户名称
:param password: 服务器的密码
:param CMD: 服务器的命令
:param port: 服务器的端口
"""
ssh = paramiko.SSHClient()
policy = paramiko.AutoAddPolicy()
ssh.set_missing_host_key_policy(policy)
ssh.connect(
hostname = ip, # 服务器的ip
port = port, # 服务器的端口
username = username, # 服务器的用户名
password = password # 用户名对应的密码
)
stdin, stdout, stderr = ssh.exec_command(command) result = stdout.read().decode() error = stderr.read().decode() print("+++++++++++++++++++++++start++++++++++++++++++++")
print("[connect success] | ip : %s" % ip)
print("result: \n %s"%result)
if error != " ":
print("error: \n %s"%error)
print("+++++++++++++++++++++++done++++++++++++++++++++") ssh.close()
#我们采用多线程
def main(host_list,command):
thread_list = []
for ip,username,password in host_list:
thread = threading.Thread(target = getConnection, args = (ip,username,password,command))
thread_list.append(thread)
for t in thread_list:
t.start()
for t in thread_list:
t.join() if __name__ == "__main__":
host_list = [
("192.168.2.186", "root", ""),
("192.168.2.88", "root", ""),
]
command = sys.argv[1]
main(host_list,command)

在这当中,我们使用了sys.argv方法来接收脚本外面执行的参数,当然,我们也可以将host_list写到一个配置文件当中使用。执行效果如下:

Python 实现远程服务器批量执行命令

上一篇:YUM源设置


下一篇:Python通过paramiko批量远程主机执行命令