1 files changed, 9 insertions, 15 deletions
diff --git a/test/vcpe/get_info.py b/test/vcpe/get_info.py
index 2d52a07b7..f86cfb204 100755
--- a/test/vcpe/get_info.py
+++ b/test/vcpe/get_info.py
@@ -1,26 +1,20 @@
 #!/usr/bin/env python
 
-import time
 import logging
 import json
-import mysql.connector
-import ipaddress
-import re
-import sys
-import base64
-from vcpecommon import *
-import preload
-import vcpe_custom_service
+from vcpecommon import * # pylint: disable=W0614
+import argparse
 
+# Run the script with [-h|--help] to get usage info
 
 logging.basicConfig(level=logging.INFO, format='%(message)s')
 
-vcpecommon = VcpeCommon()
+parser = argparse.ArgumentParser(formatter_class=
+                                 argparse.ArgumentDefaultsHelpFormatter)
+parser.add_argument('--config',help='Configuration file path',default=None)
+args = parser.parse_args()
+
+vcpecommon = VcpeCommon(cfg_file=args.config)
 nodes=['brg', 'bng', 'mux', 'dhcp']
 hosts = vcpecommon.get_vm_ip(nodes)
 print(json.dumps(hosts, indent=4, sort_keys=True))
-
-
-
-
-