Merge pull request #2 from Safecast/add-back-es-host

Added back ELASTICSEARCH_HOST.
Safecast · Sep 13, 2020 · 1678411 · 1678411
2 parents d6e3ecd + 8de37fe
commit 1678411
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -12,7 +12,12 @@
 def get_data():
     config = configparser.ConfigParser()
     config.read('config.ini')
-    es = Elasticsearch('https://{}:{}@40ad140d461d810ac41ed710b5c7a5b6.us-west-2.aws.found.io:9243/'.format(config['ELASTICSEARCH']['Username'], config['ELASTICSEARCH']['Password']))
+
+    host = os.environ.get('ELASTICSEARCH_HOST')
+    if host is None:
+        host = 'https://{}:{}@40ad140d461d810ac41ed710b5c7a5b6.us-west-2.aws.found.io:9243/'.format(
+            config['ELASTICSEARCH']['Username'], config['ELASTICSEARCH']['Password'])
+    es = Elasticsearch(host)
 
     # query
     week_ago = date.today() - timedelta(days=7)
@@ -48,4 +53,4 @@ def filter():
     return render_template('map.html', data=data)
 
 if __name__ == "__main__":
-    app.run(debug=True)
+    app.run(debug=True)