百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 优雅编程 > 正文

运维必备!用Python管理多个Linux服务器的连接与命令执行

sinye56 2025-01-05 20:04 10 浏览 0 评论

引言:

代码实现多个Linux服务器的SSH连接,支持添加、删除连接,执行命令,并返回标准输出。可以在需要时批量处理多个服务器,避免了手动逐个登录和执行命令的繁琐过程。

代码封装如下:

# -*- coding: utf-8 -*-
import paramiko  # 导入paramiko库,主要用于SSH连接到远程Linux服务器

class MultiLinuxConnection:
    def __init__(self):
        """
        初始化MultiLinuxConnection类,设置连接的列表。
        """
        self.connections = []  # 初始化一个空的列表,用于存储所有的SSH连接对象

    def add_connection(self, host, port, username, password):
        """
        添加Linux服务器连接,通过SSH进行连接,并将连接对象保存到列表中。
        
        :param host: 目标服务器的IP地址
        :param port: SSH端口,默认为22
        :param username: 登录的用户名
        :param password: 登录的密码
        """
        connection = paramiko.SSHClient()  # 创建SSHClient对象,用于建立SSH连接
        connection.set_missing_host_key_policy(paramiko.AutoAddPolicy())  # 忽略目标服务器的未知主机密钥
        connection.connect(host, port=port, username=username, password=password, look_for_keys=False)  # 连接到目标服务器
        self.connections.append(connection)  # 将连接对象添加到连接列表中

    def remove_connection(self, index):
        """
        移除指定索引处的Linux服务器连接,并关闭该连接。
        
        :param index: 要移除的连接的索引
        """
        if index < len(self.connections):  # 确保索引有效
            connection = self.connections.pop(index)  # 从连接列表中移除该连接
            connection.close()  # 关闭该连接

    def execute_command(self, index, command):
        """
        在指定的Linux服务器上执行命令,并返回输出结果。
        
        :param index: 连接列表中的索引
        :param command: 要执行的命令
        :return: 命令执行后的输出结果
        """
        if index < len(self.connections):  # 确保索引有效
            connection = self.connections[index]  # 获取指定索引的连接对象
            stdin, stdout, stderr = connection.exec_command(command)  # 执行命令
            output = stdout.read().decode().strip()  # 获取标准输出,解码为字符串并去除两端空格
            return output  # 返回命令执行的输出结果

    def execute_command_all(self, command):
        """
        在所有已连接的Linux服务器上执行命令,并返回每个服务器的输出结果。
        
        :param command: 要执行的命令
        :return: 包含所有连接输出结果的字典,key为连接索引,value为命令输出
        """
        output_dict = {}  # 用字典保存每个服务器的命令输出
        for index, connection in enumerate(self.connections):  # 遍历所有连接
            output_dict[index] = self.execute_command(index, command)  # 执行命令并保存输出
        return output_dict  # 返回输出字典

    def close_all_connections(self):
        """
        关闭所有Linux服务器的SSH连接。
        """
        for connection in self.connections:  # 遍历所有连接
            connection.close()  # 关闭每一个连接
        self.connections = []  # 清空连接列表

# 以下部分用于测试代码,实际应用时可以注释掉
if __name__ == '__main__':
    # 初始化MultiLinuxConnection对象
    li = MultiLinuxConnection()  
    
    # 添加多个Linux服务器的连接
    li.add_connection('127.0.0.1', 22, 'root', 'password')  
    li.add_connection('127.0.0.2', 22, 'root', 'password')  

    # 在第一个连接的服务器上执行"ifconfig"命令并输出结果
    output1 = li.execute_command(0, "ifconfig")  
    print(output1)  # 打印第一个服务器的命令输出

    # 可选:在所有连接的服务器上执行"df -h"命令并输出结果
    # output2 = li.execute_command_all("df -h")
    # print(output2)  # 打印所有服务器的命令输出

    # 移除第二个连接
    # li.remove_connection(1)  
    
    # 关闭所有连接
    li.close_all_connections()  

注意:

钥认证:在实际的应用场景中,建议使用SSH密钥认证而非用户名/密码认证。可以提高安全性,避免暴露密码。

ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())  # 会自动添加主机密钥,可能会有安全风险

并发执行:命令多个服务器同时执行命令时,可能会遇到性能瓶颈,尤其是当命令执行较长时间时。可以考虑使用多线程或异步编程来提高执行效率(大家自己尝试下)

from concurrent.futures import ThreadPoolExecutor

with ThreadPoolExecutor(max_workers=10) as executor:
    futures = [executor.submit(self.run_command_on_server, ip, port, username, password, command) for ip in self.connections]
    for future in futures:
        result = future.result()
        print(result)

相关推荐

程序员:JDK的安装与配置(完整版)_jdk的安装方法

对于Java程序员来说,jdk是必不陌生的一个词。但怎么安装配置jdk,对新手来说确实头疼的一件事情。我这里以jdk10为例,详细的说明讲解了jdk的安装和配置,如果有不明白的小伙伴可以评论区留言哦下...

Linux中安装jdk并配置环境变量_linux jdk安装教程及环境变量配置

一、通过连接工具登录到Linux(我这里使用的Centos7.6版本)服务器连接工具有很多我就不一一介绍了今天使用比较常用的XShell工具登录成功如下:二、上传jdk安装包到Linux服务器jdk...

麒麟系统安装JAVA JDK教程_麒麟系统配置jdk

检查检查系统是否自带java在麒麟系统桌面空白处,右键“在终端打开”,打开shell对话框输入:java–version查看是否自带java及版本如图所示,系统自带OpenJDK,要先卸载自带JDK...

学习笔记-Linux JDK - 安装&amp;配置

前提条件#检查是否存在JDKrpm-qa|grepjava#删除现存JDKyum-yremovejava*安装OracleJDK不分系统#进入安装文件目...

Linux新手入门系列:Linux下jdk安装配置

本系列文章是把作者刚接触和学习Linux时候的实操记录分享出来,内容主要包括Linux入门的一些理论概念知识、Web程序、mysql数据库的简单安装部署,希望能够帮到一些初学者,少走一些弯路。注意:L...

测试员必备:Linux下安装JDK 1.8你必须知道的那些事

1.简介在Oracle收购Sun后,Java的一系列产品就被整合到Oracle官网中,打开官网乍眼一看也不知道去哪里下载,还得一个一个的摸索尝试,而且网上大多数都是一些Oracle收购Sun前,或者就...

Linux 下安装JDK17_linux 安装jdk1.8 yum

一、安装环境操作系统:JDK版本:17二、安装步骤第一步:下载安装包下载Linux环境下的jdk1.8,请去官网(https://www.oracle.com/java/technologies/do...

在Ubuntu系统中安装JDK 17并配置环境变量教程

在Ubuntu系统上安装JDK17并配置环境变量是Java开发环境搭建的重要步骤。JDK17是Oracle提供的长期支持版本,广泛用于开发Java应用程序。以下是详细的步骤,帮助你在Ubuntu系...

如何在 Linux 上安装 Java_linux安装java的步骤

在桌面上拥抱Java应用程序,然后在所有桌面上运行它们。--SethKenlon(作者)无论你运行的是哪种操作系统,通常都有几种安装应用程序的方法。有时你可能会在应用程序商店中找到一个应用程序...

Windows和Linux环境下的JDK安装教程

JavaDevelopmentKit(简称JDK),是Java开发的核心工具包,提供了Java应用程序的编译、运行和开发所需的各类工具和类库。它包括了JRE(JavaRuntimeEnviro...

linux安装jdk_linux安装jdk软连接

JDK是啥就不用多介绍了哈,外行的人也不会进来看我的博文。依然记得读大学那会,第一次实验课就是在机房安装jdk,编写HelloWorld程序。时光飞逝啊,一下过了十多年了,挣了不少钱,买了跑车,娶了富...

linux安装jdk,全局配置,不同用户不同jdk

jdk1.8安装包链接:https://pan.baidu.com/s/14qBrh6ZpLK04QS8ogCepwg提取码:09zs上传文件解压tar-zxvfjdk-8u152-linux-...

运维大神教你在linux下安装jdk8_linux安装jdk1.7

1.到官网下载适合自己机器的版本。楼主下载的是jdk-8u66-linux-i586.tar.gzhttp://www.oracle.com/technetwork/java/javase/downl...

window和linux安装JDK1.8_linux 安装jdk1.8.tar

Windows安装JDK1.8的步骤:步骤1:下载JDK打开浏览器,找到JDK下载页面https://d.injdk.cn/download/oraclejdk/8在页面中找到并点击“下载...

最全的linux下安装JavaJDK的教程(图文详解)不会安装你来打我?

默认已经有了linux服务器,且有root账号首先检查一下是否已经安装过java的jdk任意位置输入命令:whichjava像我这个已经安装过了,就会提示在哪个位置,你的肯定是找不到。一般我们在...

取消回复欢迎 发表评论: