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写到一个配置文件当中使用。执行效果如下：