|
8
|
1 from pprint import pprint
|
|
|
2 from typing import Iterable
|
|
|
3 from apexsearch import ApexSearch
|
|
|
4
|
|
|
5
|
|
|
6 class Search:
|
|
|
7
|
|
|
8 def __init__(self):
|
|
|
9 self.apex = ApexSearch('data/apex',
|
|
|
10 tables={
|
|
|
11 "docs": {
|
|
|
12 "content": ["phrase"],
|
|
|
13 "title": "title",
|
|
|
14 "extras": ["sourceFile", "pos"],
|
|
|
15 }
|
|
|
16 },
|
|
|
17 id_field='id')
|
|
|
18
|
|
|
19 def rebuild(self, docs: Iterable):
|
|
|
20 self.apex.build_complete_index(lambda *a: docs)
|
|
|
21 print('rebuild complete')
|
|
|
22
|
|
|
23 def search(self, q:str):
|
|
|
24 res= self.apex.search(q ,target_number=100)
|
|
|
25 pprint(res)
|
|
|
26 for row in res['results']:
|
|
|
27 yield {'title':row['title'], 'snippetHtml':row['highlighted_content']} |
