Benjamin Peterson | 90f5ba5 | 2010-03-11 22:53:45 +0000 | [diff] [blame] | 1 | #! /usr/bin/env python3 |
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:] |
Andrew M. Kuchling | ac6df95 | 2003-05-13 18:14:25 +0000 | [diff] [blame] | 19 | for filename in args: |
| 20 | treat_file(filename) |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 21 | if tags: |
| 22 | fp = open('tags', 'w') |
| 23 | tags.sort() |
| 24 | for s in tags: fp.write(s) |
Guido van Rossum | ec758ea | 1991-06-04 20:36:54 +0000 | [diff] [blame] | 25 | |
Guido van Rossum | f2e1cfb | 1992-08-31 10:53:53 +0000 | [diff] [blame] | 26 | |
R David Murray | 44b548d | 2016-09-08 13:59:53 -0400 | [diff] [blame] | 27 | expr = r'^[ \t]*(def|class)[ \t]+([a-zA-Z0-9_]+)[ \t]*[:\(]' |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 28 | matcher = re.compile(expr) |
Guido van Rossum | ec758ea | 1991-06-04 20:36:54 +0000 | [diff] [blame] | 29 | |
Andrew M. Kuchling | ac6df95 | 2003-05-13 18:14:25 +0000 | [diff] [blame] | 30 | def treat_file(filename): |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 31 | try: |
Andrew M. Kuchling | ac6df95 | 2003-05-13 18:14:25 +0000 | [diff] [blame] | 32 | fp = open(filename, 'r') |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 33 | except: |
Andrew M. Kuchling | ac6df95 | 2003-05-13 18:14:25 +0000 | [diff] [blame] | 34 | sys.stderr.write('Cannot open %s\n' % filename) |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 35 | return |
Andrew M. Kuchling | ac6df95 | 2003-05-13 18:14:25 +0000 | [diff] [blame] | 36 | base = os.path.basename(filename) |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 37 | if base[-3:] == '.py': |
| 38 | base = base[:-3] |
Andrew M. Kuchling | ac6df95 | 2003-05-13 18:14:25 +0000 | [diff] [blame] | 39 | s = base + '\t' + filename + '\t' + '1\n' |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 40 | tags.append(s) |
| 41 | while 1: |
| 42 | line = fp.readline() |
Tim Peters | 70c4378 | 2001-01-17 08:48:39 +0000 | [diff] [blame] | 43 | if not line: |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 44 | break |
| 45 | m = matcher.match(line) |
| 46 | if m: |
| 47 | content = m.group(0) |
| 48 | name = m.group(2) |
Andrew M. Kuchling | ac6df95 | 2003-05-13 18:14:25 +0000 | [diff] [blame] | 49 | s = name + '\t' + filename + '\t/^' + content + '/\n' |
Guido van Rossum | 1688f37 | 2000-09-01 13:41:37 +0000 | [diff] [blame] | 50 | tags.append(s) |
Guido van Rossum | ec758ea | 1991-06-04 20:36:54 +0000 | [diff] [blame] | 51 | |
Andrew M. Kuchling | e236b38 | 2004-08-09 17:27:55 +0000 | [diff] [blame] | 52 | if __name__ == '__main__': |
| 53 | main() |