[python script] 스위치 설정 저장

파이썬 스크립트를 이용한 스위치 설정 정보 저장.(Cisco, extreme)

지원 스위치: Cisco IOS , Extreme EXOS 스위치
지원 프로토콜: telnet, ssh (익스트림 스위치 제외).
스크립트 실행 환경: python 3.6.5, netmiko, telnetlib가 필요하다.
기타사항: 시스코 스위치는 로그인 즉시 enable 상태가 되도록 설정되어 있어야함(설정 방법은 https://blog.boxcorea.com/wp/archives/2529를 참조).

스위치 접속 정보화일(switch.txt)은 아래와 같은 포맷이며, 각 열은 탭(\t) 하나로 구분한다.

# IP		PORT	USER	PASSWD		PROTOCOL	VENDOR
172.16.10.1	23 	admin	password1	telnet	cisco
172.16.10.2	22	root	password2	ssh	cisco
172.16.10.3	23	admin	password3	telnet	extreme

스크립트는 파이썬3.5, windows10에서 테스트했음.
파이썬 설치후, 아래 명령어로 netmiko를 설치한다.

pip install netmiko
pip install telnetlib

*** ver 0.0.3 업데이트
변경사항:
1.extreme 스위치 ssh 지원, netmiko 대신 paramiko 사용

pip install paramiko

2. 접속 에러 처리. 접속 안될때, 내용을 error로 저장
***

** ver 0.0.4 업데이트
변경사항:
1. thread 사용으로 성능 대폭 향상

** ver 0.0.5 업데이트
변경사항:
1. 명령행 옵션으로 임의의 스위치 정보 화일을 지원 (예. backup_switch.py [my_switch_info.txt] )

스크립트는 아래와 같다.

import telnetlib
import re
import time
import paramiko
import concurrent.futures
import sys

# backup_switch.py. ver 0.0.5
# Script by snowffox.
#
# Document: https://blog.boxcorea.com/wp/archives/2277
#
# Supported Switch is Extreme EXOS switch and Cisco IOS switch
# Supported protocol is telnet and ssh.
#
#
# switch info : ip, port, user_id, passwd, protocol, vendor(split by tab(\t) char)
#
# ver 0.0.2 : Error handling. Support SSH with netmiko
# ver 0.0.3 : Full support SSH with paramiko and hostname support.
# ver 0.0.4 : Improve performance by using Thread.
# ver 0.0.5 : Support custom file name on command line option.


def runSSH(switch, command):
    try:
        ssh=paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(hostname=switch[0], port=switch[1], username=switch[2], password=switch[3])
        stdin, stdout, stderr=ssh.exec_command(command.encode('ascii'))
        data=((stdout.read()).decode('ascii')).replace('\r\n', '\n')
    except:
        print("ERROR 001: SSH Connection error...")
        data="error"
    return data
    
def runTELNET(switch, command):
    try:
        tn=telnetlib.Telnet(switch[0], switch[1], 20)
        #tn.set_debuglevel(1)
        prompt=getPrompt(switch[5])
        tn.read_until(prompt[0].encode('ascii'))
        tn.write(switch[2].encode('ascii')+b"\n")
        tn.read_until(prompt[1].encode("ascii"))
        tn.write(switch[3].encode('ascii')+b"\n")
        tn.write(command.encode('ascii')+b"\n")
        tn.write("exit".encode('ascii')+b"\n")
        data=tn.read_all().decode('ascii')
        tn.close()
    except:
        print("ERROR 002: TELNET Connection error...\n")
        data="error"
    return data

def getPrompt(vendor):
    if vendor=='cisco':
        prompt=['Username: ','Password: ']
    elif vendor=='extreme':
        prompt=['login: ', 'password: ']
    elif vendor=='nexus':
        prompt=['login: ','Password: ']
    return prompt


def decCOMMAND(switch):
    if switch[5]=='cisco' and switch[4]=='telnet':
        command="terminal length 0\r\nshow run"
    elif switch[5]=='cisco' and switch[4]=='ssh':
        command='show run'
    elif switch[5]=='extreme'and switch[4]=='telnet':
        command="disable clipaging\r\nshow config"
    elif switch[5]=='extreme' and switch[4]=='ssh':
        command='show config'
    elif switch[5]=='nexus' and switch[4]=='telnet':
        command="terminal length 0\r\nshow run"
    elif switch[5]=='nexus' and switch[4]=='ssh':
        command='show run'
    else :
        print("%s switch is not support..." %(switch[5]))
        command="not supported"
    return command

def gethostNAME(vendors, s):
    if s=="error":
        return "error"
    l=[]
    h=""
    if not s:
        s="ERROR 003 can not extract hosname!!!"
    if str.lower(vendors)=="cisco":
        pattern="hostname"
        dli=" "
    elif str.lower(vendors)=="extreme":
        pattern="configure snmp sysName"
        dli="\""
    else:
        print("ERR 004 vendors %s Hostname not support!" %(vendors))
        hostname=vendors
        return hostname

    l=(s.strip()).split('\n')
    for i in l:
        p=re.compile(pattern)
        m=p.match(i)
        if m :
            h=i

    l=h.split(dli)
    hostname=l[1].strip()
    return hostname

def saveCONFIG(switch, data):
    host=gethostNAME(switch[5], data)
    fname=host+'_'+switch[0]+".txt"
    f=open(fname, "w")
    f.write(data)
    f.close()

def main(switch):
    print("switch information is...")
    print(switch)
    command=decCOMMAND(switch)
    if switch[4]=='ssh':
        result=runSSH(switch, command)
    elif switch[4]=='telnet':
        result=runTELNET(switch, command)
    else:
        print(switch[4])
        print("Not supported...")
        result="Not supported..."
        
    saveCONFIG(switch, result)

if __name__=="__main__":
    argn=len(sys.argv)
    if argn==1:
        print("use default switch info file 'switch.txt'...")
        fname="switch.txt"
    elif argn>2:
        print("use only one switch info file\nProgram ended...")
        exit()
    else:
        fname=sys.argv[1]
        
    try:
        f=open(fname, 'r')
    except:
        print("File '%s' not found" %(fname))
        exit()
        
    while True:
        l=f.readline().strip('\n')
        if not l:
            break
        switch=l.split('\t')
    #   print(switch)
        n=len(switch)
        if n !=6:
            print("switch information format is wrong...\n")
            print(switch)
            continue
        else:
            pool=concurrent.futures.ProcessPoolExecutor(4)
            future=pool.submit(main, (switch))
            future.done()

    f.close()

스크립트 실행 결과는 hostname_ip(x.x.x.x).txt 로 각 스위치 별로 저장된다.

윈도우용 파이썬과 위 스크립트는 아래에서 다운로드 받을 수 있다.
ver 0.0.1: backup_switch_0.0.1.zip

ver 0.0.3: backup_switch_v0.0.3

ver 0.0.4: backup_switch_0.0.4

ver 0.0.5: backup_switch_0.0.5