#通过import语句导入paramiko、time、re、socket、getpass、datetime模块 
import paramiko
import time
import re
import socket
from getpass import getpass
from datetime import datetime

#交互的模式下输入用户名和密码
username = input("Username: ")
password = getpass("password: ")
#调用datetime.now()记录当前时间
now_time = datetime.now()
#调用datetime.now()下面的子方法.month()、.day()、.year()以字符串的形式展示并将其赋值给变量date，%s 可以将整数转成字符串
date = "%s-%s-%s" % (now_time.month,now_time.day,now_time.year)
#调用datetime.now()下面的子方法.hour()、.minute()、.second()以字符串的形式展示并将其赋值给变量time_now，(%在python的格式化输出，有转换字符的作用)
time_now = "%s:%s:%s" % (now_time.hour,now_time.minute,now_time.second)

#创建两个分别代表目的不可达、认证失败的空列表
switch_with_tacacs_issue = []
switch_not_reachable = []
#设置总端口数的初始值为0
total_number_of_up_port = 0

#以只读的形式打开名为eachable_ip.txt的文件并将其内容赋值给变量iplist
iplist = open('reachable_ip.txt', 'r')
#计算变量iplist的长度并将其赋值给变量number_of_switch
number_of_switch = len(iplist.readlines())
#计算所有交换机所有的端口号
total_number_of_ports = number_of_switch * 48

#将指针设置到开头处
iplist.seek(0)
#将变量里面的内容一一赋值给变量line
for line in iplist:
    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(f"You have successfully connect to {ip}!")
#调用Paramiko的SSHClient()的invoke_shell()方法来唤醒shell,即设备命令行
        command = ssh_client.invoke_shell()
#SSH登录成功后输入特权密码查看端口信息
        command.send("su\n")
        command.send(password + "\n")
        command.send("dis ip int bri | in up\n")
#调用time模块下的sleep()函数手动让Python休眠1秒
        time.sleep(1)
#设置命令行最大回显内容数并将其赋值给变量output并用decode("ascii")将其解析为ASCII编码
        output = command.recv(65535).decode('ascii')
        print(output)
#通过正则表达式findall()方法来查看UP的G口
        search_up_port = re.findall(r'GigabitEthernet', output)
#计算UP的G口的数量并将其赋值给变量number_of_up_port
        number_of_up_port = len(search_up_port)
#打印UP的G口的数量的提示信息
        print(f"{ip} has {int(number_of_up_port)} ports up.")
#每次查找UP的G口并统计，total_number_of_up_port = number_of_up_port + total_number_of_up_port
        total_number_of_up_port += number_of_up_port
#当不符合try语句则执行except语句，这里根据认证失败来执行except下面的语句
    except paramiko.ssh_exception.AuthenticationException:
#打印某个IP认证失败信息
        print(f"TACACS is not working for {ip}!")
#将认证失败的IP添加到一个描述认证失败的列表当中
        switch_with_tacacs_issue.append(ip)
#当不符合try语句则执行except语句，这里根据目的不可达来执行except下面的语句
    except socket.error:
#打印某个IP目的不可达信息
        print(f"{ip} is not reachable.")
#将目的不可达的IP添加到一个描述目的不可达的列表当中
        switch_not_reachable.append(ip)
#关闭文件
iplist.close()

#打印认证的IP
print("--------------------------------------")
print(f"There are totally {int(total_number_of_ports)} ports available in the network.")
print(f"{int(total_number_of_up_port)} ports are currently up.")
print(f"\nport up rate is {int(total_number_of_up_port) / int(total_number_of_ports) * 100} ")
print("\nTACACS is not working for below switches: ")
for i in switch_with_tacacs_issue:
    print(i)
#打印目的不可达的IP
print("\nBelow switches are not reachable: ")
for i in switch_not_reachable:
    print(i)
#以附加的模式打开名为date.txt的文件
f = open(date + ".txt", "a+")
#在文件填写记录的时间
f.write(f"As of {date} {time_now}")
#将上述信息一一写入文件中
f.write(f"\n\nThere are totally {int(total_number_of_ports)} ports available in the network.")
f.write(f"\n{int(total_number_of_up_port)} ports are currently up.")
f.write(f"\nport up rate is {int(total_number_of_up_port) / int(total_number_of_ports) * 100} ")
f.write("\n--------------------------------------------------\n\n")
#关闭文件
f.close()