Update tools/flakes to work with pydoc

Change-Id: Ib141553017aa88d6edb65b5d51c06740699a2480

1 files changed, 13 insertions, 13 deletions

diff --git a/tools/flakes.py b/tools/flakes.py
index 7ebe10cec..4b93abc21 100644
--- a/tools/flakes.py
+++ b/tools/flakes.py
@@ -1,22 +1,22 @@
-import __builtin__
-import os
-import sys
-
-from pyflakes.scripts.pyflakes import main
-
 """
  wrapper for pyflakes to ignore gettext based warning:
      "undefined name '_'"
 
  From https://bugs.launchpad.net/pyflakes/+bug/844592
 """
+import __builtin__
+import os
+import sys
+
+from pyflakes.scripts.pyflakes import main
 
-names = os.environ.get('PYFLAKES_BUILTINS', '_')
-names = [x.strip() for x in names.split(',')]
-for x in names:
-    if not hasattr(__builtin__, x):
-        setattr(__builtin__, x, True)
+if __name__ == "__main__":
+    names = os.environ.get('PYFLAKES_BUILTINS', '_')
+    names = [x.strip() for x in names.split(',')]
+    for x in names:
+        if not hasattr(__builtin__, x):
+            setattr(__builtin__, x, True)
 
-del names, os, __builtin__
+    del names, os, __builtin__
 
-sys.exit(main())
+    sys.exit(main())