19

from pvl.snmp import snmp, lldp

    21

import logging; log = logging.getLogger('pvl.hosts-lldp')

    22

import optparse

    32

        host_snmp = host.extensions.get('snmp')

    35

        if not host_snmp :

    36

            log.debug("%s: skip non-snmp host", host)

    39

        elif host.down :

    40

            log.debug("%s: skip down host", host)

    41

            continue

    42

    43

        else :

    44

            log.debug("%s: %s", host, host_snmp)

    45

    46

        agent = lldp.LLDPAgent.apply(options, host.fqdn(), community=host_snmp.get('community'))

    49

            local = agent.local

    50

        except snmp.SNMPError as ex :

    54

        log.info("%s: %s", host, local)

    55

    59

        yield host, agent

    65

    66

    _hosts_lldp = list(hosts_lldp(options, hosts))

    67

    hosts_by_chassis = { }

    68

    69

    # first pass to discover hosts

    70

    for host, agent in _hosts_lldp :

    71

        chassis = agent.local['chassis']

    72

        log.info("%s: %s", host, chassis)

    74

        hosts_by_chassis[chassis] = host

    75

    76

    # second pass to discver links

    77

    for host, agent in _hosts_lldp :

    78

        for port, remote in agent.remotes() :

    79

            port = agent.port(port)

    80

    81

            remote_chassis = remote['chassis']

    82

            remote_host = hosts_by_chassis.get(remote_chassis)

    84

            log.info("%s: %s: %s (%s)", host, port, remote, remote_host)

    86

            yield host, merge(agent.local, port), remote, remote_host

    96

    parser.add_option_group(pvl.snmp.snmp.options(parser))

   106

    for host, local, remote, remote_host in apply_hosts_lldp(options, hosts) :

   107

        if remote_host :

   108

            print "{host:30} {local[port]:25} <-> {remote[port]:25} {remote_host:30}".format(host=host, local=local, remote=remote, remote_host=remote_host)

   109

        else :

   110

            print "{host:30} {local[port]:25} <-- {remote[port]:25} # {remote[chassis]} ({remote[sys_name]})".format(host=host, local=local, remote=remote)