Mark bindings as used by locals()

Fixes PyCQA#136
Fixes PyCQA#333

Author: jayvdb

pyflakes/checker.py

@@ -409,7 +409,6 @@ class FunctionScope(Scope):
 
     @ivar globals: Names declared 'global' in this function.
     """
-    usesLocals = False
     alwaysUsed = {'__tracebackhide__', '__traceback_info__',
                   '__traceback_supplement__'}
 
@@ -426,7 +425,6 @@ def unusedAssignments(self):
         """
         for name, binding in self.items():
             if (not binding.used and name not in self.globals
-                    and not self.usesLocals
                     and isinstance(binding, Assignment)):
                 yield name, binding
 
@@ -708,6 +706,16 @@ def handleNodeLoad(self, node):
         in_generators = None
         importStarred = None
 
+        if node.id == 'locals' and isinstance(node.parent, ast.Call):
+            # we are doing locals() call, which marks names currently
+            # in scope as used.
+            scope = self.scope
+            if isinstance(scope, GeneratorScope):
+                scope = self.scopeStack[-2]
+            if isinstance(scope, FunctionScope):
+                for binding in scope.values():
+                    binding.used = (self.scope, node)
+
         # try enclosing function scopes and global scope
         for scope in self.scopeStack[-1::-1]:
             if isinstance(scope, ClassScope):
@@ -1094,10 +1102,6 @@ def NAME(self, node):
         # Locate the name in locals / function / globals scopes.
         if isinstance(node.ctx, (ast.Load, ast.AugLoad)):
             self.handleNodeLoad(node)
-            if (node.id == 'locals' and isinstance(self.scope, FunctionScope)
-                    and isinstance(node.parent, ast.Call)):
-                # we are doing locals() call in current scope
-                self.scope.usesLocals = True
         elif isinstance(node.ctx, (ast.Store, ast.AugStore)):
             self.handleNodeStore(node)
         elif isinstance(node.ctx, ast.Del):

pyflakes/test/test_other.py

@@ -1185,6 +1185,17 @@ def a():
             return locals()
         ''')
 
+    def test_unusedVariableWithLocalsInComprehension(self):
+        """
+        Using locals() in comprehension it is perfectly valid
+        to have unused variables
+        """
+        self.flakes('''
+        def a():
+            b = 1
+            return (i for i in locals())
+        ''')
+
     def test_unusedVariableNoLocals(self):
         """
         Using locals() in wrong scope should not matter