Trie with End marker

trie_em.py

class Trie(object):
    """
    Trie with End marker
    """
    END = '_end_'
    current = dict()
    
    def add(self, word):
        """
        Adds a word to Trie
        """
        current = self.current
        for letter in word:
            current = current.setdefault(letter, {})
        current[self.END] = self.END        
    
    def find(self, word, full=False):
        """
        Finds a prefix in Trie depending on full flag
        """
        current = self.current
        for letter in word:
            if letter in current:
                current = current[letter]
            else:
                return False
        else:
            if not full:
                if self.END in current:
                    return True
                else:
                    return False
            else:
                return True