config: Add Elasticsearch hosts

config.example.py

@@ -144,6 +144,8 @@ ENABLE_ELASTIC_SEARCH_HIGHLIGHT = False
 ES_MAX_SEARCH_RESULT = 1000
 # ES index name generally (nyaa or sukebei)
 ES_INDEX_NAME = SITE_FLAVOR
+# ES hosts
+ES_HOSTS = ['localhost:9200']
 
 ################
 ## Commenting ##

import_to_es.py

@@ -18,7 +18,7 @@ from nyaa import create_app, models
 from nyaa.extensions import db
 
 app = create_app('config')
-es = Elasticsearch(timeout=30)
+es = Elasticsearch(hosts=app.config['ES_HOSTS'], timeout=30)
 ic = IndicesClient(es)
 
 def pad_bytes(in_bytes, size):

nyaa/search.py

@@ -185,7 +185,7 @@ def search_elastic(term='', user=None, sort='id', order='desc',
     if page > 4294967295:
         flask.abort(404)
 
-    es_client = Elasticsearch()
+    es_client = Elasticsearch(hosts=app.config['ES_HOSTS'])
 
     es_sort_keys = {
         'id': 'id',

sync_es.py

@@ -34,7 +34,11 @@ from elasticsearch.helpers import bulk, BulkIndexError
 from pymysqlreplication import BinLogStreamReader
 from pymysqlreplication.row_event import UpdateRowsEvent, DeleteRowsEvent, WriteRowsEvent
 from datetime import datetime
+
+from nyaa import create_app, db, models
 from nyaa.models import TorrentFlags
+app = create_app('config')
+
 import sys
 import json
 import time
@@ -262,7 +266,7 @@ class EsPoster(ExitingThread):
         self.flush_interval = flush_interval
 
     def run_happy(self):
-        es = Elasticsearch(timeout=30)
+        es = Elasticsearch(hosts=app.config['ES_HOSTS'], timeout=30)
 
         last_save = time.time()
         since_last = 0