class FindBooks(object):
file_patten = "pattens.txt"
wu_pattens = []
kmp_pattens = []
Wu = WM.WuManber(BookNameConstraint)
Kmp = kmp.KmpSearch(BookNameConstraint)
def __init__(self):
all_pattens = open(self.file_patten).readlines()
for i, item in enumerate(all_pattens):
item = item.strip(" \n")
if len(item) < 5:
self.kmp_pattens.append(item)
else:
self.wu_pattens.append(item)
self.Wu.InitPatten(self.wu_pattens)
self.Kmp.InitPattens(self.kmp_pattens)
def SearchBooks(self, string):
rWu = self.Wu.Search(string)
rKmp = self.Kmp.Search(string)
return [rWu, rKmp]
