From b09feca2094065b82f42f1398b9dac61ba43143f Mon Sep 17 00:00:00 2001
From: bile0026 <zacharybiles@gmail.com>
Date: Mon, 23 Nov 2020 21:45:25 -0600
Subject: [PATCH] use arg passed in to determine what importers to run

---
 netbox-scanner.py | 32 +++++++++++++++++++++++---------
 1 file changed, 23 insertions(+), 9 deletions(-)

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