python脚本批量保存交换机配置

原创

mob64ca12e4594b 2024-06-16 05:06:22 ©著作权

文章标签 配置信息 python 开发者 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12e4594b的原创作品，请联系作者获取转载授权，否则将追究法律责任

如何实现“python脚本批量保存交换机配置”

流程图

flowchart TD
    A[连接交换机] --> B[登录交换机]
    B --> C[保存交换机配置]
    C --> D[断开连接]

表格展示步骤

步骤	描述
1	连接交换机
2	登录交换机
3	保存交换机配置
4	断开连接

代码实现

import paramiko

# 步骤1：连接交换机
def connect_switch(ip, username, password):
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(ip, username=username, password=password)
    return ssh

# 步骤2：登录交换机
def login_switch(ssh):
    ssh.invoke_shell()
    stdin, stdout, stderr = ssh.exec_command('enable\n')
    stdin.write('password\n')
    stdin.write('terminal length 0\n')
    stdin.write('show running-config\n')
    output = stdout.read().decode('utf-8')
    return output

# 步骤3：保存交换机配置
def save_config(output, filename):
    with open(filename, 'w') as f:
        f.write(output)

# 步骤4：断开连接
def disconnect_switch(ssh):
    ssh.close()

# 主函数
def main():
    ip = '192.168.1.1'
    username = 'admin'
    password = 'admin123'
    filename = 'switch_config.txt'

    ssh = connect_switch(ip, username, password)
    output = login_switch(ssh)
    save_config(output, filename)
    disconnect_switch(ssh)

if __name__ == '__main__':
    main()

在以上代码中：