diff --git a/search_index.py b/search_index.py index bf91c88dbdc14916e4eacfa2fbca6f2f8839c822..b1b55de8255c581ad3194d1850ef7feb0e1a9fa4 100644 --- a/search_index.py +++ b/search_index.py @@ -22,9 +22,12 @@ class SearchIndex: writer.add_document(path=path, page=page, content=content) writer.commit() - def search(self, query): + def result_list(self, query): query_parser = QueryParser("content", self.ix.schema, group=OrGroup.factory(0.9)) - return self.ix.searcher().search(query_parser.parse(query))[0] + return self.ix.searcher().search(query_parser.parse(query)) + + def search(self, query): + return self.result_list(query)[0] if __name__ == "__main__":