"""
Description
   Verify IPv4 and IPv6 ARP.

   If there is any ARP failure, create a list with all the ip addresses that failed ARP and 
   insert the ipv4/ipv6 obj as the first element in the list.

   This code will loop through all created topologies, device groups that are started/enabled, ethernet, for ipv4 and ipv6.

"""

import os, sys, time, traceback
# Import the RestPy module
from ixnetwork_restpy.testplatform.testplatform import TestPlatform
from ixnetwork_restpy.files import Files
from ixnetwork_restpy.assistants.statistics.statviewassistant import StatViewAssistant

apiServerIp = '192.168.70.3'

# windows|connection_manager|linux
osPlatform = 'windows'

# For Linux API server only
username = 'admin'
password = 'admin'

try:
    testPlatform = TestPlatform(apiServerIp, log_file_name='restpy.log')

    # Console output verbosity: 'none'|request|request_response
    testPlatform.Trace = 'request_response'

    if osPlatform == 'linux':
        testPlatform.Authenticate(username, password)
        session = testPlatform.Sessions.find(Id=1)

    if osPlatform in ['windows', 'connection_manager']:
        # Windows support only one session. Id is always equal 1.
        session = testPlatform.Sessions.find(Id=1)

    # ixNetwork is the root object to the IxNetwork API tree.
    ixNetwork = session.Ixnetwork
    
    arpFailedList = []
    for topology in ixNetwork.Topology.find():
        for deviceGroup in topology.DeviceGroup.find():
            # Verify if the device group is enabled/started
            if deviceGroup.Status == 'started':
                for ethernet in deviceGroup.Ethernet.find(): 

                    ipv4Obj = ethernet.Ipv4.find()
                    if ipv4Obj:
                        currentIpObjList = []
                        # Get the index of the arp failure
                        for index, arpFailed in enumerate(ipv4Obj.SessionInfo):
                            if arpFailed == 'resolveMacFailed':
                                # With the index, get the IP address
                                currentIpObjList.append(ipv4Obj.Address.Values[index])

                        if len(currentIpObjList) > 0:
                            currentIpObjList.insert(0, ipv4Obj)
                            arpFailedList.append(currentIpObjList)

                    ipv6Obj = ethernet.Ipv6.find()
                    if ipv6Obj:
                        currentIpObjList = []
                        # Get the index of the arp failure
                        for index, arpFailed in enumerate(ipv6Obj.SessionInfo):
                            if arpFailed == 'resolveMacFailed':
                                # With the index, get the IP address
                                currentIpObjList.append(ipv6Obj.Address.Values[index])

                        if len(currentIpObjList) > 0:
                            currentIpObjList.insert(0, ipv6Obj)
                            arpFailedList.append(currentIpObjList)


    print('\nARP failures:')
    for eachFailure in arpFailedList:
        print('\t', eachFailure)

except Exception as errMsg:
    print('\nError: %s' % traceback.format_exc())
    print('\nrestPy.Exception:', errMsg)