Peter Schneider-Kamp:

Problem:
A Python program can be completed and reformatted using
Tools/scripts/pindent.py. Unfortunately there is no option for removal
of the generated "# end"-tags.  Although a few Python commands or a
"grep -v '# end '" can do wonders here, there are two drawbacks:
- not everyone has grep/time to write a Python script
- it is not checked whether the "# end"-tags were used validly

Solution:
add extra option "-e" (eliminate) to pindent.py
This commit is contained in:
Guido van Rossum 2000-06-28 22:47:22 +00:00
parent cc6a438d7f
commit 59811b189d
1 changed files with 75 additions and 10 deletions

View File

@ -1,10 +1,12 @@
#! /usr/bin/env python #! /usr/bin/env python
# This file contains a class and a main program that perform two # This file contains a class and a main program that perform three
# related (though complimentary) formatting operations on Python # related (though complimentary) formatting operations on Python
# programs. When called as "pindend -c", it takes a valid Python # programs. When called as "pindent -c", it takes a valid Python
# program as input and outputs a version augmented with block-closing # program as input and outputs a version augmented with block-closing
# comments. When called as "pindent -r" it assumes its input is a # comments. When called as "pindent -e", it assumes its input is a
# Python program with block-closing comments and outputs a commentless
# version. When called as "pindent -r" it assumes its input is a
# Python program with block-closing comments but with its indentation # Python program with block-closing comments but with its indentation
# messed up, and outputs a properly indented version. # messed up, and outputs a properly indented version.
@ -34,11 +36,12 @@
# that indentation is not significant when interpreting block-closing # that indentation is not significant when interpreting block-closing
# comments). # comments).
# Both operations are idempotent (i.e. applied to their own output # The operations are idempotent (i.e. applied to their own output
# they yield an identical result). Running first "pindent -c" and # they yield an identical result). Running first "pindent -c" and
# then "pindent -r" on a valid Python program produces a program that # then "pindent -r" on a valid Python program produces a program that
# is semantically identical to the input (though its indentation may # is semantically identical to the input (though its indentation may
# be different). # be different). Running "pindent -e" on that output produces a
# program that only differs from the original in indentation.
# Other options: # Other options:
# -s stepsize: set the indentation step size (default 8) # -s stepsize: set the indentation step size (default 8)
@ -193,6 +196,34 @@ class PythonIndenter:
# end if # end if
# end def reformat # end def reformat
def eliminate(self):
begin_counter = 0
end_counter = 0
while 1:
line = self.getline()
if not line: break # EOF
# end if
m = self.endprog.match(line)
if m:
end_counter = end_counter + 1
continue
# end if
m = self.kwprog.match(line)
if m:
kw = m.group('kw')
if kw in start:
begin_counter = begin_counter + 1
# end if
# end if
self.putline(line)
# end while
if begin_counter - end_counter < 0:
sys.stderr.write('Warning: input contained more end tags than expected\n')
elif begin_counter - end_counter > 0:
sys.stderr.write('Warning: input contained less end tags than expected\n')
# end if
# end def eliminate
def complete(self): def complete(self):
self.indentsize = 1 self.indentsize = 1
stack = [] stack = []
@ -293,12 +324,18 @@ class PythonIndenter:
# - xxx_string(s): take and return string object # - xxx_string(s): take and return string object
# - xxx_file(filename): process file in place, return true iff changed # - xxx_file(filename): process file in place, return true iff changed
def complete_filter(input= sys.stdin, output = sys.stdout, def complete_filter(input = sys.stdin, output = sys.stdout,
stepsize = STEPSIZE, tabsize = TABSIZE): stepsize = STEPSIZE, tabsize = TABSIZE):
pi = PythonIndenter(input, output, stepsize, tabsize) pi = PythonIndenter(input, output, stepsize, tabsize)
pi.complete() pi.complete()
# end def complete_filter # end def complete_filter
def eliminate_filter(input= sys.stdin, output = sys.stdout,
stepsize = STEPSIZE, tabsize = TABSIZE):
pi = PythonIndenter(input, output, stepsize, tabsize)
pi.eliminate()
# end def eliminate_filter
def reformat_filter(input = sys.stdin, output = sys.stdout, def reformat_filter(input = sys.stdin, output = sys.stdout,
stepsize = STEPSIZE, tabsize = TABSIZE): stepsize = STEPSIZE, tabsize = TABSIZE):
pi = PythonIndenter(input, output, stepsize, tabsize) pi = PythonIndenter(input, output, stepsize, tabsize)
@ -357,6 +394,14 @@ def complete_string(source, stepsize = STEPSIZE, tabsize = TABSIZE):
return output.getvalue() return output.getvalue()
# end def complete_string # end def complete_string
def eliminate_string(source, stepsize = STEPSIZE, tabsize = TABSIZE):
input = StringReader(source)
output = StringWriter()
pi = PythonIndenter(input, output, stepsize, tabsize)
pi.eliminate()
return output.getvalue()
# end def eliminate_string
def reformat_string(source, stepsize = STEPSIZE, tabsize = TABSIZE): def reformat_string(source, stepsize = STEPSIZE, tabsize = TABSIZE):
input = StringReader(source) input = StringReader(source)
output = StringWriter() output = StringWriter()
@ -380,13 +425,30 @@ def complete_file(filename, stepsize = STEPSIZE, tabsize = TABSIZE):
return 1 return 1
# end def complete_file # end def complete_file
def eliminate_file(filename, stepsize = STEPSIZE, tabsize = TABSIZE):
source = open(filename, 'r').read()
result = eliminate_string(source, stepsize, tabsize)
if source == result: return 0
# end if
import os
try: os.rename(filename, filename + '~')
except os.error: pass
# end try
f = open(filename, 'w')
f.write(result)
f.close()
return 1
# end def eliminate_file
def reformat_file(filename, stepsize = STEPSIZE, tabsize = TABSIZE): def reformat_file(filename, stepsize = STEPSIZE, tabsize = TABSIZE):
source = open(filename, 'r').read() source = open(filename, 'r').read()
result = reformat_string(source, stepsize, tabsize) result = reformat_string(source, stepsize, tabsize)
if source == result: return 0 if source == result: return 0
# end if # end if
import os import os
os.rename(filename, filename + '~') try: os.rename(filename, filename + '~')
except os.error: pass
# end try
f = open(filename, 'w') f = open(filename, 'w')
f.write(result) f.write(result)
f.close() f.close()
@ -396,8 +458,9 @@ def reformat_file(filename, stepsize = STEPSIZE, tabsize = TABSIZE):
# Test program when called as a script # Test program when called as a script
usage = """ usage = """
usage: pindent (-c|-r) [-s stepsize] [-t tabsize] [file] ... usage: pindent (-c|-e|-r) [-s stepsize] [-t tabsize] [file] ...
-c : complete a correctly indented program (add #end directives) -c : complete a correctly indented program (add #end directives)
-e : eliminate #end directives
-r : reformat a completed program (use #end directives) -r : reformat a completed program (use #end directives)
-s stepsize: indentation step (default %(STEPSIZE)d) -s stepsize: indentation step (default %(STEPSIZE)d)
-t tabsize : the worth in spaces of a tab (default %(TABSIZE)d) -t tabsize : the worth in spaces of a tab (default %(TABSIZE)d)
@ -409,7 +472,7 @@ the program acts as a filter (reads stdin, writes stdout).
def test(): def test():
import getopt import getopt
try: try:
opts, args = getopt.getopt(sys.argv[1:], 'crs:t:') opts, args = getopt.getopt(sys.argv[1:], 'cers:t:')
except getopt.error, msg: except getopt.error, msg:
sys.stderr.write('Error: %s\n' % msg) sys.stderr.write('Error: %s\n' % msg)
sys.stderr.write(usage) sys.stderr.write(usage)
@ -421,6 +484,8 @@ def test():
for o, a in opts: for o, a in opts:
if o == '-c': if o == '-c':
action = 'complete' action = 'complete'
elif o == '-e':
action = 'eliminate'
elif o == '-r': elif o == '-r':
action = 'reformat' action = 'reformat'
elif o == '-s': elif o == '-s':
@ -431,7 +496,7 @@ def test():
# end for # end for
if not action: if not action:
sys.stderr.write( sys.stderr.write(
'You must specify -c(omplete) or -r(eformat)\n') 'You must specify -c(omplete), -e(eliminate) or -r(eformat)\n')
sys.stderr.write(usage) sys.stderr.write(usage)
sys.exit(2) sys.exit(2)
# end if # end if