Merge "Sync everything from oslo-incubator."

1 files changed, 14 insertions, 5 deletions

diff --git a/tools/flakes.py b/tools/flakes.py
index 7f96116ca..191bd6eab 100644
--- a/tools/flakes.py
+++ b/tools/flakes.py
@@ -4,12 +4,21 @@
 
  Synced in from openstack-common
 """
+
+__all__ = ['main']
+
+import __builtin__ as builtins
 import sys
 
-import pyflakes.checker
-from pyflakes.scripts import pyflakes
+import pyflakes.api
+from pyflakes import checker
+
+
+def main():
+    checker.Checker.builtIns = (set(dir(builtins)) |
+                                set(['_']) |
+                                set(checker._MAGIC_GLOBALS))
+    sys.exit(pyflakes.api.main())
 
 if __name__ == "__main__":
-    orig_builtins = set(pyflakes.checker._MAGIC_GLOBALS)
-    pyflakes.checker._MAGIC_GLOBALS = orig_builtins | set(['_'])
-    sys.exit(pyflakes.main())
+    main()