summaryrefslogtreecommitdiff
path: root/netdisco/__main__.py
blob: 64037f0fd0b39f9b3438b372ca57f76be68375c5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
"""Command line tool to print discocvered devices or dump raw data."""
from pprint import pprint
import sys

from netdisco.discovery import NetworkDiscovery


def main():
    """Handle command line execution."""
    netdisco = NetworkDiscovery()

    netdisco.scan()

    print("Discovered devices:")
    count = 0
    for dev in netdisco.discover():
        count += 1
        print('{}:'.format(dev))
        pprint(netdisco.get_info(dev))
        print()
    print("Discovered {} devices".format(count))

    # Pass in command line argument dump to get the raw data
    if sys.argv[-1] == 'dump':
        print()
        print()
        print("Raw Data")
        print()
        netdisco.print_raw_data()

    netdisco.stop()


if __name__ == '__main__':
    main()