搞定docker所需的通过环境变量控制

elastic/es_connect.py

@@ -4,6 +4,7 @@ Django版本的ES连接和操作模块
 """
 from elasticsearch import Elasticsearch
 from elasticsearch_dsl import connections
+import os
 from .documents import AchievementDocument, UserDocument, GlobalDocument
 from .indexes import ACHIEVEMENT_INDEX_NAME, USER_INDEX_NAME, GLOBAL_INDEX_NAME
 import hashlib
@@ -11,8 +12,11 @@ import time
 from datetime import datetime, timezone, timedelta
 import threading
 
-# 使用Django的ES连接配置
-connections.create_connection(hosts=['localhost:9200'])
+# 使用环境变量配置ES连接，默认为本机
+_ES_URL = os.environ.get('ELASTICSEARCH_URL', 'http://localhost:9200')
+if not (_ES_URL.startswith('http://') or _ES_URL.startswith('https://')):
+    _ES_URL = 'http://' + _ES_URL
+connections.create_connection(hosts=[_ES_URL])
 
 # 获取默认的ES客户端
 es = connections.get_connection()