mirror of
https://github.com/python/cpython.git
synced 2024-11-27 11:55:13 +08:00
4d92158f4c
Part 3 of 3, continuing PR #7689. This covers 14 idlelib modules and their tests, rpc to zoomheight except for run (already done) and tooltip (being done separately).
97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
'''Complete the current word before the cursor with words in the editor.
|
|
|
|
Each menu selection or shortcut key selection replaces the word with a
|
|
different word with the same prefix. The search for matches begins
|
|
before the target and moves toward the top of the editor. It then starts
|
|
after the cursor and moves down. It then returns to the original word and
|
|
the cycle starts again.
|
|
|
|
Changing the current text line or leaving the cursor in a different
|
|
place before requesting the next selection causes AutoExpand to reset
|
|
its state.
|
|
|
|
There is only one instance of Autoexpand.
|
|
'''
|
|
import re
|
|
import string
|
|
|
|
|
|
class AutoExpand:
|
|
wordchars = string.ascii_letters + string.digits + "_"
|
|
|
|
def __init__(self, editwin):
|
|
self.text = editwin.text
|
|
self.bell = self.text.bell
|
|
self.state = None
|
|
|
|
def expand_word_event(self, event):
|
|
"Replace the current word with the next expansion."
|
|
curinsert = self.text.index("insert")
|
|
curline = self.text.get("insert linestart", "insert lineend")
|
|
if not self.state:
|
|
words = self.getwords()
|
|
index = 0
|
|
else:
|
|
words, index, insert, line = self.state
|
|
if insert != curinsert or line != curline:
|
|
words = self.getwords()
|
|
index = 0
|
|
if not words:
|
|
self.bell()
|
|
return "break"
|
|
word = self.getprevword()
|
|
self.text.delete("insert - %d chars" % len(word), "insert")
|
|
newword = words[index]
|
|
index = (index + 1) % len(words)
|
|
if index == 0:
|
|
self.bell() # Warn we cycled around
|
|
self.text.insert("insert", newword)
|
|
curinsert = self.text.index("insert")
|
|
curline = self.text.get("insert linestart", "insert lineend")
|
|
self.state = words, index, curinsert, curline
|
|
return "break"
|
|
|
|
def getwords(self):
|
|
"Return a list of words that match the prefix before the cursor."
|
|
word = self.getprevword()
|
|
if not word:
|
|
return []
|
|
before = self.text.get("1.0", "insert wordstart")
|
|
wbefore = re.findall(r"\b" + word + r"\w+\b", before)
|
|
del before
|
|
after = self.text.get("insert wordend", "end")
|
|
wafter = re.findall(r"\b" + word + r"\w+\b", after)
|
|
del after
|
|
if not wbefore and not wafter:
|
|
return []
|
|
words = []
|
|
dict = {}
|
|
# search backwards through words before
|
|
wbefore.reverse()
|
|
for w in wbefore:
|
|
if dict.get(w):
|
|
continue
|
|
words.append(w)
|
|
dict[w] = w
|
|
# search onwards through words after
|
|
for w in wafter:
|
|
if dict.get(w):
|
|
continue
|
|
words.append(w)
|
|
dict[w] = w
|
|
words.append(word)
|
|
return words
|
|
|
|
def getprevword(self):
|
|
"Return the word prefix before the cursor."
|
|
line = self.text.get("insert linestart", "insert")
|
|
i = len(line)
|
|
while i > 0 and line[i-1] in self.wordchars:
|
|
i = i-1
|
|
return line[i:]
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from unittest import main
|
|
main('idlelib.idle_test.test_autoexpand', verbosity=2)
|