# A subroutine for extracting a function name from a code object
# XXX The functions getcodename() and getfuncname() are now obsolete
# XXX as code and function objects now have a name attribute --
# XXX co.co_name and f.func_name.
# XXX getlineno() is now also obsolete because of the new attribute
# XXX of code objects, co.co_firstlineno.
# Extract the function or class name from a code object.
# This is a bit of a hack, since a code object doesn't contain
# the name directly. So what do we do:
# - get the filename (which *is* in the code object)
# - look in the code string to find the first SET_LINENO instruction
# (this must be the first instruction)
# - get the line from the file
# - if the line starts with 'class' or 'def' (after possible whitespace),
# extract the following identifier
# This breaks apart when the function was read from <stdin>
# or constructed by exec(), when the file is not accessible,
# and also when the file has been modified or when a line is
# continued with a backslash before the function or class name.
# Because this is a pretty expensive hack, a cache is kept.
SET_LINENO
= 127 # The opcode (see "opcode.h" in the Python source)
identchars
= string
.ascii_letters
+ string
.digits
+ '_' # Identifier characters
_namecache
= {} # The cache
key
= `co`
# arbitrary but uniquely identifying string
if _namecache
.has_key(key
): return _namecache
[key
]
filename
= co
.co_filename
if ord(code
[0]) == SET_LINENO
:
lineno
= ord(code
[1]) |
ord(code
[2]) << 8
line
= linecache
.getline(filename
, lineno
)
if len(words
) >= 2 and words
[0] in ('def', 'class'):
for i
in range(len(name
)):
if name
[i
] not in identchars
:
# Use the above routine to find a function's name.
return getcodename(func
.func_code
)
# A part of the above code to extract just the line number from a code object.
if ord(code
[0]) == SET_LINENO
:
return ord(code
[1]) |
ord(code
[2]) << 8