| #! /usr/bin/env python |
| |
| # ptags |
| # |
| # Create a tags file for Python programs, usable with vi. |
| # Tagged are: |
| # - functions (even inside other defs or classes) |
| # - classes |
| # - filenames |
| # Warns about files it cannot open. |
| # No warnings about duplicate tags. |
| |
| import sys, re, os |
| |
| tags = [] # Modified global variable! |
| |
| def main(): |
| args = sys.argv[1:] |
| for filename in args: |
| treat_file(filename) |
| if tags: |
| fp = open('tags', 'w') |
| tags.sort() |
| for s in tags: fp.write(s) |
| |
| |
| expr = '^[ \t]*(def|class)[ \t]+([a-zA-Z0-9_]+)[ \t]*[:\(]' |
| matcher = re.compile(expr) |
| |
| def treat_file(filename): |
| try: |
| fp = open(filename, 'r') |
| except: |
| sys.stderr.write('Cannot open %s\n' % filename) |
| return |
| base = os.path.basename(filename) |
| if base[-3:] == '.py': |
| base = base[:-3] |
| s = base + '\t' + filename + '\t' + '1\n' |
| tags.append(s) |
| while 1: |
| line = fp.readline() |
| if not line: |
| break |
| m = matcher.match(line) |
| if m: |
| content = m.group(0) |
| name = m.group(2) |
| s = name + '\t' + filename + '\t/^' + content + '/\n' |
| tags.append(s) |
| |
| if __name__ == '__main__': |
| main() |