traverseda
/
cpython
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Clarify the purpose of getsourcefile(). 

Add getabsfile() for getting a most-normalized path.
This commit is contained in:
Ka-Ping Yee 2001-03-02 02:08:53 +00:00
parent 0a8c29be4b
commit c113c24e19
1 changed files with 19 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
Lib/inspect.py
View File

@ -169,7 +169,7 @@ def getdoc(object):
return string.join(lines, '\n') return string.join(lines, '\n')
def getfile(object): def getfile(object):
"""Try to guess which (text or binary) file an object was defined in.""" """Work out which source or compiled file an object was defined in."""
if ismodule(object): if ismodule(object):
if hasattr(object, '__file__'): if hasattr(object, '__file__'):
return object.__file__ return object.__file__
@ -192,6 +192,21 @@ def getfile(object):
raise TypeError, 'arg is not a module, class, method, ' \ raise TypeError, 'arg is not a module, class, method, ' \
'function, traceback, frame, or code object' 'function, traceback, frame, or code object'
def getsourcefile(object):
"""Return the Python source file an object was defined in, if it exists."""
filename = getfile(object)
if string.lower(filename[-4:]) in ['.pyc', '.pyo']:
filename = filename[:-4] + '.py'
if string.lower(filename[-3:]) == '.py' and os.path.exists(filename):
return filename
def getabsfile(object):
"""Return an absolute path to the source file or compiled file for an object.
The idea is for each object to have a unique origin, so this routine normalizes
the result as much as possible."""
return os.path.normcase(os.path.abspath(getsourcefile(object) or getfile(object)))
modulesbyfile = {} modulesbyfile = {}
def getmodule(object): def getmodule(object):
@ -199,15 +214,14 @@ def getmodule(object):
if isclass(object): if isclass(object):
return sys.modules.get(object.__module__) return sys.modules.get(object.__module__)
try: try:
file = os.path.abspath(getsourcefile(object)) file = getabsfile(object)
except TypeError: except TypeError:
return None return None
if modulesbyfile.has_key(file): if modulesbyfile.has_key(file):
return sys.modules[modulesbyfile[file]] return sys.modules[modulesbyfile[file]]
for module in sys.modules.values(): for module in sys.modules.values():
if hasattr(module, '__file__'): if hasattr(module, '__file__'):
modulesbyfile[ modulesbyfile[getabsfile(module)] = module.__name__
os.path.abspath(getsourcefile(module))] = module.__name__
if modulesbyfile.has_key(file): if modulesbyfile.has_key(file):
return sys.modules[modulesbyfile[file]] return sys.modules[modulesbyfile[file]]
main = sys.modules['__main__'] main = sys.modules['__main__']
@ -221,13 +235,6 @@ def getmodule(object):
if builtinobject is object: return builtin if builtinobject is object: return builtin
except AttributeError: pass except AttributeError: pass
def getsourcefile(object):
"""Try to guess which Python source file an object was defined in."""
filename = getfile(object)
if filename[-4:] == '.pyc':
filename = filename[:-4] + '.py'
return filename
def findsource(object): def findsource(object):
"""Return the entire source file and starting line number for an object. """Return the entire source file and starting line number for an object.
@ -571,7 +578,7 @@ def getframeinfo(frame, context=1):
start = min(start, len(lines) - context) start = min(start, len(lines) - context)
lines = lines[start:start+context] lines = lines[start:start+context]
index = lineno - 1 - start index = lineno - 1 - start
except: except IOError:
lines = index = None lines = index = None
else: else:
lines = index = None lines = index = None