elasticsearch8

import os from elasticsearch import Elasticsearch # --- Configuration for connecting to Elasticsearch --- # For Elastic Cloud (recommended) CLOUD_ID = os.environ.get("ELASTIC_CLOUD_ID", "") API_KEY = os.environ.get("ELASTIC_API_KEY", "") # Base64 encoded API Key ID and Key # For self-managed Elasticsearch ES_HOST = os.environ.get("ELASTICSEARCH_HOST", "http://localhost:9200") ES_USERNAME = os.environ.get("ELASTICSEARCH_USERNAME", "elastic") ES_PASSWORD = os.environ.get("ELASTICSEARCH_PASSWORD", "changeme") client = None if CLOUD_ID and API_KEY: print("Connecting to Elastic Cloud...") client = Elasticsearch( cloud_id=CLOUD_ID, api_key=API_KEY, ) elif ES_HOST: print(f"Connecting to self-managed Elasticsearch at {ES_HOST}...") client = Elasticsearch( hosts=[ES_HOST], basic_auth=(ES_USERNAME, ES_PASSWORD), verify_certs=False # CAUTION: Only for local dev/testing without proper CA config ) else: raise ValueError( "Elasticsearch connection details not provided. Set ELASTIC_CLOUD_ID/ELASTIC_API_KEY " "or ELASTICSEARCH_HOST/ELASTICSEARCH_USERNAME/ELASTICSEARCH_PASSWORD environment variables." ) # --- Verify Connection --- if client.ping(): print("Successfully connected to Elasticsearch!") else: print("Could not connect to Elasticsearch. Please check your configuration.") exit(1) # --- Index a Document --- index_name = "my-documents" doc_id = "1" document = { "author": "John Doe", "text": "Elasticsearch is a powerful open-source search and analytics engine.", "timestamp": "2024-05-15T12:00:00Z" } print(f"Indexing document with ID '{doc_id}' into index '{index_name}'...") response = client.index(index=index_name, id=doc_id, document=document) print(f"Indexed document: Result - {response['result']}") # --- Search for Documents --- print(f"Searching for documents in index '{index_name}'...") search_query = {"match": {"text": "search engine"}} response = client.search(index=index_name, query=search_query) print(f"Found {response['hits']['total']['value']} hits:") for hit in response['hits']['hits']: print(f" ID: {hit['_id']}, Source: {hit['_source']}") # --- Clean Up (Optional) --- # client.indices.delete(index=index_name, ignore=[400, 404]) # print(f"Index '{index_name}' deleted (if it existed).") client.close() print("Client connection closed.")