From: Benjamin Peterson <benjamin@python.org>
Date: Sun, 28 Jun 2009 19:27:55 +0000 (+0000)
Subject: return locals and cells in get_locals() not bound globals, though
X-Git-Tag: v2.7a1~889
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=552e7a7e2f526fa0637a3e14f47354c567dfe26e;p=python

return locals and cells in get_locals() not bound globals, though
---

diff --git a/Lib/symtable.py b/Lib/symtable.py
index 44d70a35fd..ca73f58651 100644
--- a/Lib/symtable.py
+++ b/Lib/symtable.py
@@ -3,7 +3,7 @@
 import _symtable
 from _symtable import (USE, DEF_GLOBAL, DEF_LOCAL, DEF_PARAM,
      DEF_IMPORT, DEF_BOUND, OPT_IMPORT_STAR, OPT_EXEC, OPT_BARE_EXEC,
-     SCOPE_OFF, SCOPE_MASK, FREE, GLOBAL_IMPLICIT, GLOBAL_EXPLICIT)
+     SCOPE_OFF, SCOPE_MASK, FREE, GLOBAL_IMPLICIT, GLOBAL_EXPLICIT, CELL, LOCAL)
 
 import weakref
 
@@ -137,7 +137,9 @@ class Function(SymbolTable):
 
     def get_locals(self):
         if self.__locals is None:
-            self.__locals = self.__idents_matching(lambda x:x & DEF_BOUND)
+            locs = (LOCAL, CELL)
+            test = lambda x: ((x >> SCOPE_OFF) & SCOPE_MASK) in locs
+            self.__locals = self.__idents_matching(test)
         return self.__locals
 
     def get_globals(self):
diff --git a/Lib/test/test_symtable.py b/Lib/test/test_symtable.py
index 792d9c3343..2de30ae664 100644
--- a/Lib/test/test_symtable.py
+++ b/Lib/test/test_symtable.py
@@ -92,7 +92,7 @@ class SymtableTest(unittest.TestCase):
         func = self.spam
         self.assertEqual(func.get_parameters(), ("a", "b", "kw", "var"))
         self.assertEqual(func.get_locals(),
-                         ("a", "b", "bar", "internal", "kw", "var", "x"))
+                         ("a", "b", "internal", "kw", "var", "x"))
         self.assertEqual(func.get_globals(), ("bar", "glob"))
         self.assertEqual(self.internal.get_frees(), ("x",))