elasticsearch:
  host: localhost
  port: 9200
  # 用于数据导入的目标索引
  index: elasticsearch_data2
  # 用于搜索的索引列表
  search_indices: 
    - elasticsearch_data
    - elasticsearch_data1
    - elasticsearch_data2

"""
加载配置文件
"""
def load_config():
    config_path = os.path.join(os.path.dirname(__file__), 'config.yml')
    try:
        with open(config_path, 'r', encoding='utf-8') as f:
            return yaml.safe_load(f)
    except Exception as e:
        print(f"加载配置文件出错: {e}")
        sys.exit(1)

class Searcher:
    """
    初始化 Elasticsearch 连接
    """
    def __init__(self):
        # 加载配置
        self.config = load_config()
        es_config = self.config['elasticsearch']

        self.es = Elasticsearch(
            hosts=[{
                'host': es_config['host'],
                'port': es_config['port'],
                'scheme': 'http'
            }],
            basic_auth=(es_config['username'], es_config['password']),
            verify_certs=False,
            ssl_show_warn=False
        )
        # 使用search_indices配置用于搜索
        self.indices = es_config.get('search_indices', [es_config.get('index')])