diff options
author | Guido van Rossum <guido@python.org> | 1999-06-10 14:39:39 (GMT) |
---|---|---|
committer | Guido van Rossum <guido@python.org> | 1999-06-10 14:39:39 (GMT) |
commit | a3b4a33f3b37870096c8062af1dd6bae051af30e (patch) | |
tree | bc5a50309e930af5799b26c1d521c273534b13f1 /Lib/pyclbr.py | |
parent | ea827e916c1a92820254b5adbc4b6580e29787da (diff) | |
download | cpython-a3b4a33f3b37870096c8062af1dd6bae051af30e.zip cpython-a3b4a33f3b37870096c8062af1dd6bae051af30e.tar.gz cpython-a3b4a33f3b37870096c8062af1dd6bae051af30e.tar.bz2 |
Co-production with Tim Peters, implementing a suggestion by Mark
Hammond: record top-level functions (as Function instances, a simple
subclass of Class). You must use the new interface readmodule_ex() to
get these, though.
Diffstat (limited to 'Lib/pyclbr.py')
-rw-r--r-- | Lib/pyclbr.py | 37 |
1 files changed, 31 insertions, 6 deletions
diff --git a/Lib/pyclbr.py b/Lib/pyclbr.py index 3d5e0fa..a6c210e 100644 --- a/Lib/pyclbr.py +++ b/Lib/pyclbr.py @@ -116,7 +116,27 @@ class Class: def _addmethod(self, name, lineno): self.methods[name] = lineno +class Function(Class): + '''Class to represent a top-level Python function''' + def __init__(self, module, name, file, lineno): + Class.__init__(self, module, name, None, file, lineno) + def _addmethod(self, name, lineno): + assert 0, "Function._addmethod() shouldn't be called" + def readmodule(module, path=[], inpackage=0): + '''Backwards compatible interface. + + Like readmodule_ex() but strips Function objects from the + resulting dictionary.''' + + dict = readmodule_ex(module, path, inpackage) + res = {} + for key, value in dict.items(): + if not isinstance(value, Function): + res[key] = value + return res + +def readmodule_ex(module, path=[], inpackage=0): '''Read a module file and return a dictionary of classes. Search for MODULE in PATH and sys.path, read and parse the @@ -189,19 +209,24 @@ def readmodule(module, path=[], inpackage=0): if m.start("Method") >= 0: # found a method definition or function thisindent = _indent(m.group("MethodIndent")) + meth_name = m.group("MethodName") + lineno = lineno + \ + countnl(src, '\n', + last_lineno_pos, start) + last_lineno_pos = start # close all classes indented at least as much while classstack and \ classstack[-1][1] >= thisindent: del classstack[-1] if classstack: - # and we know the class it belongs to - meth_name = m.group("MethodName") - lineno = lineno + \ - countnl(src, '\n', - last_lineno_pos, start) - last_lineno_pos = start + # it's a class method cur_class = classstack[-1][0] cur_class._addmethod(meth_name, lineno) + else: + # it's a function + f = Function(module, meth_name, + file, lineno) + dict[meth_name] = f elif m.start("String") >= 0: pass |