优化敏感字查询

This commit is contained in:
zhl 2021-12-01 16:38:44 +08:00
parent 9685db79be
commit 78a8114871

View File

@ -43,6 +43,7 @@ class TextFilter(object):
# Singleton
_instance = None
root = None
def __new__(cls, *args, **kw):
"""单例模式"""
@ -51,11 +52,12 @@ class TextFilter(object):
return cls._instance
def __init__(self):
self.root = Node()
self.sensitive_word = SensitiveWords().sensitive_word_dict
for word_type in self.sensitive_word.keys():
for word in self.sensitive_word[word_type]:
self.add_word(word, word_type)
if self.root == None:
self.root = Node()
self.sensitive_word = SensitiveWords().sensitive_word_dict
for word_type in self.sensitive_word.keys():
for word in self.sensitive_word[word_type]:
self.add_word(word, word_type)
def add_word(self, word, word_type=u'common'):
"""向字典树里添加敏感词汇及敏感词类型"""