From e9af59a50e880070db6f841921dc654f771532ad Mon Sep 17 00:00:00 2001 From: Kefu Chai Date: Thu, 19 Nov 2020 13:37:00 +0800 Subject: [PATCH] doc/script/gen_state_diagram: only run if called as a script Signed-off-by: Kefu Chai --- doc/scripts/gen_state_diagram.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/doc/scripts/gen_state_diagram.py b/doc/scripts/gen_state_diagram.py index 868eaa287d3b4..2054bfb1f09d5 100755 --- a/doc/scripts/gen_state_diagram.py +++ b/doc/scripts/gen_state_diagram.py @@ -235,7 +235,8 @@ class StateMachineRenderer(object): yield("%s -> %s %s;" % (fro, to, append(appendix))) -INPUT_GENERATOR = do_filter(line for line in sys.stdin) -RENDERER = StateMachineRenderer() -RENDERER.read_input(INPUT_GENERATOR) -RENDERER.emit_dot() +if __name__ == '__main__': + INPUT_GENERATOR = do_filter(line for line in sys.stdin) + RENDERER = StateMachineRenderer() + RENDERER.read_input(INPUT_GENERATOR) + RENDERER.emit_dot() -- 2.39.5