zhanghongliang/zcensor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化敏感字查询

Browse Source
This commit is contained in:
zhl 2021-12-01 16:38:44 +08:00
parent 9685db79be
commit 78a8114871
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lib/text_filter.py
View File

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