whoosh_quickstart_schema_index_search_example.py

1import os
2from whoosh.index import create_in
3from whoosh.fields import Schema, TEXT, ID
4from whoosh.qparser import QueryParser
5
6# 1. Define the schema
7# "stored=True" means the value of the field is saved in the index 
8# so it can be returned in search results.
9schema = Schema(title=TEXT(stored=True), path=ID(stored=True), content=TEXT)
10
11# 2. Create the index directory and the index
12if not os.path.exists("indexdir"):
13    os.mkdir("indexdir")
14ix = create_in("indexdir", schema)
15
16# 3. Add documents to the index
17writer = ix.writer()
18writer.add_document(title=u"First document", path=u"/a",
19                    content=u"This is the first document we've added!")
20writer.add_document(title=u"Second document", path=u"/b",
21                    content=u"The second one is even more interesting!")
22writer.commit()
23
24# 4. Search the index
25with ix.searcher() as searcher:
26    query = QueryParser("content", ix.schema).parse("first")
27    results = searcher.search(query)
28    
29    # Print the results
30    print(f"Found {len(results)} results:")
31    for hit in results:
32        print(hit["title"])