mirror of https://github.com/python/cpython
102 lines
3.1 KiB
Python
102 lines
3.1 KiB
Python
|
# Extension to format a paragraph
|
||
|
|
|
||
|
|
import string
|
||
|
|
import re
|
||
|
|
|
||
|
|
class FormatParagraph:
|
||
|
|
|
||
|
|
menudefs = [
|
||
|
|
('edit', [
|
||
|
|
('Format Paragraph', '<<format-paragraph>>'),
|
||
|
|
])
|
||
|
|
]
|
||
|
|
|
||
|
|
keydefs = {
|
||
|
|
'<<format-paragraph>>': ['<Alt-q>'],
|
||
|
|
}
|
||
|
|
|
||
|
|
def __init__(self, editwin):
|
||
|
|
self.editwin = editwin
|
||
|
|
|
||
|
|
def format_paragraph_event(self, event):
|
||
|
|
text = self.editwin.text
|
||
|
|
try:
|
||
|
|
first = text.index("sel.first")
|
||
|
|
last = text.index("sel.last")
|
||
|
|
except TclError:
|
||
|
|
first = last = None
|
||
|
|
if first and last:
|
||
|
|
data = text.get(first, last)
|
||
|
|
else:
|
||
|
|
first, last, data = find_paragraph(text, text.index("insert"))
|
||
|
|
newdata = reformat_paragraph(data)
|
||
|
|
text.tag_remove("sel", "1.0", "end")
|
||
|
|
if newdata != data:
|
||
|
|
text.mark_set("insert", first)
|
||
|
|
text.delete(first, last)
|
||
|
|
text.insert(first, newdata)
|
||
|
|
else:
|
||
|
|
text.mark_set("insert", last)
|
||
|
|
text.see("insert")
|
||
|
|
|
||
|
|
def find_paragraph(text, mark):
|
||
|
|
lineno, col = map(int, string.split(mark, "."))
|
||
|
|
line = text.get("%d.0" % lineno, "%d.0 lineend" % lineno)
|
||
|
|
while is_all_white(line):
|
||
|
|
lineno = lineno + 1
|
||
|
|
line = text.get("%d.0" % lineno, "%d.0 lineend" % lineno)
|
||
|
|
first_lineno = lineno
|
||
|
|
while not is_all_white(line):
|
||
|
|
lineno = lineno + 1
|
||
|
|
line = text.get("%d.0" % lineno, "%d.0 lineend" % lineno)
|
||
|
|
last = "%d.0" % lineno
|
||
|
|
# Search back to beginning of paragraph
|
||
|
|
lineno = first_lineno - 1
|
||
|
|
line = text.get("%d.0" % lineno, "%d.0 lineend" % lineno)
|
||
|
|
while not is_all_white(line):
|
||
|
|
lineno = lineno - 1
|
||
|
|
line = text.get("%d.0" % lineno, "%d.0 lineend" % lineno)
|
||
|
|
first = "%d.0" % (lineno+1)
|
||
|
|
return first, last, text.get(first, last)
|
||
|
|
|
||
|
|
def reformat_paragraph(data, limit=72):
|
||
|
|
lines = string.split(data, "\n")
|
||
|
|
i = 0
|
||
|
|
n = len(lines)
|
||
|
|
while i < n and is_all_white(lines[i]):
|
||
|
|
i = i+1
|
||
|
|
if i >= n:
|
||
|
|
return data
|
||
|
|
indent1 = get_indent(lines[i])
|
||
|
|
if i+1 < n and not is_all_white(lines[i+1]):
|
||
|
|
indent2 = get_indent(lines[i+1])
|
||
|
|
else:
|
||
|
|
indent2 = indent1
|
||
|
|
new = lines[:i]
|
||
|
|
partial = indent1
|
||
|
|
while i < n and not is_all_white(lines[i]):
|
||
|
|
# XXX Should take double space after period (etc.) into account
|
||
|
|
words = re.split("(\s+)", lines[i])
|
||
|
|
for j in range(0, len(words), 2):
|
||
|
|
word = words[j]
|
||
|
|
if not word:
|
||
|
|
continue # Can happen when line ends in whitespace
|
||
|
|
if len(string.expandtabs(partial + word)) > limit and \
|
||
|
|
partial != indent1:
|
||
|
|
new.append(string.rstrip(partial))
|
||
|
|
partial = indent2
|
||
|
|
partial = partial + word + " "
|
||
|
|
if j+1 < len(words) and words[j+1] != " ":
|
||
|
|
partial = partial + " "
|
||
|
|
i = i+1
|
||
|
|
new.append(string.rstrip(partial))
|
||
|
|
# XXX Should reformat remaining paragraphs as well
|
||
|
|
new.extend(lines[i:])
|
||
|
|
return string.join(new, "\n")
|
||
|
|
|
||
|
|
def is_all_white(line):
|
||
|
|
return re.match(r"^\s*$", line) is not None
|
||
|
|
|
||
|
|
def get_indent(line):
|
||
|
|
return re.match(r"^(\s*)", line).group()
|