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 set( p for p in products if p.startswith(word + " ") )

    def GetWordWildcard(self, word):
        return set( 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()