From: Kefu Chai <kchai@redhat.com>
Date: Thu, 19 Nov 2020 05:37:00 +0000 (+0800)
Subject: doc/script/gen_state_diagram: only run if called as a script
X-Git-Tag: v16.1.0~542^2~5
X-Git-Url: http://git-server-git.apps.pok.os.sepia.ceph.com/?a=commitdiff_plain;h=e9af59a50e880070db6f841921dc654f771532ad;p=ceph.git

doc/script/gen_state_diagram: only run if called as a script

Signed-off-by: Kefu Chai <kchai@redhat.com>
---

diff --git a/doc/scripts/gen_state_diagram.py b/doc/scripts/gen_state_diagram.py
index 868eaa287d3..2054bfb1f09 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()