feat(cli): add flag to configure log level

This commit is contained in:
Cyrille Nofficial 2022-11-09 20:21:03 +01:00
parent befb4bacb3
commit c755d019e8

View File

@ -13,7 +13,6 @@ import paho.mqtt.client as mqtt
from . import depthai as cam # pylint: disable=reimported
logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO)
_DEFAULT_CLIENT_ID = "robocar-depthai"
@ -52,6 +51,10 @@ def _parse_args_cli() -> argparse.Namespace:
parser.add_argument("-W", "--image-width", help="image width",
type=int,
default=_get_env_int_value("IMAGE_WIDTH", 126))
parser.add_argument("--log", help="Log level",
type=str,
default="info",
choices=["info", "debug"])
args = parser.parse_args()
return args
@ -72,9 +75,12 @@ def execute_from_command_line() -> None:
Cli entrypoint
:return:
"""
logging.basicConfig(level=logging.INFO)
args = _parse_args_cli()
if args.log == "info":
logging.basicConfig(level=logging.INFO)
elif args.log == "debug":
logging.basicConfig(level=logging.DEBUG)
client = _init_mqtt_client(broker_host=args.mqtt_broker_host,
broker_port=args.mqtt_broker_port,