mirror of
https://github.com/lopes/netbox-scanner.git
synced 2025-07-21 04:44:37 +02:00
use arg passed in to determine what importers to run
This commit is contained in:
parent
25f17b1101
commit
b09feca209
@ -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)
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user