use arg passed in to determine what importers to run

This commit is contained in:
bile0026 2020-11-23 21:45:25 -06:00
parent 25f17b1101
commit b09feca209

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,8 +11,14 @@ from urllib3 import disable_warnings
from urllib3.exceptions import InsecureRequestWarning from urllib3.exceptions import InsecureRequestWarning
from nbs import NetBoxScanner from nbs import NetBoxScanner
argument = str(sys.argv[1])
if argument == 'nmap':
from nbs.nmap import Nmap from nbs.nmap import Nmap
if argument == 'netxms':
from nbs.netxms import NetXMS from nbs.netxms import NetXMS
if argument == 'prime':
from nbs.prime import Prime from nbs.prime import Prime
@ -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']
if argument == 'nmap':
nmap = config['NMAP'] nmap = config['NMAP']
#netxms = config['NETXMS'] if argument == 'netxms':
#prime = config['PRIME'] 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
if argument == 'nmap':
argsp = subparsers.add_parser('nmap', help='Nmap module') argsp = subparsers.add_parser('nmap', help='Nmap module')
#argsp = subparsers.add_parser('netxms', help='NetXMS module') if argument == 'netxms':
#argsp = subparsers.add_parser('prime', help='Cisco Prime module') 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)