import paramiko
import time
import csv
import os
# 確認(rèn)當(dāng)前腳本所在目錄，并構(gòu)建ips.csv文件的路徑
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
IPS_FILE = os.path.join(BASE_DIR, "ips.csv")
# 讀取IP地址列表，忽略空行和#開頭的行，并去除重復(fù)的IP地址
ip = []
print ("正在讀取IP配置文件...")
with open(IPS_FILE, "r", encoding="utf-8") as f:
    reader = csv.reader(f)
    for row in reader:
        if not row or row[0].strip().startswith("#"):
            continue
        ip.append(row[0].strip())
set_ip = set(ip)
if len(set_ip) == len(ip):
    print ("讀取到的IP地址列表,共", len(ip), "個(gè)IP地址,沒有重復(fù)的IP地址")
else:
    count_duplicates = len(ip) - len(set_ip)
    ip = list(set_ip)
    print (f"讀取到的IP地址列表,共{len(ip)}個(gè)IP地址,有{count_duplicates}個(gè)重復(fù)的IP地址,已自動(dòng)去重")
username = "hao"
password = "admin12345"
ssh_client = paramiko.SSHClient ()
ssh_client.set_missing_host_key_policy (paramiko.AutoAddPolicy ())
for i in ip:
    print ("正在連接至", i)
    ssh_client.connect (
        hostname=i,username=username,password=password,
        look_for_keys=False,allow_agent=False
        )
    command = ssh_client.invoke_shell ()
    import re
    time.sleep(1.5)
    output = command.recv(65535).replace(b'\x00',b'').decode('ASCII').replace('\r\n','').strip()
    match = re.search(r'[<\[](.*?)[>\]]',output,re.M)
    sysname = match.group(1) if match else "未知系統(tǒng)"
    print ("你已成功連接至", i, "系統(tǒng)名稱:", sysname)
    #向設(shè)備發(fā)送指令，可自行編輯修改你要發(fā)送的命令
    time.sleep (0.1)
    command.send ("sys\n")
    command.send ("dis version\n")
    time.sleep (0.5)
    output_text = command.recv(65535).decode('ASCII').replace('\r','').replace('\x00','')
    print (output_text)
    # 保存回顯結(jié)果到TXT文件
    filename = os.path.join(BASE_DIR, f"[{sysname}]_{i}_{time.strftime('%Y%m%d%H%M%S')}_output.txt")
    with open(filename, 'w', encoding='utf-8') as f:
        f.write(f"交換機(jī)IP: {i} 系統(tǒng)名稱:{sysname}\n")
        f.write("=" * 50 + "\n")
        f.write(output)
        f.write(output_text)
        f.write("\n" + "=" * 50 + "\n")
        f.write(f"檢查時(shí)間: {time.strftime('%Y-%m-%d %H:%M:%S')}")
    print(f"回顯結(jié)果已保存到文件: {filename}")
    ssh_client.close ()
    print ("已關(guān)閉", i, "的連接")
    print ("---------------------------------------------------------------")
print  ("================================================================")
print ("檢查結(jié)束")