add a helper to get a function's dictionary

--HG--
branch : trunk
This commit is contained in:
Benjamin Peterson 2009-09-03 16:45:28 -05:00
parent 499a982860
commit 86da34b874
3 changed files with 14 additions and 0 deletions

View File

@ -153,6 +153,7 @@ initpkg(__name__,
'builtin._istext' : ('./builtin/builtin31.py', '_istext'),
'builtin._getimself' : ('./builtin/builtin31.py', '_getimself'),
'builtin._getcode' : ('./builtin/builtin31.py', '_getcode'),
'builtin._getfuncdict' : ('./builtin/builtin31.py', '_getfuncdict'),
'builtin.builtins' : ('./builtin/builtin31.py', 'builtins'),
'builtin.execfile' : ('./builtin/builtin31.py', 'execfile'),
'builtin.callable' : ('./builtin/builtin31.py', 'callable'),

View File

@ -25,6 +25,9 @@ if sys.version_info >= (3, 0):
def _getcode(function):
return function.__code__
def _getfuncdict(function):
return getattr(function, "__dict__", None)
def execfile(fn, globs=None, locs=None):
if globs is None:
back = sys._getframe(1)
@ -64,6 +67,9 @@ else:
def _getcode(function):
return function.func_code
def _getfuncdict(function):
return getattr(function, "__dict__", None)
import __builtin__ as builtins
def print_(*args, **kwargs):
""" minimal backport of py3k print statement. """

View File

@ -105,6 +105,13 @@ def test_execfile(tmpdir):
py.builtin.execfile(str(test_file))
assert A.x == 3
def test_getfuncdict():
def f():
pass
f.x = 4
assert py.builtin._getfuncdict(f)["x"] == 4
assert py.builtin._getfuncdict(2) is None
def test_callable():
class A: pass
assert py.builtin.callable(test_callable)