mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-10-09 08:31:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 13

Close #15153: Added inspect.getgeneratorlocals to simplify whitebox testing of generator state updates

Browse source
This commit is contained in:
Nick Coghlan 2012-06-23 19:52:05 +10:00
parent 766e62266e
commit 04e2e3f231
5 changed files with 98 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

18
Lib/inspect.py
View file

@ -1259,6 +1259,8 @@ def getattr_static(obj, attr, default=_sentinel):
raise AttributeError(attr)
# ------------------------------------------------ generator introspection
GEN_CREATED = 'GEN_CREATED'
GEN_RUNNING = 'GEN_RUNNING'
GEN_SUSPENDED = 'GEN_SUSPENDED'
@ -1282,6 +1284,22 @@ def getgeneratorstate(generator):
return GEN_SUSPENDED
def getgeneratorlocals(generator):
"""
Get the mapping of generator local variables to their current values.
A dict is returned, with the keys the local variable names and values the
bound values."""
if not isgenerator(generator):
raise TypeError("'{!r}' is not a Python generator".format(generator))
frame = getattr(generator, "gi_frame", None)
if frame is not None:
return generator.gi_frame.f_locals
else:
return {}
###############################################################################
### Function Signature Object (PEP 362)
###############################################################################