annotate Demo/scripts/pp.py @ 25838:8540887bcec6 2.2

Backport of 2.183: Fix memory leak in add_subclass() found by NealN with valgrind.
author Guido van Rossum <guido@python.org>
date Fri, 18 Oct 2002 13:42:21 +0000
parents 73758db2f464
children 11c701038805
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
4573
73758db2f464 /usr/local/bin/python -> /usr/bin/env python
Guido van Rossum <guido@python.org>
parents: 1267
diff changeset
1 #! /usr/bin/env python
912
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
2
934
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
3 # Emulate some Perl command line options.
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
4 # Usage: pp [-a] [-c] [-d] [-e scriptline] [-F fieldsep] [-n] [-p] [file] ...
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
5 # Where the options mean the following:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
6 # -a : together with -n or -p, splits each line into list F
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
7 # -c : check syntax only, do not execute any code
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
8 # -d : run the script under the debugger, pdb
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
9 # -e scriptline : gives one line of the Python script; may be repeated
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
10 # -F fieldsep : sets the field separator for the -a option [not in Perl]
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
11 # -n : runs the script for each line of input
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
12 # -p : prints the line after the script has run
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
13 # When no script lines have been passed, the first file argument
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
14 # contains the script. With -n or -p, the remaining arguments are
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
15 # read as input to the script, line by line. If a file is '-'
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
16 # or missing, standard input is read.
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
17
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
18 # XXX To do:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
19 # - add -i extension option (change files in place)
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
20 # - make a single loop over the files and lines (changes effect of 'break')?
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
21 # - add an option to specify the record separator
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
22 # - except for -n/-p, run directly from the file if at all possible
912
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
23
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
24 import sys
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
25 import string
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
26 import getopt
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
27
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
28 FS = ''
934
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
29 SCRIPT = []
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
30 AFLAG = 0
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
31 CFLAG = 0
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
32 DFLAG = 0
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
33 NFLAG = 0
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
34 PFLAG = 0
912
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
35
934
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
36 try:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
37 optlist, ARGS = getopt.getopt(sys.argv[1:], 'acde:F:np')
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
38 except getopt.error, msg:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
39 sys.stderr.write(sys.argv[0] + ': ' + msg + '\n')
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
40 sys.exit(2)
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
41
912
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
42 for option, optarg in optlist:
934
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
43 if option == '-a':
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
44 AFLAG = 1
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
45 elif option == '-c':
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
46 CFLAG = 1
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
47 elif option == '-d':
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
48 DFLAG = 1
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
49 elif option == '-e':
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
50 for line in string.splitfields(optarg, '\n'):
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
51 SCRIPT.append(line)
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
52 elif option == '-F':
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
53 FS = optarg
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
54 elif option == '-n':
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
55 NFLAG = 1
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
56 PFLAG = 0
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
57 elif option == '-p':
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
58 NFLAG = 1
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
59 PFLAG = 1
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
60 else:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
61 print option, 'not recognized???'
912
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
62
934
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
63 if not ARGS: ARGS.append('-')
912
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
64
934
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
65 if not SCRIPT:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
66 if ARGS[0] == '-':
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
67 fp = sys.stdin
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
68 else:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
69 fp = open(ARGS[0], 'r')
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
70 while 1:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
71 line = fp.readline()
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
72 if not line: break
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
73 SCRIPT.append(line[:-1])
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
74 del fp
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
75 del ARGS[0]
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
76 if not ARGS: ARGS.append('-')
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
77
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
78 if CFLAG:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
79 prologue = ['if 0:']
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
80 epilogue = []
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
81 elif NFLAG:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
82 # Note that it is on purpose that AFLAG and PFLAG are
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
83 # tested dynamically each time through the loop
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
84 prologue = [ \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
85 'LINECOUNT = 0', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
86 'for FILE in ARGS:', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
87 ' \tif FILE == \'-\':', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
88 ' \t \tFP = sys.stdin', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
89 ' \telse:', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
90 ' \t \tFP = open(FILE, \'r\')', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
91 ' \tLINENO = 0', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
92 ' \twhile 1:', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
93 ' \t \tLINE = FP.readline()', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
94 ' \t \tif not LINE: break', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
95 ' \t \tLINENO = LINENO + 1', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
96 ' \t \tLINECOUNT = LINECOUNT + 1', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
97 ' \t \tL = LINE[:-1]', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
98 ' \t \taflag = AFLAG', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
99 ' \t \tif aflag:', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
100 ' \t \t \tif FS: F = string.splitfields(L, FS)', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
101 ' \t \t \telse: F = string.split(L)' \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
102 ]
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
103 epilogue = [ \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
104 ' \t \tif not PFLAG: continue', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
105 ' \t \tif aflag:', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
106 ' \t \t \tif FS: print string.joinfields(F, FS)', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
107 ' \t \t \telse: print string.join(F)', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
108 ' \t \telse: print L', \
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
109 ]
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
110 else:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
111 prologue = ['if 1:']
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
112 epilogue = []
912
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
113
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
114 # Note that we indent using tabs only, so that any indentation style
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
115 # used in 'command' will come out right after re-indentation.
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
116
934
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
117 program = string.joinfields(prologue, '\n') + '\n'
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
118 for line in SCRIPT:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
119 program = program + (' \t \t' + line + '\n')
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
120 program = program + (string.joinfields(epilogue, '\n') + '\n')
912
60e3118d3073 Initial revision
Guido van Rossum <guido@python.org>
parents:
diff changeset
121
934
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
122 import tempfile
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
123 tfn = tempfile.mktemp()
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
124 try:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
125 fp = open(tfn, 'w')
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
126 fp.write(program)
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
127 fp.close()
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
128 if DFLAG:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
129 import pdb
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
130 pdb.run('execfile(' + `tfn` + ')')
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
131 else:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
132 execfile(tfn)
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
133 finally:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
134 import os
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
135 try:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
136 os.unlink(tfn)
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
137 except:
d383d2733176 Almost totally rewritten.
Guido van Rossum <guido@python.org>
parents: 912
diff changeset
138 pass