#!/bin/env python
import argparse
import sys
import cisco
IP = 'IP-Address'
MAC = 'Mac-Address'
Vlan = 'Vlan'
If = 'Interface'
Desc = 'Description'
IP_info = {IP:'None', MAC:'None', Vlan:'None', If:'None', Desc:'None'}

def get_ARP_Table(ipaddr):
    arpCmd = 'sh ip arp ' + ipaddr
    arpCmdResult = CLI(arpCmd, False)
        # CLI 명령 시, 매개변수2에 False를 지정하지 않으면 화면에 해당 명령 결과값이 함께 출력된다.
        # Default 값은 True.
    arpCmdResultList = arpCmdResult.get_output()
        # CLI의 결과값을 cisco.cli.CLI type에서 각 Line별 값으로 List Type으로 변경
    for arp in arpCmdResultList:
        if (-1<arp.find(args.ip)):  # 확인하고자 하는 IP값이 존재하는 ARP Table 확인
            return arp             # 요청한 IP 정보가 있는 List의 문자열을 리턴
    else:
        print ' %s : Not found IP Address Infomation' % args.ip # 만약 ARP Table에 존재하지 않으면 종료
        sys.exit()

def get_IP_MAC_info(info):
    info_list = info.split()            # ARP Table의 문자열 데이터를 분리
    IP_info[IP] = info_list[0]          # 첫번째 항목이 IP
    IP_info[MAC] = info_list[2]         # 세번째 항목이 MAC
    IP_info[Vlan] = info_list[3][4:]    # 네번째 항목이 VLAN정보이나,
                                        # 앞의 'VLAN'글자를 빼고 출력하기 위해 4번째부터 저장

def get_Interface_info():
    macCmd = 'sh mac address-table addr ' + IP_info[MAC]
    macCmdResult = CLI(macCmd, False)
    macCmdResultList = macCmdResult.get_output()
    if (len(macCmdResultList) > 6):             # Mac Address Table이 존재하지 않을 경우에는 1리턴, 존재시 7이상 리턴.(결과값라인)
        IP_info[Int] = macCmdResultList[5][58:] # Interface 정보만 가져옴.
        get_Description_info(IP_info[Int])      # Interface정보가 있는 경우에는 해당 Interface의 Desc를 가져옴.

def get_Description_info(iInfo):
    if(iInfo.find('Eth') == 0 or iInfo.find('Po')==0): # Eht나 Po인 경우에만 Desc확인
        intCmd = 'sh int desc | inc ' + iInfo
        intCmdResult = CLI(intCmd, False)
        intCmdResultList = intCmdResult.get_output()
        IP_info[Desc] = intCmdResultList[0][25:]

def show_IP_info():
    print '================================================'
    print '             IP Info : NetworkZIGI              '
    print '================================================'
    print '%-15s : %s' % (IP,IP_info[IP])
    print '%-15s : %s' % (MAC,IP_info[MAC])
    print '%-15s : %s' % (Vlan, IP_info[Vlan])
    print '%-15s : %s' % (Int, IP_info[Int])
    print '%-15s : %s' % (Desc,IP_info[Desc])

parser = argparse.ArgumentParser('Args',description='Args Desc')
parser.add_argument('ip')
args = parser.parse_args()

iparp = get_ARP_Table(args.ip)
get_IP_MAC_info(iparp)
get_Interface_info()
show_IP_info()