python登陆华为交换机配置snmp

好久没有写文章了，重拾旧爱啊。多写文章，多总结，有利于提高，也有利于日后的应用。这次写文章是由于近期在做项目的时候要上一台网管设备，所以要为所有的交换机配置snmp，设备不多也就四十多台。我在现场的时候是这样想的，有没有什么方法可以快速的（最省事儿的）方法，最好可以翘着二档腿儿喝着茶就能配的方法，初步的构想python批量登陆进行配置，但是现场要求尽快配置好，所有就没有时间操练python了，但是回来之后查了查资料，可以实现。
并且做些简单的分析：

图片.png

我们需要做的事情基本上有三个，第一，读取excel文件获取相应的位置信息和ip地址信息；第二，登陆交换机执行命令；第三，如果有交换机没有配置成功可以捕获到这个ip以备后续分析。针对这三个问题我进行了如下的测试：

1. python读取excel文件

#encoding:utf-8
import xlrd

def read_excel():
    
    #打开问文件
    workbook = xlrd.open_workbook(r'message.xlsx')

    #获取所有sheet
    print workbook.sheet_names()
    sheet1_name = workbook.sheet_names()[0]

    #获取指定sheet
    sheet1 = workbook.sheet_by_index(0)
    sheet1 = workbook.sheet_by_name('Sheet1')

    #输出sheet名，行数，列数
    print sheet1.name,sheet1.nrows,sheet1.ncols

    #输出整行整列内容
    rows = sheet1.row_values(0)
    cols = sheet1.col_values(1)
    print rows
    print cols

    #获取指定单元格内容
    print sheet1.cell(0,0).value
    print sheet1.cell_value(0,0)
    print sheet1.row(1)[0].value

    #获取单元格数据类型
    print sheet1.cell(0,0).ctype

if __name__ == '__main__':
    read_excel()

这是xlrd库的应用，上面是通过这个库进行excel进行读取的一些简单测试，不过这些测试足够我们的目标脚本使用。

2. python登陆交换机

#encoding:utf-8
import paramiko

readGroup = "000000"
writeGroup = "1111111"
udpDomain = "192.168.3.101"
contact = "Mr.Liu-12345"
location = "F2-R110"

ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect('192.168.3.201',22,'huawei','[email protected]', allow_agent=False, look_for_keys=False)
stdin,stdout,stderr = ssh.exec_command('dis cu | in ssh')
print stdout.read()
def deploy():

def main():

if __name__ == '__main__':
    main()

这个小脚本是通过paramiko库进行ssh登陆，通过exec_command()执行命令，但是exec_command()无法连续执行多条命令，最后的脚本中我们使用的是invoke_shell()返回的ssh命令行的对象调用sendall()方法进行执行命令。

3. 实现批量配置snmp

#encoding:utf-8

#基本思路:
#通过read_excel()读取表格中的数据
#将读取到命令通过make_code()整合到命令中
#main函数中for循环来逐一读取交换机IP地址和位置信息
#并将ip地址存放入数组中将位置信息
#调用log_switch()登陆交换机进行配置

import paramiko
import xlrd
import time

#读取相关文件
def read_excel():

    print '[*]reading excel...'

    #打开文件
    workbook = xlrd.open_workbook(r'message.xlsx')
    #通过sheet名称确定sheet
    sheet1 = workbook.sheet_by_name('Sheet1')
    #获取列内容
    col0 = sheet1.col_values(0)
    col1 = sheet1.col_values(1)
    #去除字段名
    ip_list = col0[1:]
    location_list = col1[1:]

    return ip_list,location_list


def make_code(location_list):

    print '[*]making shell code...'
    shell_code_list = []

    for ip_address in location_list:
        shell_code = ['n',
        'sys',
        'snmp-agent community read ci public',
        'snmp-agent community write ci private',
        'snmp-agent sys-info contact Mr.Liu-Tel:123',
        'snmp-agent sys-info location '+ip_address,
        'snmp-agent sys-info version all',
        'snmp-agent target-host trap address udp-domain 192.168.1.101 udp-port 5000 params securityname public',
        'q',
        'sa',
        'Y'
        ]
        shell_code_list.append(shell_code)

    return shell_code_list

#登陆交换机部署snmp信息
def deploy(ip,shell_code,error_list):

    try:
        #交换机账号密码
        username = 'huawei'
        password = 'Bdff2018'

        #登陆获取会话
        print '[*]logging:'+ip
        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(ip,22,username,password,allow_agent=False,look_for_keys=False)
        ssh_shell = ssh.invoke_shell()

        #配置
        print '[*]deploying at:'+ip
        for i in range(0,len(shell_code)):
            #print shell_code[i]
            res = ssh_shell.sendall(shell_code[i]+'\n')
            #time.sleep(float(1))
        print '[+]Finished it.'
        ssh.close()
    #输出错误
    except Exception as e:
        error_list.append(ip)
        print '[-]error: '+ ip
        print e

def main():

    #定义错误列表，将未配置成功的ipappend进去
    error_list = []
    
    #ip_list是通过excel文件中获取到的交换机IP地址
    #location_list是通过excel中获取到的交换机位置信息
    ip_list,location_list = read_excel()

    #shell_code_list是将excel表中读取到的信息整合到命令中
    shell_code_list = make_code(location_list)

    #登陆配置
    for i in range(0,len(ip_list)):
        result = deploy(ip_list[i],shell_code_list[i],error_list)
    print '[*]Ok, well done.'

    #输出配置失败的ip地址
    if error_list:
        print '[!]Error ips are as follow:'
        for ip in  error_list:
            print '[!]-->'+ip

if __name__ == '__main__':
    main()

基本上就是这样，整体比较顺利，实现也还好，最好在批量发送命令之间有一定的间歇，不然容易出现socket wrong报错，其实可以通过多线程进行优化一下，等有时间优化一下再做升级。

4. 运行结果