【dump-Switch-config.py】備份CISCO config的小程式！

#!.\venv\Scripts\python.exe
# -*- coding: utf-8 -*-
#
# for Python version 3.10.0
# Author：William Huang
# Date：2022/10/24
# program：dump-Switch-config.py
#
# 

def main():
    import telnetlib
    from datetime import datetime

    # CISCO switch清單
    swList = open("backup-Switches.txt","r")
    # 記錄dump config失敗的switch
    failedList = open("Fail-to-created.txt","a+")

    for sw in swList:
        HOST = sw.strip()

        # 顯示處理中的switch
        print("Dumping " + HOST + "'s config...")

        # 登入switch的帳號
        user = <username>       # 帳號字串，請換成真正帳號

        # 登入switch的密碼
        password = <password>    #密碼字串，請換成真正密碼

        # switch prompt登入帳號
        loginPrompt=[b"Username: "]

        # switch prompt登入密碼
        pwdPrompt=[b"Password: "]

        # 判別user mode or enable mode
        pmp = [b"\>|\#"]

        # 存入config檔名格式：年月日-時分秒-switch-ip.conf
        configfile = open(datetime.today().strftime("%Y%m%d-%H%M%S-") + "switch-" + HOST + ".conf", "w")

        try:
            # 建立連到HOST的telnet連線
            tn = telnetlib.Telnet(HOST)

            # 等待輸入帳號的prompt
            tn.expect(loginPrompt)

            # 輸入帳號
            tn.write(user.encode('ascii') + b"\n")   # b"\n" == enter

            # 等待輸入密碼的prompt
            tn.expect(pwdPrompt)

            # 輸入密碼
            tn.write(password.encode('ascii') + b"\n")

            # 等待輸入指令的prompt，這裡是">"
            tn.expect(pmp)

            # 輸入enable指令
            tn.write("enable".encode('ascii') + b"\n")

            # 等待輸入密碼的prompt
            tn.expect(pwdPrompt)

            # 輸入密碼
            tn.write(password.encode('ascii') + b"\n")

            # 等待輸入指令的prompt，這裡是"#"
            tn.expect(pmp)

            # 輸入螢幕顯示不分頁指令
            tn.write(b"terminal length 0" + b"\n")

            # 等待輸入指令的prompt，這裡是"#"
            tn.expect(pmp)

            # 輸入show running-config指令
            tn.write(b"show running-config" + b"\n")

            # 等待指令結束回應"#"
            # 再讀取、儲存回應字串到dmpconfig
            dmpconfig = tn.read_until(b"#")    # 指令+前置處理字串+「running-config(用\r\n分行)」+後綴prompt

            # 結束enable mode
            tn.write(b"exit" + b"\n")

            # 讀取user mode prompt，這裡是">"
            tn.expect(pmp)

            # 結束user mode
            tn.write(b"exit" + b"\n")

            # 從dmpconfig字串，擷取running-config
            for _ in dmpconfig.split(b"\r\n")[4:-2]:  # 以\r\n分列，去掉頭尾非config列
                # 結果寫到configfile檔，格式見前述
                configfile.write(_.decode()+"\n")

        except (OSError,EOFError) as e:  # OSError、EOFError
            # 如果出現錯誤，寫入錯誤訊息到failedList檔
            failedList.write("Failed to create " + datetime.today().strftime("%Y%m%d-%H%M%S-") + HOST + ".conf!\n\r")

            # 如果出現錯誤，在螢幕顯示出錯的switch
            print("\nFailed to create " + datetime.today().strftime("%Y%m%d-%H%M%S-") + HOST + ".conf!")

            #  如果出現錯誤，在螢幕顯示完整錯誤訊息
            print(e)

            continue

        configfile.close()

    swList.close()
    failedList.close()

if __name__ == '__main__':
    main()