mirror of https://github.com/python/cpython
125 lines
4.2 KiB
Python
125 lines
4.2 KiB
Python
import string
|
|
|
|
class AutoIndent:
|
|
|
|
def __init__(self, text, prefertabs=0, spaceindent=4*" "):
|
|
self.text = text
|
|
self.prefertabs = prefertabs
|
|
self.spaceindent = spaceindent
|
|
text.bind("<<newline-and-indent>>", self.autoindent)
|
|
text.bind("<<indent-region>>", self.indentregion)
|
|
text.bind("<<dedent-region>>", self.dedentregion)
|
|
text.bind("<<comment-region>>", self.commentregion)
|
|
text.bind("<<uncomment-region>>", self.uncommentregion)
|
|
|
|
def config(self, **options):
|
|
for key, value in options.items():
|
|
if key == 'prefertabs':
|
|
self.prefertabs = value
|
|
elif key == 'spaceindent':
|
|
self.spaceindent = value
|
|
else:
|
|
raise KeyError, "bad option name: %s" % `key`
|
|
|
|
def autoindent(self, event):
|
|
text = self.text
|
|
line = text.get("insert linestart", "insert")
|
|
i, n = 0, len(line)
|
|
while i < n and line[i] in " \t":
|
|
i = i+1
|
|
indent = line[:i]
|
|
lastchar = text.get("insert -1c")
|
|
if lastchar == ":":
|
|
if not indent:
|
|
if self.prefertabs:
|
|
indent = "\t"
|
|
else:
|
|
indent = self.spaceindent
|
|
elif indent[-1] == "\t":
|
|
indent = indent + "\t"
|
|
else:
|
|
indent = indent + self.spaceindent
|
|
text.insert("insert", "\n" + indent)
|
|
text.see("insert")
|
|
return "break"
|
|
|
|
def indentregion(self, event):
|
|
head, tail, chars, lines = self.getregion()
|
|
for pos in range(len(lines)):
|
|
line = lines[pos]
|
|
if line:
|
|
i, n = 0, len(line)
|
|
while i < n and line[i] in " \t":
|
|
i = i+1
|
|
line = line[:i] + " " + line[i:]
|
|
lines[pos] = line
|
|
self.setregion(head, tail, chars, lines)
|
|
return "break"
|
|
|
|
def dedentregion(self, event):
|
|
head, tail, chars, lines = self.getregion()
|
|
for pos in range(len(lines)):
|
|
line = lines[pos]
|
|
if line:
|
|
i, n = 0, len(line)
|
|
while i < n and line[i] in " \t":
|
|
i = i+1
|
|
indent, line = line[:i], line[i:]
|
|
if indent:
|
|
if indent == "\t" or indent[-2:] == "\t\t":
|
|
indent = indent[:-1] + " "
|
|
elif indent[-4:] == " ":
|
|
indent = indent[:-4]
|
|
else:
|
|
indent = string.expandtabs(indent, 8)
|
|
indent = indent[:-4]
|
|
line = indent + line
|
|
lines[pos] = line
|
|
self.setregion(head, tail, chars, lines)
|
|
return "break"
|
|
|
|
def commentregion(self, event):
|
|
head, tail, chars, lines = self.getregion()
|
|
for pos in range(len(lines)):
|
|
line = lines[pos]
|
|
if not line:
|
|
continue
|
|
lines[pos] = '##' + line
|
|
self.setregion(head, tail, chars, lines)
|
|
|
|
def uncommentregion(self, event):
|
|
head, tail, chars, lines = self.getregion()
|
|
for pos in range(len(lines)):
|
|
line = lines[pos]
|
|
if not line:
|
|
continue
|
|
if line[:2] == '##':
|
|
line = line[2:]
|
|
elif line[:1] == '#':
|
|
line = line[1:]
|
|
lines[pos] = line
|
|
self.setregion(head, tail, chars, lines)
|
|
|
|
def getregion(self):
|
|
text = self.text
|
|
head = text.index("sel.first linestart")
|
|
tail = text.index("sel.last -1c lineend +1c")
|
|
if not (head and tail):
|
|
head = text.index("insert linestart")
|
|
tail = text.index("insert lineend +1c")
|
|
chars = text.get(head, tail)
|
|
lines = string.split(chars, "\n")
|
|
return head, tail, chars, lines
|
|
|
|
def setregion(self, head, tail, chars, lines):
|
|
text = self.text
|
|
newchars = string.join(lines, "\n")
|
|
if newchars == chars:
|
|
text.bell()
|
|
return
|
|
text.tag_remove("sel", "1.0", "end")
|
|
text.mark_set("insert", head)
|
|
text.delete(head, tail)
|
|
text.insert(head, newchars)
|
|
text.tag_add("sel", head, "insert")
|