tyler.durden/netbox-scanner
1
0
Fork 0
mirror of https://github.com/lopes/netbox-scanner.git synced 2025-07-22 05:14:39 +02:00
Code Issues Packages Projects Releases Wiki Activity

use arg passed in to determine what importers to run

Browse Source
This commit is contained in:
bile0026 2020-11-23 21:45:25 -06:00
parent 25f17b1101
commit b09feca209
1 changed files with 23 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
netbox-scanner.py
View File

@ -1,6 +1,7 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import logging import logging
import sys
from configparser import ConfigParser from configparser import ConfigParser
from argparse import ArgumentParser from argparse import ArgumentParser
@ -10,9 +11,15 @@ from urllib3 import disable_warnings
from urllib3.exceptions import InsecureRequestWarning from urllib3.exceptions import InsecureRequestWarning
from nbs import NetBoxScanner from nbs import NetBoxScanner
from nbs.nmap import Nmap
from nbs.netxms import NetXMS argument = str(sys.argv[1])
from nbs.prime import Prime
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') local_config = expanduser('~/.netbox-scanner.conf')
@ -27,16 +34,22 @@ else:
raise FileNotFoundError('Configuration file was not found.') raise FileNotFoundError('Configuration file was not found.')
netbox = config['NETBOX'] netbox = config['NETBOX']
nmap = config['NMAP'] if argument == 'nmap':
#netxms = config['NETXMS'] nmap = config['NMAP']
#prime = config['PRIME'] if argument == 'netxms':
netxms = config['NETXMS']
if argument == 'prime':
prime = config['PRIME']
parser = ArgumentParser(description='netbox-scanner') parser = ArgumentParser(description='netbox-scanner')
subparsers = parser.add_subparsers(title='Commands', dest='command') subparsers = parser.add_subparsers(title='Commands', dest='command')
subparsers.required = True subparsers.required = True
argsp = subparsers.add_parser('nmap', help='Nmap module') if argument == 'nmap':
#argsp = subparsers.add_parser('netxms', help='NetXMS module') argsp = subparsers.add_parser('nmap', help='Nmap module')
#argsp = subparsers.add_parser('prime', help='Cisco Prime module') if argument == 'netxms':
argsp = subparsers.add_parser('netxms', help='NetXMS module')
if argument == 'prime':
argsp = subparsers.add_parser('prime', help='Cisco Prime module')
args = parser.parse_args() args = parser.parse_args()
logfile = '{}/netbox-scanner-{}.log'.format( logfile = '{}/netbox-scanner-{}.log'.format(
@ -50,6 +63,7 @@ logging.basicConfig(
) )
logging.getLogger().addHandler(logging.StreamHandler()) logging.getLogger().addHandler(logging.StreamHandler())
# useful if you have ssl_verify set to no
disable_warnings(InsecureRequestWarning) disable_warnings(InsecureRequestWarning)