From bb4b8dceaa295aa790bc642c4ecbd07ac6b88919 Mon Sep 17 00:00:00 2001 From: Rasmus Wriedt Larsen Date: Thu, 23 Jul 2020 13:38:58 +0200 Subject: [PATCH] Python: CG trace: More robust logging setup --- .../recorded-call-graph-metrics/src/cg_trace/main.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/python/tools/recorded-call-graph-metrics/src/cg_trace/main.py b/python/tools/recorded-call-graph-metrics/src/cg_trace/main.py index 1fadfea15e3..5f5a7881500 100644 --- a/python/tools/recorded-call-graph-metrics/src/cg_trace/main.py +++ b/python/tools/recorded-call-graph-metrics/src/cg_trace/main.py @@ -31,8 +31,18 @@ def record_calls(code, globals): return all_calls_sorted, captured_stdout, captured_stderr, exit_status +def setup_logging(): + # code we run can also set up logging, so we need to set the level directly on our + # own pacakge + sh = logging.StreamHandler(stream=sys.stderr) + + pkg_logger = logging.getLogger("cg_trace") + pkg_logger.addHandler(sh) + pkg_logger.setLevel(logging.INFO) + + def main(args=None) -> int: - logging.basicConfig(stream=sys.stderr, level=logging.DEBUG) + setup_logging() # from . import bytecode_reconstructor # logging.getLogger(bytecode_reconstructor.__name__).setLevel(logging.INFO)