Poodletooth-iLand/otp/chat/WhiteList.py

from bisect import bisect_left

class WhiteList:
    def __init__(self, words):
        self.words = words
        self.numWords = len(self.words)

    def cleanText(self, text):
        text = text.strip('.,?!')
        return text.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)

        if i == self.numWords:
            return False

        return self.words[i].startswith(text)

    def prefixCount(self, text):
        text = self.cleanText(text)
        i = bisect_left(self.words, text)
        j = i
        while j < self.numWords and self.words[j].startswith(text):
            j += 1

        return j - i

    def prefixList(self, text):
        text = self.cleanText(text)
        i = bisect_left(self.words, text)
        j = i
        while j < self.numWords and self.words[j].startswith(text):
            j += 1

        return self.words[i:j]