Make Elasticsearch searched fields configurable

This allows fields stuch as "content" to be searchable when using Elasticsearch
Add config item ELASTICSEARCH_FIELDS, list of fields to search.
Defaults to just "name" to preserve existing behaviour
Available fields are:
- "name"
- "username"
- "content"
- "message"
This commit is contained in:
Simon Edwards 2015-07-24 13:29:24 +01:00
vanhempi ceec6b6789
commit 8e73a32875
2 muutettua tiedostoa jossa 6 lisäystä ja 3 poistoa

Näytä tiedosto

@ -99,6 +99,7 @@ SEARCH_TYPE = 'simple' # simple is not good for large wikis
# SEARCH_TYPE = 'elasticsearch'
ELASTICSEARCH_URL = 'http://127.0.0.1:9200'
ELASTICSEARCH_FIELDS = ["name"]
# SEARCH_TYPE = 'whoosh'
WHOOSH_INDEX = '/tmp/whoosh'

Näytä tiedosto

@ -14,7 +14,8 @@ def whoosh(app):
def elasticsearch(app):
from flask.ext.elastic import Elastic
return ElasticSearch(Elastic(app))
fields = app.config.get('ELASTICSEARCH_FIELDS')
return ElasticSearch(Elastic(app), fields)
class Search(object):
@ -125,8 +126,9 @@ class WhooshSearch(BaseSearch):
class ElasticSearch(BaseSearch):
def __init__(self, elastic):
def __init__(self, elastic, fields):
self.elastic = elastic
self.fields = fields
def index(self, index, doc_type, id_=None, body=None):
return self.elastic.index(index=index, doc_type=doc_type, id=id_, body=body)
@ -144,7 +146,7 @@ class ElasticSearch(BaseSearch):
res = self.elastic.search(index='wiki', body={"query": {
"multi_match": {
"query": query,
"fields": ["name"]
"fields": self.fields
}}})
return [hit["_source"] for hit in res['hits']['hits']]