mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-10-31 00:37:54 +00:00
47 lines
No EOL
1.4 KiB
Python
Executable file
47 lines
No EOL
1.4 KiB
Python
Executable file
from bisect import bisect_left
|
|
|
|
class WhiteList:
|
|
|
|
def setWords(self, words):
|
|
self.words = words
|
|
self.numWords = len(self.words)
|
|
|
|
def cleanText(self, text):
|
|
return text.strip('.,?!').lower()
|
|
|
|
def isWord(self, text):
|
|
return self.cleanText(text) in self.words
|
|
|
|
def isPrefix(self, text):
|
|
text = self.cleanText(text)
|
|
i = bisect_left(self.words, text)
|
|
|
|
return i != self.numWords and self.words[i].startswith(text)
|
|
|
|
def getReplacement(self, text, av=None, garbler=None):
|
|
return '\x01WLRed\x01%s\x02' % text if not garbler else garbler.garble(av, text)
|
|
|
|
def processText(self, text, av=None, garbler=None):
|
|
if (not self.words) or text.startswith('~'):
|
|
return text
|
|
|
|
words = text.split(' ')
|
|
newWords = []
|
|
|
|
for word in words:
|
|
if (not word) or self.isWord(word):
|
|
newWords.append(word)
|
|
else:
|
|
newWords.append(self.getReplacement(word, av, garbler))
|
|
|
|
lastWord = words[-1]
|
|
|
|
if (not lastWord) or self.isPrefix(lastWord):
|
|
newWords[-1] = lastWord
|
|
else:
|
|
newWords[-1] = self.getReplacement(lastWord, av, garbler)
|
|
|
|
return ' '.join(newWords)
|
|
|
|
def processThroughAll(self, text, av=None, garbler=None):
|
|
return self.processText(text, av, garbler) |