#batch processing with seed file
from netmiko import ConnectHandler
from netmiko.ssh_exception import  NetMikoTimeoutException
from paramiko.ssh_exception import SSHException
from netmiko.ssh_exception import  AuthenticationException
import datetime
import os
import time
import getpass



#delete Result_ISE_Port_Status.csv file

if os.path.exists("IP.txt"):
  pass
else:
    print("IP.txt file NOT FOUND!\n")
    input("Press Any Key to Exit...")
    os._exit(0)
try:
    if os.path.exists("Result_ISE_Port_Status.csv"):
        os.remove("Result_ISE_Port_Status.csv")
except:
    input("Close the Result_ISE_Port_Status.csv file and relaunch the program.")
    os._exit(0)
    
rowtotal = 0
#GET TOTAL ROW
list = open('IP.txt')
for data in list:
    rowtotal += 1  
list.close()

seed = open('IP.txt')

pushlog = open("Result_ISE_Port_Status.csv","a+")
pushlog.write("Hostname,IP,Source_Int,ISE_OPEN?\n")
pushlog.close()
def my_account():
    global mgmt_int
    device=ConnectHandler(device_type="cisco_ios", ip=IP, username=user_login, password=user_password)
    output=device.send_command("show run | in hostname")
    output=output.split(" ")
    hostname=output[1]
    start_time = time.time()
    print("Connection established!\n")
    print ("    Hostname :",hostname)
    print ("   Logged-in :",user_login)
    pushlog = open("Result_ISE_Port_Status.csv","a+")
    pushlog.write("%s,%s," % (hostname,IP.strip()))
    pushlog.close()
    showarp = device.send_command("show arp | in %s_" % IP.strip())
    showarp.strip()
    if showarp == "":
        showroute = device.send_command("show ip route | in %s" % IP.strip())
        showroute = showroute.split("connected,")
        mgmt_int = showroute[1]
    else:
        showarp = showarp.split("ARPA")
        mgmt_int = showarp[1]    
    mgmt_int = mgmt_int.strip()
    print("  Source Int : %s" % mgmt_int)
    showvlan = device.send_command('telnet %s 49 /so %s' % (ISE_IP.strip(),mgmt_int),expect_string=r"Open|#")
    if "Open" in showvlan:
        pushlog = open("Result_ISE_Port_Status.csv","a+")
        pushlog.write("%s,Yes\n" % mgmt_int)
        pushlog.close()
        print("    ISE Port : Open\n")
        device.send_command_timing('\x036')
        print("Terminating Connection to %s tcp 49..." % ISE_IP.strip())
        device.send_command("disco 1",expect_string="[confirm]")
        device.send_command("y")
        print("Connection Successfully Terminated!")
    else:
        print("    ISE Port : Closed\n")
        pushlog = open("Result_ISE_Port_Status.csv","a+")
        pushlog.write("%s,No\n" % mgmt_int)
        pushlog.close()
    print("Elapsed Time : {} seconds".format(time.time() - start_time))
    device.disconnect()



#MAIN PROGRAM
print ("\n********************************************************************")
print ("                  GLOBAL IT : AAA PreCheck Script")
print ("********************************************************************\n")
#
user_login = input("     Username: ")
while (user_login =="" ):
    user_login = input("     Username: ")
user_password = getpass.getpass(prompt="     Password: ")
#
ISE_IP = input("       ISE IP: ")
while (ISE_IP==""):
    ISE_IP = input("       ISE IP: ")
   

x = datetime.datetime.now()
print ("\nTOTAL DEVICES TO BE CHECKED :  {}                    Date : {}".format(rowtotal,x.strftime("%x")))
rowcount = 0
for IP in seed:
    rowcount += 1
    completionRate = (rowcount / rowtotal) * 100
    print ('--------------------------------------------------------------------')
    print("Checking : {} of {}                          Progress  : {:.0f} percent".format(rowcount,rowtotal,completionRate))
    print ('\nIP :' + (IP.strip()) + "\n")
    try:
        my_account()
    except (AuthenticationException):
        print('Authentication Failure!')
        pushlog = open("Result_ISE_Port_Status.csv","a+")
        pushlog.write("Login_Error_AuthFail,%s\n" % (IP.strip()))
        pushlog.close()
    except (NetMikoTimeoutException):
        print('\n' + 'Timeout to device: ' + (IP))
        pushlog = open("Result_ISE_Port_Status.csv","a+")
        pushlog.write("Login_Error_Timeout,%s\n" % (IP.strip()))
        pushlog.close()
    except:
        pass
    print ('--------------------------------------------------------------------')
   
print("Runtime: {}".format(datetime.datetime.now()   - x))
input("Press Any Key to Exit...")