DNA-center-API.py

import sys
import json
import requests
import os
from requests.auth import HTTPBasicAuth
requests.packages.urllib3.disable_warnings()

DNAC=os.environ.get('DNAC', input("DNA CENTER FQDN:", ))
DNAC_PORT=os.environ.get('DNAC_PORT', input("DNA API port: ", ))
DNAC_USER=os.environ.get('DNAC_USER', input("DNA username: ", ))
DNAC_PASSWORD=os.environ.get('DNAC_PASSWORD',input("DNA passord: ", ))

# -------------------------------------------------------------------
# Helper functions
# -------------------------------------------------------------------
def get_auth_token(controller_ip=DNAC, username=DNAC_USER, password=DNAC_PASSWORD):
    """ Authenticates with controller and returns a token to be used in subsequent API invocations
    """

    login_url = "https://{0}:{1}/dna/system/api/v1/auth/token".format(controller_ip, DNAC_PORT)
    result = requests.post(url=login_url, auth=HTTPBasicAuth(DNAC_USER, DNAC_PASSWORD), verify=False)
    result.raise_for_status()

    token = result.json()["Token"]
    return {
        "controller_ip": controller_ip,
        "token": token
    }

def create_url(path, controller_ip=DNAC):
    """ Helper function to create a DNAC API endpoint URL
    """

    return "https://%s:%s/dna/intent/api/v1/%s" % (controller_ip, DNAC_PORT, path)

def get_url(url):

    url = create_url(path=url)
    print(url)
    token = get_auth_token()
    headers = {'X-auth-token' : token['token']}
    try:
        response = requests.get(url, headers=headers, verify=False)
    except requests.exceptions.RequestException as cerror:
        print("Error processing request", cerror)
        sys.exit(1)

    return response.json()

def list_network_devices():
    return get_url("network-device")

def ip_to_id(ip):
    return get_url("network-device/ip-address/%s" % ip)['response']['id']

def get_modules(id):
       return get_url("network-device/module?deviceId=%s" % id)

def print_info(modules):
    print("{0:30}{1:15}{2:25}{3:5}".format("Module Name","Serial Number","Part Number","Is Field Replaceable?"))
    for module in modules['response']:
        print("{moduleName:30}{serialNumber:15}{partNumber:25}{moduleType:5}".format(moduleName=module['name'],
                                                           serialNumber=module['serialNumber'],
                                                           partNumber=module['partNumber'],
                                                           moduleType=module['isFieldReplaceable']))

def printdevices(response):
    print("{0:42}{1:17}{2:12}{3:18}{4:12}{5:16}{6:15}".
       format("hostname","mgmt IP","serial",
       "platformId","SW Version","role","Uptime"))
    for device in response['response']:
       uptime = 'N/A' if device['upTime'] is None else device['upTime']
       print("{0:42}{1:17}{2:12}{3:18}{4:12}{5:16}{6:15}".
           format(str(device['hostname']),
                  str(device['managementIpAddress']),
                   str(device['serialNumber']),
                   str(device['platformId']),
                   str(device['softwareVersion']),
                   str(device['role']), str(uptime)))

def devices_module(response):
    for ip in response['response']:
        device_id=ip_to_id(ip['managementIpAddress'])
        print(device_id)
        modules=get_modules(device_id)
        print_info(modules)
    return

if __name__ == '__main__':
    response = list_network_devices()
    devices_module(response)
    printdevices(response)