diff --git a/dsSchoolBuddy/ElasticSearch/Utils/EsSearchUtil.py b/dsSchoolBuddy/ElasticSearch/Utils/EsSearchUtil.py index bbdbb0aa..9747ddf0 100644 --- a/dsSchoolBuddy/ElasticSearch/Utils/EsSearchUtil.py +++ b/dsSchoolBuddy/ElasticSearch/Utils/EsSearchUtil.py @@ -1,7 +1,13 @@ import logging +import warnings +from Config.Config import ES_CONFIG from ElasticSearch.Utils.ElasticsearchConnectionPool import ElasticsearchConnectionPool +# 抑制HTTPS相关警告 +warnings.filterwarnings('ignore', message='Connecting to .* using TLS with verify_certs=False is insecure') +warnings.filterwarnings('ignore', message='Unverified HTTPS request is being made to host') + # 初始化日志 logger = logging.getLogger(__name__) logger.setLevel(logging.INFO) @@ -43,3 +49,28 @@ class EsSearchUtil: # 释放连接回连接池 self.es_pool.release_connection(conn) + +# 添加main函数进行测试 +if __name__ == "__main__": + try: + # 创建EsSearchUtil实例 + search_util = EsSearchUtil(ES_CONFIG) + + # 查询"混凝土" + query = "混凝土" + logger.info(f"开始查询关键词: {query}") + results = search_util.text_search(query, size=5) + + # 打印结果 + logger.info(f"查询到 {len(results['hits']['hits'])} 条结果") + for i, hit in enumerate(results['hits']['hits'], 1): + logger.info(f"结果 {i}:") + logger.info(f"得分: {hit['_score']}") + logger.info(f"内容: {hit['_source'].get('user_input', '无内容')}") + logger.info("-" * 50) + + print(f"查询 '{query}' 完成,共找到 {len(results['hits']['hits'])} 条结果") + + except Exception as e: + logger.error(f"测试失败: {str(e)}") + print(f"测试失败: {str(e)}") diff --git a/dsSchoolBuddy/ElasticSearch/Utils/__pycache__/__init__.cpython-310.pyc b/dsSchoolBuddy/ElasticSearch/Utils/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 00000000..e0a7340c Binary files /dev/null and b/dsSchoolBuddy/ElasticSearch/Utils/__pycache__/__init__.cpython-310.pyc differ