simon987/od-database
1
0
Fork 0
mirror of https://github.com/simon987/od-database.git synced 2025-12-16 08:09:04 +00:00
Code Issues Projects Releases Wiki Activity

Filter by extension type

Browse Source
This commit is contained in:
Simon
2018-06-26 19:02:46 -04:00
parent a0bd45c829
commit 8ea57967e6
4 changed files with 39 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
search/search.py
View File

@@ -16,7 +16,7 @@ class SearchEngine:
def import_json(self, in_str: str, website_id: int):
raise NotImplementedError
def search(self, query, page, per_page, sort_order) -> {}:
def search(self, query, page, per_page, sort_order, extension) -> {}:
raise NotImplementedError
def reset(self):
@@ -142,9 +142,11 @@ class ElasticSearchEngine(SearchEngine):
action_string = '{"index":{}}\n'
return "\n".join("".join([action_string, ujson.dumps(doc)]) for doc in docs)
def search(self, query, page, per_page, sort_order) -> {}:
def search(self, query, page, per_page, sort_order, extensions) -> {}:
filters = []
if extensions:
filters.append({"terms": {"ext": extensions}})
sort_by = ElasticSearchEngine.SORT_ORDERS.get(sort_order, [])
page = self.es.search(body={