From dbfd415918999d8d71cfb3e332166d73d839ce07 Mon Sep 17 00:00:00 2001 From: David Dorfman Date: Wed, 8 May 2024 22:15:43 +0300 Subject: [PATCH] Move logging setup to main (fixes #95) --- code2flow/engine.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/code2flow/engine.py b/code2flow/engine.py index ad13a43..b3e81ad 100644 --- a/code2flow/engine.py +++ b/code2flow/engine.py @@ -672,7 +672,7 @@ def code2flow(raw_source_paths, output_file, language=None, hide_legend=True, exclude_namespaces=None, exclude_functions=None, include_only_namespaces=None, include_only_functions=None, no_grouping=False, no_trimming=False, skip_parse_errors=False, - lang_params=None, subset_params=None, level=logging.INFO): + lang_params=None, subset_params=None): """ Top-level function. Generate a diagram based on source code. Can generate either a dotfile or an image. @@ -708,8 +708,6 @@ def code2flow(raw_source_paths, output_file, language=None, hide_legend=True, include_only_functions = include_only_functions or [] assert isinstance(include_only_functions, list) - logging.basicConfig(format="Code2Flow: %(message)s", level=level) - sources, language = get_sources_and_language(raw_source_paths, language) output_ext = None @@ -857,6 +855,7 @@ def main(sys_argv=None): subset_params = SubsetParams.generate(args.target_function, args.upstream_depth, args.downstream_depth) + logging.basicConfig(format="Code2Flow: %(message)s", level=level) code2flow( raw_source_paths=args.sources, output_file=args.output, @@ -871,5 +870,4 @@ def main(sys_argv=None): skip_parse_errors=args.skip_parse_errors, lang_params=lang_params, subset_params=subset_params, - level=level, )