robocars/robocar-oak-camera
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat(cli): add flag to configure log level

Browse Source
This commit is contained in:
Cyrille Nofficial 2022-11-09 20:21:03 +01:00
parent befb4bacb3
commit c755d019e8
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
camera/cli.py
View File

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