82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
|
# A subroutine for extracting a function name from a code object
|
||
|
|
# (with cache)
|
||
|
|
|
||
|
|
import sys
|
||
|
|
from stat import *
|
||
|
|
import string
|
||
|
|
import os
|
||
|
|
import linecache
|
||
|
|
|
||
|
|
# 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
|
||
|
|
|
||
|
|
def getcodename(co):
|
||
|
|
try:
|
||
|
|
return co.co_name
|
||
|
|
except AttributeError:
|
||
|
|
pass
|
||
|
|
key = `co` # arbitrary but uniquely identifying string
|
||
|
|
if _namecache.has_key(key): return _namecache[key]
|
||
|
|
filename = co.co_filename
|
||
|
|
code = co.co_code
|
||
|
|
name = ''
|
||
|
|
if ord(code[0]) == SET_LINENO:
|
||
|
|
lineno = ord(code[1]) | ord(code[2]) << 8
|
||
|
|
line = linecache.getline(filename, lineno)
|
||
|
|
words = line.split()
|
||
|
|
if len(words) >= 2 and words[0] in ('def', 'class'):
|
||
|
|
name = words[1]
|
||
|
|
for i in range(len(name)):
|
||
|
|
if name[i] not in identchars:
|
||
|
|
name = name[:i]
|
||
|
|
break
|
||
|
|
_namecache[key] = name
|
||
|
|
return name
|
||
|
|
|
||
|
|
# Use the above routine to find a function's name.
|
||
|
|
|
||
|
|
def getfuncname(func):
|
||
|
|
try:
|
||
|
|
return func.func_name
|
||
|
|
except AttributeError:
|
||
|
|
pass
|
||
|
|
return getcodename(func.func_code)
|
||
|
|
|
||
|
|
# A part of the above code to extract just the line number from a code object.
|
||
|
|
|
||
|
|
def getlineno(co):
|
||
|
|
try:
|
||
|
|
return co.co_firstlineno
|
||
|
|
except AttributeError:
|
||
|
|
pass
|
||
|
|
code = co.co_code
|
||
|
|
if ord(code[0]) == SET_LINENO:
|
||
|
|
return ord(code[1]) | ord(code[2]) << 8
|
||
|
|
else:
|
||
|
|
return -1
|