# -*- coding: utf-8 -*-
import paramiko
from datetime import datetime
def backup_device_config(host, username, password):
    """
    備份網(wǎng)絡(luò)設(shè)備配置
    :param host: 設(shè)備IP地址
    :param username: 登錄用戶名
    :param password: 登錄密碼
    """
    # 創(chuàng)建SSH對象
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    # 連接設(shè)備
    ssh.connect(host, username=username, password=password)
    # 執(zhí)行命令，獲取配置文件
    stdin, stdout, stderr = ssh.exec_command('show running-config')
    # 獲取當(dāng)前時間作為備份文件名
    now = datetime.now().strftime("%Y%m%d%H%M%S")
    backup_file = f"{host}_{now}.cfg"
    # 保存配置文件
    with open(backup_file, 'w') as f:
        f.write(stdout.read().decode())
    # 關(guān)閉連接
    ssh.close()
    print(f"設(shè)備 {host} 配置備份完成，備份文件：{backup_file}")
def restore_device_config(host, username, password, backup_file):
    """
    恢復(fù)網(wǎng)絡(luò)設(shè)備配置
    :param host: 設(shè)備IP地址
    :param username: 登錄用戶名
    :param password: 登錄密碼
    :param backup_file: 備份文件路徑
    """
    # 創(chuàng)建SSH對象
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    # 連接設(shè)備
    ssh.connect(host, username=username, password=password)
    # 讀取備份文件內(nèi)容
    with open(backup_file, 'r') as f:
        config = f.read()
    # 分割配置文件，避免命令行長度限制
    config_lines = config.split('\n')
    # 登錄設(shè)備，進(jìn)入特權(quán)模式
    stdin, stdout, stderr = ssh.exec_command('enable')
    # 輸入特權(quán)密碼
    stdin.write(password + '\n')
    stdin.flush()
    # 進(jìn)入配置模式
    stdin, stdout, stderr = ssh.exec_command('configure terminal')
    # 恢復(fù)配置
    for line in config_lines:
        stdin.write(line + '\n')
        stdin.flush()
    # 保存配置
    stdin, stdout, stderr = ssh.exec_command('end')
    stdin, stdout, stderr = ssh.exec_command('write memory')
    # 關(guān)閉連接
    ssh.close()
    print(f"設(shè)備 {host} 配置恢復(fù)完成，使用備份文件：{backup_file}")
if __name__ == '__main__':
    # 設(shè)備信息
    host = '192.168.1.1'
    username = 'admin'
    password = 'admin'
    # 備份配置
    backup_device_config(host, username, password)
    # 恢復(fù)配置
    # backup_file = '192.168.1.1_20210101010101.cfg'
    # restore_device_config(host, username, password, backup_file)