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 = "123" #用户名对应的密码
)
#远程执行命令
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", "123"),
        ("192.168.2.88", "root", "123"),
    ]
    command = sys.argv[1]
    main(host_list,command)

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