Python批量使用SSH登录交换机

引言

在网络管理中，需要登录到多个交换机上执行相同的配置或者命令是很常见的需求。手动一个一个登录交换机并逐个执行命令是非常繁琐的，因此我们可以使用Python编写脚本来实现批量登录交换机并执行相同的操作。

本文将介绍如何使用Python的paramiko库来实现批量使用SSH登录交换机的方案，并提供代码示例。

方案概述

本方案的主要思路是使用paramiko库来建立SSH连接并执行命令。具体步骤如下：

  1. 读取交换机的IP地址、用户名和密码等登录信息。
  2. 使用paramiko库连接到交换机。
  3. 执行相应的命令或者配置。
  4. 关闭SSH连接。
  5. 重复上述步骤直到所有交换机都处理完毕。

方案详解

读取登录信息

首先，我们需要准备一个文件来存储交换机的登录信息，例如一个名为switch_info.txt的文本文件。每一行代表一个交换机的登录信息，格式为IP地址,用户名,密码。示例文件内容如下：

192.168.1.1,admin,password
192.168.1.2,admin,password
192.168.1.3,admin,password

接下来，我们使用Python来读取这个文件并将每一行的登录信息保存到一个列表中。代码如下：

switch_info = []

with open('switch_info.txt', 'r') as f:
    for line in f:
        ip, username, password = line.strip().split(',')
        switch_info.append((ip, username, password))

建立SSH连接并执行命令

接下来，我们使用paramiko库来建立SSH连接并执行命令。代码示例如下：

import paramiko

for ip, username, password in switch_info:
    # 建立SSH连接
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    client.connect(ip, username=username, password=password)

    # 执行命令
    stdin, stdout, stderr = client.exec_command('show version')
    output = stdout.read().decode()

    # 输出结果
    print(f'{ip}:\n{output}')

    # 关闭SSH连接
    client.close()

上述代码中，我们首先通过paramiko.SSHClient()创建一个SSH客户端对象，然后使用set_missing_host_key_policy方法设置自动添加新的主机密钥。接着，我们使用connect方法连接到交换机，传入IP地址、用户名和密码等登录信息。之后，我们使用exec_command方法执行命令，并通过stdout.read().decode()读取命令输出。最后，我们通过close方法关闭SSH连接。

配置交换机

除了执行命令，我们还可以使用Python来配置交换机。例如，我们可以通过SSH登录交换机并修改交换机的配置。代码示例如下：

import paramiko

config_commands = [
    'interface GigabitEthernet1/0/1',
    'switchport mode access',
    'switchport access vlan 10',
    'no shutdown'
]

for ip, username, password in switch_info:
    # 建立SSH连接
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    client.connect(ip, username=username, password=password)

    # 执行配置命令
    for command in config_commands:
        stdin, stdout, stderr = client.exec_command(command)
        output = stdout.read().decode()
        print(f'{ip}:\n{output}')

    # 关闭SSH连接
    client.close()

上述代码中，我们将要执行的配置命令存储在一个列表中，然后逐个执行这些命令。执行过程与执行命令类似，只需将命令替换为配置命令。

关系图

下面是一个使用mermaid语法绘制的关系图，展示了本方案中的关键步骤和流程。

erDiagram
    登录信息 ||..|{ 建立SSH连接
    建立SSH连接 ||..|{ 执行命令
    建立SSH连接 ||..|{ 配置交换机