##使用import语句导入paramiko模块、time模块和getpass模块
import paramiko
import time
#方便将getpass.getpass("password: ")简写成getpass("password: ")
from getpass import getpass
import sys
import socket

#使用input()函数以及getpass模块实现交互式输入用户名和密码
username = input("Username: ")
password = getpass("password: ")
#获取运行 Python 程序的命令行的第一个参数(在这里具体指python3.8后面的ip_list.txt)
ip_file = sys.argv[1]
#获取运行 Python 程序的命令行的第二个参数(在这里具体指python3.8后面的cmd.txt)
cmd_file = sys.argv[2]

#创建名为switch_with_authentication_issue和switch_not_reachable的两个空列表
switch_with_authentication_issue = []
switch_not_reachable = []

#以只读的权限读取名为ip_list.txt的文件并将文件内容赋值给变量iplist
iplist = open(ip_file)
#一行一行读取ip_list.txt文件内容并将其赋值给变量line
for line in iplist.readlines():
#使用try...except...语句进行异常处理
    try:
#将ip进行去除首尾空行处理
        ip = line.strip()
#调用Paramiko的SSHClient()方法，将其赋值给变量ssh_client
        ssh_client = paramiko.SSHClient()
#使用ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())来让Paramiko接受服务端提供的公钥
        ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
#调用paramiko.SSHClient()的函数connect（）进行SSH登录，look_for_keys=False用来解决避免Paramiko使用已存在密钥来登录设备
        ssh_client.connect(hostname=ip, username=username, password=password, look_for_keys=False)
#打印提示信息
        print("--------------------------------------------------------")
        print(f"You have successfully connect to {ip}!!!")
#调用Paramiko的SSHClient()的invoke_shell()方法来唤醒shell,即设备命令行
        command = ssh_client.invoke_shell()
#设置屏显长度为0
        command.send('screen-length 0 temporary\n')
#以只读的权限读取名为cmd.txt的文件并将文件内容赋值给变量cmmand
        cmdlist = open(cmd_file)
#将文件指针移到开头位置
        cmdlist.seek(0)
#一行一行读取cmd.txt文件内容并将其赋值给变量line
        for line in cmdlist.readlines():
#执行cmd.txt文件里命令
            command.send(line + "\n")
#调用time模块下的sleep()函数手动让Python休眠2秒
        time.sleep(2)
#关闭文件
        cmdlist.close()
#设置命令行最大回显内容数并将其赋值给变量output并用decode("ascii")将其解析为ASCII编码
        output = command.recv(65535).decode('ascii')
        print(output)
#若符合认证错误则执行except语句
    except paramiko.ssh_exception.AuthenticationException:
#打印认证失败交换机的IP
        print(f"User authentication failed ip is: {ip}.")
#将认证失败交换机的IP添加到名为switch_with_authentication_issue的空列表中
        switch_with_authentication_issue.append(ip)
#若符合目的不可达错误则执行except语句
    except socket.error:
#打印目的不可达的IP
        print(f"{ip} is not reachable!")
#将目的不可达的IP添加到名为switch_not_reachable的列表
        switch_not_reachable.append(ip)
#关闭文件ip_list.txt文件
iplist.close()
#断开SSH连接
ssh_client.close()
#一个一个打印认证失败设备交换机的全部IP
print("----------------------------------------------")
print("User authentication failed for below switches: ")
for i in switch_with_authentication_issue:
    print(i)
#一个一个打印目的不可达设备交换机的全部IP
print("----------------------------------------------")
print("Below switches are not reachable: ")
for i in switch_not_reachable:
    print(i)