Lib/idlelib/AutoExpand.py

   1 import string
   2 import re
   3
   4 ###$ event <<expand-word>>
   5 ###$ win <Alt-slash>
   6 ###$ unix <Alt-slash>
   7
   8 class AutoExpand:
   9
  10     keydefs = {
  11         '<<expand-word>>': ['<Alt-slash>'],
  12     }
  13
  14     unix_keydefs = {
  15         '<<expand-word>>': ['<Meta-slash>', '<Alt-slash>'],
  16     }
  17
  18     menudefs = [
  19         ('edit', [
  20             ('E_xpand word', '<<expand-word>>'),
  21          ]),
  22     ]
  23
  24     wordchars = string.letters + string.digits + "_"
  25
  26     def __init__(self, editwin):
  27         self.text = editwin.text
  28         self.state = None
  29
  30     def expand_word_event(self, event):
  31         curinsert = self.text.index("insert")
  32         curline = self.text.get("insert linestart", "insert lineend")
  33         if not self.state:
  34             words = self.getwords()
  35             index = 0
  36         else:
  37             words, index, insert, line = self.state
  38             if insert != curinsert or line != curline:
  39                 words = self.getwords()
  40                 index = 0
  41         if not words:
  42             self.text.bell()
  43             return "break"
  44         word = self.getprevword()
  45         self.text.delete("insert - %d chars" % len(word), "insert")
  46         newword = words[index]
  47         index = (index + 1) % len(words)
  48         if index == 0:
  49             self.text.bell()            # Warn we cycled around
  50         self.text.insert("insert", newword)
  51         curinsert = self.text.index("insert")
  52         curline = self.text.get("insert linestart", "insert lineend")
  53         self.state = words, index, curinsert, curline
  54         return "break"
  55
  56     def getwords(self):
  57         word = self.getprevword()
  58         if not word:
  59             return []
  60         before = self.text.get("1.0", "insert wordstart")
  61         wbefore = re.findall(r"\b" + word + r"\w+\b", before)
  62         del before
  63         after = self.text.get("insert wordend", "end")
  64         wafter = re.findall(r"\b" + word + r"\w+\b", after)
  65         del after
  66         if not wbefore and not wafter:
  67             return []
  68         words = []
  69         dict = {}
  70         # search backwards through words before
  71         wbefore.reverse()
  72         for w in wbefore:
  73             if dict.get(w):
  74                 continue
  75             words.append(w)
  76             dict[w] = w
  77         # search onwards through words after
  78         for w in wafter:
  79             if dict.get(w):
  80                 continue
  81             words.append(w)
  82             dict[w] = w
  83         words.append(word)
  84         return words
  85
  86     def getprevword(self):
  87         line = self.text.get("insert linestart", "insert")
  88         i = len(line)
  89         while i > 0 and line[i-1] in self.wordchars:
  90             i = i-1
  91         return line[i:]