netbox-scanner.py

#!/usr/bin/env python3

import logging
import os
import sys
import docker
import ipaddress

from configparser import ConfigParser
from argparse import ArgumentParser
from os.path import expanduser, isfile
from datetime import datetime
from urllib3 import disable_warnings
from urllib3.exceptions import InsecureRequestWarning

from nbs import NetBoxScanner

argument = str(sys.argv[1])

if argument == 'nmap':
    from nbs.nmap import Nmap
if argument == 'netxms':
    from nbs.netxms import NetXMS
if argument == 'prime':
    from nbs.prime import Prime


local_config = expanduser('~/.netbox-scanner.conf')
global_config = '/opt/netbox/netbox-scanner.conf'
dir_config = './netbox-scanner.conf'
config = ConfigParser()

if isfile(local_config):
    config.read(local_config)
elif isfile(global_config):
    config.read(global_config)
elif isfile(dir_config):
    config.read(dir_config)
else:
    raise FileNotFoundError('Configuration file was not found.')

netbox = config['NETBOX']
tag = 'netbox-scanner'
cleanup = False
if argument == 'nmap':
    nmap = config['NMAP']
    tag = nmap['tag']
    cleanup = nmap.getboolean('cleanup')
if argument == 'netxms':
    netxms = config['NETXMS']
    tag = netxms['tag']
if argument == 'prime':
    prime = config['PRIME']
    tag = prime['tag']
# if argument == 'docker':
#     dockerConf = config['Docker']
#     tag = dockerConf['tag']
#     cleanup = dockerConf.getboolean('cleanup')

parser = ArgumentParser(description='netbox-scanner')
subparsers = parser.add_subparsers(title='Commands', dest='command')
subparsers.required = True
if argument == 'nmap':
    argsp = subparsers.add_parser('nmap', help='Nmap module')
if argument == 'netxms':
    argsp = subparsers.add_parser('netxms', help='NetXMS module')
if argument == 'prime':
    argsp = subparsers.add_parser('prime', help='Cisco Prime module')
if argument == 'docker':
    argsp = subparsers.add_parser('docker', help='Docker module')
args = parser.parse_args()

if not os.path.isdir(netbox['logs']):
    os.makedirs(netbox['logs'])
logfile = f'{netbox["logs"]}/netbox-scanner-{datetime.now().isoformat()}.log'
logging.basicConfig(
    filename=logfile,
    level=logging.INFO,
    format='%(asctime)s\tnetbox-scanner\t%(levelname)s\t%(message)s'
)
logging.getLogger().addHandler(logging.StreamHandler())

# useful if you have tls_verify set to no
disable_warnings(InsecureRequestWarning)

with open(nmap['networks'], 'r') as file:
    networks = file.readlines()

def cmd_nmap(s):  # nmap handler
    h = Nmap(nmap['unknown'], networks)
    h.run()
    s.sync(h.hosts)


def cmd_netxms(s):  # netxms handler
    h = NetXMS(
        netxms['address'],
        netxms['username'],
        netxms['password'],
        netxms.getboolean('tls_verify'),
        netxms['unknown']
    )
    h.run()
    s.sync(h.hosts)


def cmd_prime(s):  # prime handler
    h = Prime(
        prime['address'],
        prime['username'],
        prime['password'],
        prime.getboolean('tls_verify'),
        prime['unknown']
    )
    h.run()  # set access_point=True to process APs
    s.sync(h.hosts)

def cmd_docker(s):

    if not config.has_section('dockerdef'):
        config.add_section('dockerdef')
    dockerDef = config['dockerdef'];
    dockerDef.setdefault('cluster_type', 'Docker')
    dockerDef.setdefault('cluster_prefix', 'Docker')
    dockerDef.setdefault('vrf_prefix', 'Docker')
    scanner.init_docker(dockerDef)

    for s in config.sections():
        if not s.startswith('Docker:'):
            continue

        dockerSection = dict(config.items(s))
        scanner.sync_docker(dockerSection, dockerDef)

if __name__ == '__main__':
    scanner = NetBoxScanner(
        netbox['address'],
        netbox['token'],
        netbox['tls_verify'],
        tag,
        cleanup
    )

    if args.command == 'nmap':
        cmd_nmap(scanner)
    elif args.command == 'netxms':
        scanner.tag = 'netxms'
        scanner.cleanup = netxms.getboolean('cleanup')
        cmd_netxms(scanner)
    elif args.command == 'prime':
        scanner.tag = prime['tag']
        scanner.cleanup = prime.getboolean('cleanup')
        cmd_prime(scanner)
    elif args.command == 'docker':
        cmd_docker(scanner)

    exit(0)