From d24ea744f23f84a65bb63952806d098477a0ec9b Mon Sep 17 00:00:00 2001
From: Chuck SmileyRekiere <chuck.smileyrekiere@candelatech.com>
Date: Fri, 1 Jul 2022 16:36:50 -0600
Subject: [PATCH] lf_create_rig_json.py : added logger

Signed-off-by: Chuck SmileyRekiere <chuck.smileyrekiere@candelatech.com>
---
 .../tools/ct_rig_json/lf_create_rig_json.py   | 42 +++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/py-scripts/tools/ct_rig_json/lf_create_rig_json.py b/py-scripts/tools/ct_rig_json/lf_create_rig_json.py
index 961b5c25..fed8b76e 100644
--- a/py-scripts/tools/ct_rig_json/lf_create_rig_json.py
+++ b/py-scripts/tools/ct_rig_json/lf_create_rig_json.py
@@ -5,6 +5,22 @@ Usage: lf_create_rig_json.py --lf_mgr <lanforge ip> --lf_mgr_port <lanforge port
 '''
 
 import argparse
+import logging
+import importlib
+import os
+import sys
+
+if sys.version_info[0] != 3:
+    print("This script requires Python 3")
+    exit(1)
+
+
+sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../../../")))
+
+
+logger = logging.getLogger(__name__)
+lf_logger_config = importlib.import_module("py-scripts.lf_logger_config")
+
 
 
 class lf_create_rig_json():
@@ -99,8 +115,32 @@ Usage: lf_create_rig_json.py ----lf_mgr <lanforge ip> --lf_mgr_port <lanforge po
     parser.add_argument('--upstream_port', help='--upstream_port <1.1.eth2> need to include self and resource', default='1.1.eth2')
     parser.add_argument('--test_timeout', help='--test_timeout 600', default='600')
 
+    parser.add_argument('--log_level',
+                        default=None,
+                        help='Set logging level: debug | info | warning | error | critical')
+
+    # logging configuration
+    parser.add_argument(
+        "--lf_logger_config_json",
+        help="--lf_logger_config_json <json file> , json configuration of logger")
+
+
     args = parser.parse_args()
 
+    # set up logger
+    logger_config = lf_logger_config.lf_logger_config()
+
+    # set the logger level to debug
+    if args.log_level:
+        logger_config.set_level(level=args.log_level)
+
+    # lf_logger_config_json will take presidence to changing debug levels
+    if args.lf_logger_config_json:
+        # logger_config.lf_logger_config_json = "lf_logger_config.json"
+        logger_config.lf_logger_config_json = args.lf_logger_config_json
+        logger_config.load_lf_logger_config()
+
+
     if args.test_server is None:
         _test_server = args.lf_mgr
     else:
@@ -130,6 +170,8 @@ Usage: lf_create_rig_json.py ----lf_mgr <lanforge ip> --lf_mgr_port <lanforge po
                                   _test_timeout=_test_timeout)
     rig_json.create()
 
+    logger.info("created {file}".format(file=_file))
+
 
 if __name__ == '__main__':
     main()