Guido van Rossum | f06ee5f | 1996-11-27 19:52:01 +0000 | [diff] [blame] | 1 | #! /usr/bin/env python |
Guido van Rossum | ec758ea | 1991-06-04 20:36:54 +0000 | [diff] [blame] | 2 | |
| 3 | # ptags |
| 4 | # |
| 5 | # Create a tags file for Python programs, usable with vi. |
| 6 | # Tagged are: |
| 7 | # - functions (even inside other defs or classes) |
| 8 | # - classes |
| 9 | # - filenames |
| 10 | # Warns about files it cannot open. |
| 11 | # No warnings about duplicate tags. |
| 12 | |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 13 | import sys, re, os |
Guido van Rossum | ec758ea | 1991-06-04 20:36:54 +0000 | [diff] [blame] | 14 | |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 15 | tags = [] # Modified global variable! |
Guido van Rossum | ec758ea | 1991-06-04 20:36:54 +0000 | [diff] [blame] | 16 | |
| 17 | def main(): |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 18 | args = sys.argv[1:] |
| 19 | for file in args: treat_file(file) |
| 20 | if tags: |
| 21 | fp = open('tags', 'w') |
| 22 | tags.sort() |
| 23 | for s in tags: fp.write(s) |
Guido van Rossum | ec758ea | 1991-06-04 20:36:54 +0000 | [diff] [blame] | 24 | |
Guido van Rossum | f2e1cfb | 1992-08-31 10:53:53 +0000 | [diff] [blame] | 25 | |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 26 | expr = '^[ \t]*(def|class)[ \t]+([a-zA-Z0-9_]+)[ \t]*[:\(]' |
| 27 | matcher = re.compile(expr) |
Guido van Rossum | ec758ea | 1991-06-04 20:36:54 +0000 | [diff] [blame] | 28 | |
| 29 | def treat_file(file): |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 30 | try: |
| 31 | fp = open(file, 'r') |
| 32 | except: |
| 33 | sys.stderr.write('Cannot open %s\n' % file) |
| 34 | return |
| 35 | base = os.path.basename(file) |
| 36 | if base[-3:] == '.py': |
| 37 | base = base[:-3] |
| 38 | s = base + '\t' + file + '\t' + '1\n' |
| 39 | tags.append(s) |
| 40 | while 1: |
| 41 | line = fp.readline() |
Tim Peters | 70c4378 | 2001-01-17 08:48:39 +0000 | [diff] [blame] | 42 | if not line: |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 43 | break |
| 44 | m = matcher.match(line) |
| 45 | if m: |
| 46 | content = m.group(0) |
| 47 | name = m.group(2) |
| 48 | s = name + '\t' + file + '\t/^' + content + '/\n' |
| 49 | tags.append(s) |
Guido van Rossum | ec758ea | 1991-06-04 20:36:54 +0000 | [diff] [blame] | 50 | |
| 51 | main() |