archive/pillar
0
0
Fork 0
Code Issues 42 Pull Requests Projects Releases Wiki Activity

Allow importing pillar.api.search.index outside of app context

Browse Source
This commit is contained in:
Sybren A. Stüvel 2017-12-08 16:07:10 +01:00
parent 3bdd5197f5
commit c15fffa11f
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
pillar/api/search/index.py
View File

@ -21,13 +21,15 @@ class ResetIndexTask(object):
"""
Clear and build index / mapping
"""
index = ''
index_key = ''
"""Key into the ELASTIC_INDICES dict in the app config."""
doc_types = []
name = 'remove index'
def __init__(self):
if not self.index:
if not self.index_key:
raise ValueError("No index specified")
if not self.doc_types:
@ -40,16 +42,16 @@ class ResetIndexTask(object):
)
def execute(self):
idx = es.Index(self.index)
index = current_app.config['ELASTIC_INDICES'][self.index_key]
idx = es.Index(index)
try:
idx.delete(ignore=404)
log.info("Deleted index %s", self.index)
log.info("Deleted index %s", index)
except AttributeError:
log.warning("Could not delete index '%s', ignoring", self.index)
log.warning("Could not delete index '%s', ignoring", index)
except NotFoundError:
log.warning("Could not delete index '%s', ignoring", self.index)
log.warning("Could not delete index '%s', ignoring", index)
# create doc types
for dt in self.doc_types:
@ -60,15 +62,13 @@ class ResetIndexTask(object):
class ResetNodeIndex(ResetIndexTask):
index = current_app.config['ELASTIC_INDICES']['NODE']
index_key = 'NODE'
doc_types = [documents.Node]
class ResetUserIndex(ResetIndexTask):
index = current_app.config['ELASTIC_INDICES']['USER']
index_key = 'USER'
doc_types = [documents.User]
def reset_node_index():
resettask = ResetNodeIndex()
resettask.execute()