1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
from searchparser import SearchQueryParser
products = [
"grape juice",
"grape jelly",
"orange juice",
"orange jujubees",
"strawberry jam",
"prune juice",
"prune butter",
"orange marmalade",
"grapefruit juice",
]
class FruitSearchParser(SearchQueryParser):
def GetWord(self, word):
return {p for p in products if p.startswith(word + " ")}
def GetWordWildcard(self, word):
return {p for p in products if p.startswith(word[:-1])}
def GetQuotes(self, search_string, tmp_result):
result = set()
# I have no idea how to use this feature...
return result
def GetNot(self, not_set):
return set(products) - not_set
parser = FruitSearchParser()
tests = """\
grape or orange
grape*
not(grape*)
prune and grape""".splitlines()
for t in tests:
print(t.strip())
print(parser.Parse(t))
print("")
|
